Python - pandas 04. 선택과 인덱싱

import pandas as pd

df["age"]              # 열 1개 → Series
df[["name", "age"]]    # 열 여러 개 → DataFrame
df[0:3]                # 행 슬라이스 (위치) → 3행
df[df["age"] > 30]     # 불리언 인덱싱 → 행 필터

df.loc[3]                      # 인덱스 라벨이 3인 행
df.loc[3, "amount"]            # 행 라벨 3, 열 "amount"
df.loc[2:5, ["name", "age"]]   # 라벨 2~5 (5 포함!), 두 열
df.loc[df["age"] > 30, "city"] # 불리언 + 열 지정
df.loc[:, "a":"c"]             # 열 라벨 a~c (c 포함)

df.iloc[0]                # 첫 행
df.iloc[0, 2]             # 0번 행, 2번 열
df.iloc[2:5, [0, 1]]      # 위치 2,3,4 (5 제외), 0·1번 열
df.iloc[-1]               # 마지막 행
df.iloc[::2]              # 짝수 위치 행

df.at[3, "amount"]     # 라벨로 값 1개 (가장 빠름)
df.iat[0, 2]           # 위치로 값 1개

df[df["age"] > 30]
df[(df["age"] > 30) & (df["city"] == "Seoul")]   # & | ~, 각 조건 괄호 필수
df[df["city"].isin(["Seoul", "Busan"])]
df[df["amount"].between(100, 500)]
df[df["name"].str.startswith("K")]
df[~df["category"].isna()]

df.query("age > 30 and city == 'Seoul'")
df.query("amount.between(100, 500)")
threshold = 1000
df.query("amount > @threshold")     # @변수명으로 파이썬 변수 참조

# 나쁨 — SettingWithCopyWarning, 반영 보장 안 됨
df[df["age"] > 30]["bonus"] = 100
 
# 좋음 — .loc 한 번에 행+열 지정
df.loc[df["age"] > 30, "bonus"] = 100

seoul = df.loc[df["city"] == "Seoul"].copy()

g = df.set_index(["city", "category"]).sort_index()
 
g.loc["Seoul"]                    # 1레벨 선택
g.loc[("Seoul", "Food")]          # 튜플로 두 레벨 동시 지정
g.loc[(slice(None), "Food"), :]   # 2레벨만 지정 (= g.xs("Food", level="category"))
g.xs("Food", level="category")