반응형

파이썬, 중첩 컴프리헨션 이용하기

 

글. 수알치 오상문

 

아래 예제를 보자.

data1의 자료를 [1, 2, 3, 4, 5, 6, 7, 8, 9] 형태로 변경하려고 한다.

간편하게 이중 for 문을 이용하면 이렇게 작성할 수 있다.

 

data1 = [[1,2,3], [4,5], [6,7,8,9]]

 

data2 = []
for item in data1:
    for v in item:
        data2.append(v)


print(data2) 

 

[결과]

[1, 2, 3, 4, 5, 6, 7, 8, 9]

 

이 구조를 중첩 컴프리헨션 구조로 바꾸면 다음과 같다.

(가독성은 떨어지지만 코드가 간단해졌다.)

 

data1 = [[1,2,3], [4,5], [6,7,8,9]]


data2 = [for item in data1 for v in item]


print(data2)

 

[결과] 

[1, 2, 3, 4, 5, 6, 7, 8, 9]


반응형

+ Recent posts