본문 바로가기
개발 이야기/Python

Python 이중 for loop 를 itertools 를 활용하여 개선하기

by _ppuing 2024. 2. 26.
반응형

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)]

 

반응형

댓글