[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은 첫 번째 열을 가리..
2024. 2. 13.
