[Python] 파이썬 2차원 배열에서 특정 열(column) 추출하기

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

 

1. inline for loop

b = [i[0] for i in a]
print(b)
# [0, 2, 4, 6, 8]

 

2. zip 이용하기

b = list(zip(*a))[0]
print(b)
# (0, 2, 4, 6, 8)

 

3. numpy 이용하기

import numpy as np
b = np.array(a).T[0]
print(b)
# [0 2 4 6 8]