[python] list 요소 2개씩 묶기

다음과 같이 짝수개 요소가 들어있는 list를 보고, 두개씩 짝지어서 return하고 싶을 때 간단한 기록용

 

#for문을 쓰면 직관적이지만, 깔끔하지 못함

test = [1, 2, 3, 4, 5, 6, 7, 8]
new = []
for i in range(0, len(test), 2):
    new.append([test[i], test[i+1]])
    
print(new) #[[1, 2], [3, 4], [5, 6], [7, 8]]

 

한줄로 묶기위해선 zip과 list slicing을 활용

# 한 줄로 작성
test = [1,2,3,4,5,6,7,8]
result = [[x, y] for x, y in zip(test[0::2], test[1::2])]
print(result) # [[1, 2], [3, 4], [5, 6], [7, 8]]