Numpy 배열에서 특정 열을 가져오는 방법(예제 포함)

다음 구문을 사용하여 NumPy 배열에서 특정 열을 가져올 수 있습니다.

 #get column in index position 2 from NumPy array
my_array[:, 2]

다음 예에서는 이 구문을 실제로 사용하는 방법을 보여줍니다.

예시 1: NumPy 배열에서 열 가져오기

다음 코드는 NumPy 배열에서 특정 열을 가져오는 방법을 보여줍니다.

 import numpy as np

#create NumPy array
data = np. array ([[1, 2, 3, 4], [5, 6, 7, 8], [9, 10, 11, 12]])

#view NumPy array
data

array([[ 1, 2, 3, 4],
       [5, 6, 7, 8],
       [9, 10, 11, 12]])

#get column in index position 2
data[:, 2]

array([3,7,11])

NumPy 배열에서 열을 가져와 열 벡터 로 검색하려면 다음 구문을 사용할 수 있습니다.

 #get column in index position 2 (as a column vector)
data[:, [2]]

array([[ 3],
       [7],
       [11]])

예시 2: NumPy 배열에서 여러 열 가져오기

다음 코드는 NumPy 배열에서 여러 열을 가져오는 방법을 보여줍니다.

 import numpy as np

#create NumPy array
data = np. array ([[1, 2, 3, 4], [5, 6, 7, 8], [9, 10, 11, 12]])

#view NumPy array
data

array([[ 1, 2, 3, 4],
       [5, 6, 7, 8],
       [9, 10, 11, 12]])

#get columns in index positions 1 and 3 from NumPy array
data[:, [1,3]]

array([[ 2, 4],
       [6, 8],
       [10, 12]])

예시 3: NumPy 배열에서 범위 내의 열 가져오기

다음 코드는 NumPy 배열에서 특정 범위의 열을 가져오는 방법을 보여줍니다.

 import numpy as np

#create NumPy array
data = np. array ([[1, 2, 3, 4], [5, 6, 7, 8], [9, 10, 11, 12]])

#view NumPy array
data

array([[ 1, 2, 3, 4],
       [5, 6, 7, 8],
       [9, 10, 11, 12]])

#get columns in index positions 0 through 3 (not including 3)
data[:, 0:3]

array([[ 1, 2, 3],
       [5, 6, 7],
       [9, 10, 11]])

범위의 마지막 값(이 경우 3)은 반환된 열 범위에 포함되지 않습니다.

추가 리소스

다음 튜토리얼에서는 NumPy에서 다른 일반적인 작업을 수행하는 방법을 설명합니다.

NumPy 배열에 함수를 매핑하는 방법
NumPy 배열에 열을 추가하는 방법