[Pandas] 데이터 추출(열, Column)

한 열 단위 데이터 추출

Series : 1개 열
DataFrame : 2개 열 이상

열단위 Data 얻기

import pandas as pd

df = pd.read_csv('input.csv',sep=',')

column_df = df['country']

print(type(colum_df)) # Series

print(column_df.head()) # 5줄

print(column_df.tail()) # 하위 5줄

2열 이상 데이터 추출

columns_df = df[['country', 'continent', 'year']] # List 전달

print(type(columns_df)) # DataFrame

print(columns_df.head())

print(columns_df.tail())