Python 사용시 다차원 배열 또는 튜플을 많이 사용 하곤 합니다.

이렇게 다차원으로 구성된 배열의 일부 Column 만을 뽑아 내고 싶은 경우가 종종 있습니다. 

이런 경우에는 zip() 함수를 사용하면 편리 합니다 .

아래 코드는 그 예제이며 tuple 이나 list 에서 똑같이 동작 합니다. 

print "-"*30 + "Tuple" + "-"*30

Multidimensional_tuple=((1,2,3),(4,5,6),(7,8,9))
print Multidimensional_tuple

Col1, Col2, Col3 = zip(*Multidimensional_tuple)

print Col1
print Col2
print Col3

# -------------------------

print "-"*30 + "Array" + "-"*30
Multidimensional_array=[[1,2,3],[4,5,6],[7,8,9]]
print Multidimensional_array

Col1, Col2, Col3 = zip(*Multidimensional_array)

print Col1
print Col2
print Col3

 위 코드에 따른 결과는 아래와 같습니다. 


각각의 모든 column 이 아니라 특정 column 만 추출하고 싶다면 다음과 같이 안 쓰는 부분들을 _(underscore) 로 처리 하면 됩니다. 

아래 예는 2 번째 column 만 추출 하는 경우의 코드 입니다.

_, Col2, _ = zip(*Multidimensional_array)


+ Recent posts