개발 이야기/Python
Python 이중 for loop 를 itertools 를 활용하여 개선하기
_ppuing
2024. 2. 26. 09:25
반응형
a = [1,2,3,4]
b = ['A', 'B']
의 두 리스트에 대해 (1,A), (1,B), (2,A), (2,B), ... 반복을 하려면 무지성으로 구현했을 때
for x in a:
for y in b:
function(x, y)
처럼 구현할 수 있다.
하지만 itertools 의 product 를 활용하면 좀 더 깔끔하게 구현이 가능하다
[function(x, y) for x, y in product(a, b)]
반응형