from itertools
combinations_with_replacement(iter, r) : A~E 5명의 후보가 있다. 중복을 허용해 2명의 대표를 뽑는 방법
from itertools import combinations_with_replacement
list(combinations_with_replacement('ABC', 2)) #결과 : 'AA' 'AB' 'AC' 'BB' 'BC' 'CC' product(iter, r) : 1~5 숫자카드가 있다. 중복을 허용해 2자리수를 만드는 방법.
from itertools import product
list(product('ABC', repeat=2)) #결과 : 'AA' 'AB' 'AC' 'BA' 'BB' 'BC' 'CA' 'CB' 'CC'import heapq
heapq.heapify( 리스트 )
a = [4,1,3,10]
heapq.heapify(a) #결과: a는 [1, 4, 3, 10]-> 힙에서 인덱스 0이 최솟값이라고 해서, 인덱스 1에 두 번째, 인덱스 2에 세 번째로 작은 원소가 있는 것이 아니다. 두 번째로 작은 원소를 얻으려면 heappop()을 통해 최솟값을 삭제한 후 heap[0]로 접근하는 방법을 사용하거나, 인덱스 1을 인덱스 2와 비교하는 방법을 사용해야 한다. (최소값만 보장)
from collections import defaultdict
defaultdict(int)
defaultdict(list)딕셔너리에 items() 메서드를 사용해주면 {"key" : value}의 형태를 [(key, value)]의 형태로 만들어 준다.
d[k] = v # (k, v) 추가
d.pop(k) # key가 k인 쌍 찾아 제거
k in d # 현재 hashmap에 key가 k인 쌍이 있는지 확인key 가능 type immutable한 값 -> int, char 등의 primitive type과 string, tuple 등 가능, mutable한 list, dict 등의 type은 가변적이기 때문에 불가
print(d.items()) #dict_items([(1, 3), (2, 4), (-1, 6)])
print(d.keys()) #dict_keys([1, 2, -1])
print(d.values()) #dict_values([3, 4, 6])sorted(d) #['blue', 'green', 'red']: list 형태로 returnsorted(d.items(), key=lambda x : x[1]) #[('red', 3), ('blue', 3), ('green', 1)]: list 형태로 returnfor key in d:
d[key].sort()s=set()
s.add(e) # 데이터 e 추가
s.remove(e) # 데이터 중 숫자 e 찾아 제거
e in s # 현재 hashset에 숫자 e가 들어 있는지 확인# a: {1,2,6}, b: {2,6,9}
a|b #{1,2,6,9}
a&b #{2,6}
a-b #{1}이진 탐색 모듈
해당 값이 위치한 마지막 index 표시
import bisect
mylist = [1, 2, 3, 3, 7, 9, 11, 33]
print(bisect.bisect(mylist, 3)) # 4 (bisect = bisect_right)target 값 존재하지 않아도 index 출력하므로 target 값과 동일한지 check하는 로직 필요요
for _ in range(m):
target = int(input())
index = bisect.bisect_right(lst, target)
if lst[index-1] == target:
print(index)
else:
print(-1)
import bisect
mylist = [1, 2, 3, 3, 7, 9, 11, 33]
print(bisect.bisect_right(mylist, 3) - bisect.bisect_left(mylist, 3))import math
math.gcd( 숫자들 )
math.gcd(12,8) //결과: 4요소가 딕셔너리 키로 저장되고 개수가 딕셔너리 값으로 저장되는 컬렉션
from collections import Counter
Counter( [리스트] or [문자열] ).most_common( 출력갯수 )
"별똥별".count("별") #결과: 2문자열을 정렬하고 싶을 땐 sort한 다음에 join으로 묶어줘야 한다. (sort하면 리스트를 반환한다)
orders[i] = "".join(sorted(orders[i]))문자열 값을 변경하고 싶을땐 list로 변환 후 idx에 해당하는 값을 수정하고 join으로 묶어준다.
find() 함수는 왼쪽부터 찾고, rfind() 함수는 오른쪽부터 찾는다.
- 문자가 없다면 -1을 반환
- 문자가 여러 개 있다면 맨 마지막 인덱스 반환
"별똥별".find("별") #결과: 0
"별똥별".rfind("별") #결과: 2- 문자열.swapcase() # 소문자는 대문자로, 대문자는 소문자로 서로 바꿔주는 메서드
"abc".upper() #결과: "ABC"
"ABC".lower() #결과: "abc"
"aBc".swapcase() #결과: "AbC"- 문자열.strip()
- 문자열.replace(old, new)
- 문자 -> 아스키 코드 : ord('A')
- 아스키 코드 -> 문자 : chr(65)
#1. bin(x)[2:]): 내장함수 bin 사용
#2.
def change_10to2(n):
tmp = []
while n > 0:
tmp.append(str(n%2))
n //= 2
tmp = tmp[::-1]
return tmpint(문자열 값, n)각 iterables 의 요소들을 모으는 이터레이터를 만듭니다. 튜플의 이터레이터를 돌려주는데, i 번째 튜플은 각 인자로 전달된 시퀀스나 이터러블의 i 번째 요소를 포함합니다.
list1 = [1, 2, 3, 4]
list2 = [100, 120, 30, 300]
list3 = [392, 2, 33, 1]
answer = []
for number1, number2, number3 in zip(list1, list2, list3):
print(number1 + number2 + number3)animals = ['cat', 'dog', 'lion']
sounds = ['meow', 'woof', 'roar']
answer = dict(zip(animals, sounds)) # {'cat': 'meow', 'dog': 'woof', 'lion': 'roar'}for 문과 if문을 한번에
mylist = [3, 2, 6, 7]
answer = [number**2 for number in mylist if number % 2 == 0]flag 대신 사용. for loop가 break없이 빠져나올 경우 else문 실행. continue는 break 처럼 동작하지 않음.
a = 3
print(f'{a} 입니다')가장 큰 수 표현
inf = int(1e9)
print(inf > inf) #False
print(inf+1 > inf) #Trueimport sys
sys.exit(0)