열 선택

데이터 프레임에서 열 이름을 확인하고 필요한 열 선택.

실습 준비

import pandas as pd
sales = pd.read_excel("sales.xlsx")

sales.xlsx를 판다스로 읽고 열 이름을 확인하는 파이썬 코드를 작성해줘.

열 이름

sales.columns: 데이터 프레임의 열 이름 확인.

sales.columns  # 열 이름

Index(['product', 'region', 'qty', 'unit_price'], dtype='object')

product(제품), region(지역), qty(수량), unit_price(단가)의 네 열로 구성.

열 하나 선택

sales의 qty 열을 선택해줘.

sales["열 이름"]: 열 하나를 시리즈(Series)로 선택. 다음은 qty 열 선택.

sales["qty"]

0     3
1     5
2     8
3    10
4     4
5     6
Name: qty, dtype: int64

시리즈는 하나의 열을 담는 1차원 자료구조. 행 인덱스와 선택한 열의 값 표시.

속성으로 열 선택

열 이름이 qty처럼 변수 이름으로 사용할 수 있는 형태이면 점을 붙여 sales.qty로도 선택 가능.

sales.qty

0     3
1     5
2     8
3    10
4     4
5     6
Name: qty, dtype: int64

sales["qty"]와 같은 시리즈 반환.

여러 열 선택

sales의 product와 qty 두 열을 함께 선택해줘.

sales[["열 이름1", "열 이름2"]]: 열 이름을 리스트로 묶어 데이터 프레임으로 선택.

sales[["product", "qty"]]

  product  qty
0     노트북    3
1     모니터    5
2     키보드    8
3     마우스   10
4      웹캠    4
5     헤드셋    6

선택한 열은 리스트에 지정한 순서로 표시. 이중 대괄호로 열 하나를 선택해도 데이터 프레임 유지.

sales[["qty"]]

   qty
0    3
1    5
2    8
3   10
4    4
5    6