pandas DataFrame 기초
pandas DataFrame 기초
DataFrame은 행과 열로 이뤄진 표 형태의 핵심 자료구조예요. 열은 인덱스(레이블)로 접근하고, 행은 조건이나 위치로 선택할 수 있어요. DataFrame의 다양한 조회 방법을 익히면 데이터 탐색이 훨씬 빨라져요.
열 선택과 추가
. 표기나 []로 열을 선택하고, 새 열을 할당할 수 있어요.
import pandas as pd
df = pd.DataFrame({"a": [1, 2, 3], "b": [4, 5, 6]})
df["c"] = df["a"] + df["b"] # 새 열 추가
print(df["a"]) # 열 선택 (Series)
조건으로 행 선택
불리언 조건으로 원하는 행만 뽑을 수 있어요.
df2 = df[df["a"] > 1]
여러 조건은 &(AND), |(OR)로 묶고, 각 조건은 괄호로 감싸요.
df3 = df[(df["a"] > 1) & (df["b"] < 10)]
loc과 iloc
엄밀한 위치 기반 선택은 loc(라벨)과 iloc(정수 위치)을 써요. loc는 행·열 이름으로, iloc는 위치 번호로 접근해요.
print(df.loc[1]) # 라벨 1 행
print(df.iloc[0, 1]) # 0행 1열 값
요약 통계 확인
describe()로 수치 컬럼의 요약 통계를 한 번에 볼 수 있어요.
print(df.describe())
head(), tail(), shape, info() 같은 메서드로 데이터 구조를 빠르게 파악할 수 있어요.