combinations, permutations에서 요소 추출

combinations이나 permutations 을 사용하면 튜플 자료형이 나오는데, for i ,j in arr로 두 개를 뽑아서 사용할 수도 있지만, 튜플 안의 자료를 모두 뽑아서 써야할 상황이 생긴다. 그때마다 요긴하게 쓰일 수 있는 트릭이다. 

from itertools import *  # 모듈 모두 불러오기

numbers = list(map(str,range(3)))

arr = list(permutations(numbers, 2))
print(arr)
# [('0', '1'), ('0', '2'), ('1', '0'), ('1', '2'), ('2', '0'), ('2', '1')]

arr = list(map(lambda x: int(''.join(x)), arr))
print(arr)
# [1, 2, 10, 12, 20, 21]

 

set([int(('').join(p)) for p in arr])

이렇게 작성해도 된다.