[Python] 2차원 배열 모든 요소에서 첫 번째 열을 선택하고 싶을 때는? [:,0] 사용하기

 

 

---

 

데이터 처리를 위해 배열을 자주 사용할 때 익혀두면 좋은 문법입니다.

배열을 요소로 가진 배열, 2차원 배열 또는 그 이상의 배열을 다룰 때에 사용할 수 있지요.

 

아래, Numpy 2차원 배열인 train_input을 이용하여 예시를 들어보겠습니다.

import numpy as np

# 4x3의 배열 생성
train_input = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9], [10, 11, 12]])

result = train_input[:,0]

result 부분을 확인해 주세요.

 train_input[:,0] 는 train_input의 모든 행 요소에서 첫 번째 열의 값을 선택하라는 뜻입니다.

 

여기서 : 은 전체 행이나 열을 선택하라는 의미이고, 0은 첫 번째 열을 가리킵니다.

콤마, 는 Numpy배열에서 차원을 구분할 때 사용하는데요, 첫 번째 위치는 행, 두 번째 위치는 열을 나타냅니다.

 

결과는 아래와 같습니다.

print(result)

#[1, 4, 7, 10]

 

만약  train_input의 모든 행 요소에서 두 번째 열의 값들을 선택하고 싶다면 0 대신 1을 작성하면 됩니다.

# 모든 행에서 두 번째 열을 선택
result = a[:,1]

print(result)

#[2, 5, 8, 11]

 

 

---