[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.
[Python] 파이썬 리스트 안에 for문 사용하는 형식, list comprehension
설명 #기본 형식 list = [ 표현식 for 반복자 in 반복 가능한 자료형] 파이썬에서 list 안에서 for문을 사용하는 경우, list comprehension 이라고 부릅니다. 반복 가능한 자료형에는 리스트 뿐만 아니라 튜플, 문자열, 딕셔너리, Set, range 객체 등이 있습니다. 예시 0부터 9까지의 숫자 중 짝수만 포함하는 리스트 생성 even_numbers = [num for num in range(10) if num % 2 == 0] # [0, 2, 4, 6, 8] 출력 print(even_numbers)
2024. 1. 11.
