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 배열에 열을 추가하는 방법

의견을 추가하다

이메일 주소는 공개되지 않습니다. 필수 필드는 *로 표시됩니다