[파이썬] itertools 라이브러리

itertools는 파이썬에서 반복되는 데이터를 처리하는 기능을 포함하는 라이브러리다.

 

itertools 중에서 유용하게 사용할 수 있는 클래스에 대하여 알아본다.

 

1.  permutations

iterable 객체에서 r개의 데이터를 뽑아 일렬로 나열하는 모든 경우를 계산한다.

사용 예:

from itertools import permutations

data = [ 1,2,3]
print(list(permutations(data,2)))  #[(1, 2), (1, 3), (2, 1), (2, 3), (3, 1), (3, 2)]

---

2. product

iterable 객체에서 중복을 허용하여 r개의 데이터를 뽑아 일렬로 나열하는 모든 경우를 계산한다.

사용 예:

from itertools import product

data = [1,2,3]
print(list(product(data, repeat=2))) #[(1, 1), (1, 2), (1, 3), (2, 1), (2, 2), (2, 3), (3, 1), (3, 2), (3, 3)]

---

3. combinations

iterable 객체에서 중복을 허용하지 않고 r개의 데이터를 뽑는 모든 경우를 계산한다.

from itertools import combinations

data = [1,2,3]
print(list(combinations(data,2))) #[(1, 2), (1, 3), (2, 3)]

---

4. combinations_with_replacement

iterable 객체에서 중복을 허용하여 r개의 데이터를 뽑는 모든 경우를 계산한다.

사용 예:

from itertools import combinations_with_replacement

data = [1,2,3]
print(list(combinations_with_replacement(data,2))) #[(1, 1), (1, 2), (1, 3), (2, 2), (2, 3), (3, 3)]