[Python] 코딩테스트 문법 정리

📌 딕셔너리 정렬

sorted(dict.items(), key=lambda x: (-x[1], x[0]))

 

📌 defaultdict()

from collections import defaultdict

int_dict = defaultdict(int)

 

📌 아스키코드 변환 : ord(), chr()

ord('a')  # 97
chr(97)   # 'a'

 

📌 이진수 변환

bin(42)  # '0b101010'

 

📌 절대값 : abs()

abs(-10)  # 10

 

📌 heapq

import heapq

arr = [3, 2, 1]
heapq.heapify(arr)  # [1, 2, 3]
heapq.heappush(arr, 4)  # [1, 2, 3, 4]
heapq.heappop(arr)  # 1

 

📌 재귀 깊이 제한 설정

import sys

sys.setrecursionlimit(10**6)

 

📌 무한대

float('inf')  # 양의 무한대
float('-inf')  # 음의 무한대

 

📌 eval()

eval("1+2")  # 3
evel(len([1, 2, 3, 4]))  # 4

 

📌 Unpacking

📂 가변 인자

• 인자의 갯수가 몇 개가 될지 확실하지 않을 때 사용

list_ = [1, 2, 3, 4, 5]
first, *rest, last = list_
print(rest)  # [2, 3, 4]

📂 List Unpacking

• 컨테이너형 구조(ex: tuple, set, ...)에서 모두 적용 가능

list_ = [1, 2, 3, 4, 5]
print(*list_, sep=' / ')  # 1 / 2 / 3 / 4 / 5

 

📌 Set

• 값을 찾기 위해서는 set에서 in을 사용해야 함
  • list를 사용하면 순차적으로 탐색함을 주의!

 

📌 Dictionary

fruit = ['apple', 'grape', 'orange', 'banana']
price = [3200, 15200, 9800, 5000]

dict_ = dict(zip(fruit, price)) # {'apple' : 3200, 'grape' : 15200, 'orange' : 9000, 'banana' : 5000}

 

📌 Combination/Permutation

import itertools

list_ = [1, 2, 3, 4]

iter = itertools.combinations(list_, 2) # 12 13 14 23 24 34
iter = itertools.permutations(list_, 2) # 12 13 14 21 23 24 31 32 34 41 42 43
iter = itertools.combinations_with_replacement(list_, 2) # 11 12 13 14 22 23 24 33 34 44
iter = itertools.product(list_, repeat=2) # 11 12 13 14 21 22 23 24 31 32 33 34 41 42 43 44

 

📌 Counter

from collections import Counter

Counter('hello world').most_common() # [('l', 3), ('o', 2), ('h', 1), ('e', 1), (' ', 1), ('w', 1), ('r', 1), ('d', 1)]
Counter('hello world').most_common(2) # [('l', 3), ('o', 2)]