[ Pandas ] Dataframe 필요한 열(column) 추출
import pandas as pd data = { 'A':[1,2,3,4,5], 'B':['a','b','c','d','e'], 'C':['apple','banana','grape','strawberry','peach'] } df = pd.DataFrame(data) 1. 컬럼명 사용 df[['A', 'B']] 2. iloc 사용 # 0번째, 2번째 컬럼의 모든행 df.iloc[:, [0,2]] 3. drop 사용 : drop은 컬럼 제거할 때 사용하는 방법이지만 추출할때도 사용 가능. (원하는 컬럼 제외하고 나머지를 제거) # 'A', 'C' 추출 df.drop(['B', 'D'], axis=1) axis = 0 : 행을 제거 / axis = 1 : 열을 제거
2023. 5. 7.
