[python3] itertools 이터레이터를 만드는 함수 종류 살펴보기

 

1. Itertools

• 파이썬에서 제공하는 자신만의 반복자를 생성해주는 함수이다.
• 특정 배열이나 순열에 대한 조합을 만들어서 이를 이용하는 알고리즘 문제를 풀때 유용하게 사용된다.
• 효율적인 반복자를 구할 수 있다.
• 무한 이터레이터 / 조합형 이터레이터 / 종료 이터레이터 가 있지만, 종료 이터레이터는 너무 많으니 아래 링크로 첨부해두고, 자주 사용할 것 같은 이터레이터만 작성할 것이다.

 

2. 무한 이터레이터

 2.1 Count()

from itertools import count
a=count(10,10)

#-------호출----------------------
next(a)
#결과 : 10

next(a)
#결과 : 20 
.
.
.

• count(시작숫자, 더해줄숫자) 입력해주면 무한으로 계속 규칙에 맞게 호출된다.

 

2.2 Cycle()

from itertools import cycle
a = cycle([1,2,3,4,5])

#--------결과호출------------
next(a)
# 결과 : 1

next(a)
# 결과 : 2

• a 의 cycle에 입력된 리스트가 반복되어 cycle에 들어간다. 따라서 5 다음의 next는 1로 다시 처음으로 돌아가는 무한 반복 사이클을 생성한다.

 

2.3 repeat()

from itertools import repeat
a = repeat("AAA", 4)

print(next(a))
print(next(a))
print(next(a))
print(next(a))
print(next(a))

• repeat(반복할것, 반복횟수) 로 지정
• 위의 예시에서는 4번 의 AAA를 출력한 후, 다섯번째는 출력을 못하고 오류가 나는 것을 볼 수 있다.

 

3. 조합 이터레이터

3.1 product()

from itertools import product

a = ["a", "b", "c"]
pd = list(product(a, repeat = 3))

• product(list, repeat = n) repeat 부분에는 몇개 씩 조합할건지 쓰는 거라고 이해하면 된다.
• 데카르트의 곱 . 중첩된 for루프와 동일
• 두 개 이상의 리스트의 모든 조합을 구할 때 주로 사용된다.
• 리스트 a 에 있는 모든 수의 조합 방법이 pd에 들어갔다. 
• 중복이 허용되는 것을 볼 수가 있다. 
• len(pd) 는 27이 나온다 (3개를 중복을 허용한 채 조합해서 나올 수 있는 모든 경우의 수)

st = list(product('ABCD', repeat=2))
st

• 리스트이던, str형태이던 상관없다.

# 두 개 이상의 조합을 구할 때
from itertools import product

a = ["a", "b", "c"]
b = [1,2,3]

pd = list(product(a, b, repeat=1))

 

3.2 permutations()

from itertools import permutations

a = [1, 2, 3]
permut = list(permutations(a, 3))
print(permut)

• 한 리스트 내 에서만 중복을 허용한 채로 모든 경우의 수를 구한다.
• 순열을 표현할 때 사용되는 메서드 이다.

 

3.3 combinations()

from itertools import combinations

a = [1, 2, 3]
combi = list(combinations(a, 2))
print(combi)

• 조합을 구하는 메서드 이다.
• 리스트에서 중복을 허용하지 않고 모든 경우의 수를 구한다.

# 갯수 별로 조합을 반복할 수 있다.
a = [1, 2, 3]
for i in range(1, len(a) + 1):
    print(list(combinations(a, i)))

• 바로 위에 처럼 for문 안에 넣고, 갯수별로 조합을 생성할 수 있다. 

 

3.4 combinations_with_replacement()

combinations과 동일한데, 자기자신과의 조합까지 출력해준다. 아래 두 메서드의 차이점을 보면 이해가 될 것이다.

https://docs.python.org/ko/3.8/library/itertools.html#itertools.count

 

 

참고자료

https://docs.python.org/ko/3.8/library/itertools.html#itertools.count

itertools — 효율적인 루핑을 위한 이터레이터를 만드는 함수 — Python 3.8.11 문서

https://velog.io/@davkim1030/Python-%EC%88%9C%EC%97%B4-%EC%A1%B0%ED%95%A9-product-itertools