파이썬 리스트 2개 한번에 for문 돌리기

파이썬으로 다양한 코드를 짜다 보면

리스트를 정말 많이 사용한다.

 

특히 계속 꺼내서 쓰는 일이 많아서 그런데,

서로 다른 2개의 리스트에서 꺼내 쓸 일이 생긴다.

 

예를들어 아래처럼 파이토치에서 loss를 별도로 계산하고자 할 때 생긴다.

output 여러개, label 여러개를 가져왔는데 하나씩 꺼내서 loss를 계산해야 된다.

 

대충 이런 상황에서 다음과 같이 코딩하면 에러가 난다.

for i, j in [1], [2]:
	print(i + j)

error

 

이럴땐 zip() 함수를 사용해서 묶어주면 파이썬이 알아먹는다.

for i, j in zip([1], [2]):
	print(i + j)

output

 

zip(리스트1, 리스트2, ....) 하면 된다.

 

끝.