Pandas DataFrame은 방대한 데이터를 다룰 때 그중 필요한 부분만 추출해 분석하는 경우가 매우 많습니다. 이번 글에서는 CSV 파일로부터 데이터를 불러온 뒤, 대괄호([])를 사용하여 원하는 열(column)만 골라내는 방법을 예제와 함께 살펴보겠습니다.
1. 샘플 데이터 준비
설명을 위해 Microsoft Excel로 열어본 다음과 같은 CSV 파일(SalesData.csv)이 있다고 가정합니다.
2. CSV 파일을 DataFrame으로 불러오기
가장 먼저 pd.read_csv() 함수를 사용해 CSV 파일의 데이터를 Pandas DataFrame으로 로드합니다.
dataFrame = pd.read_csv("C:\\Users\\amit_\\Desktop\\SalesData.csv")3. 대괄호를 이용한 하위 집합 선택
DataFrame에서 특정 열을 선택할 때는 대괄호 안에 열 이름을 지정하면 됩니다.
단일 열 선택
대괄호 안에 하나의 열 이름을 넣으면 전체 데이터셋에서 해당 열 하나만 가져올 수 있습니다.
dataFrame['Car']
여러 열 선택
두 개 이상의 열을 선택할 때는 열 이름들을 리스트 형태로 감싸서 전달합니다. 즉, 대괄호 안에 리스트를 한 번 더 작성하는 이중 대괄호 구조를 사용합니다.
dataFrame[['Car', 'Units']]
전체 예제 코드
지금까지 설명한 내용을 하나의 코드로 정리하면 다음과 같습니다.
import pandas as pd
# CSV 파일의 데이터를 Pandas DataFrame으로 불러오기
dataFrame = pd.read_csv("C:\\Users\\amit_\\Desktop\\SalesData.csv")
print("\nCSV 파일 읽기...\n", dataFrame)
# 단일 열('Car')만 표시
res1 = dataFrame['Car']
print("\n'Car' 열 하나만 표시 : \n", res1)
# 두 개의 열('Car', 'Units') 표시
res2 = dataFrame[['Car', 'Units']]
print("\n두 개의 열 표시 : \n", res2)실행 결과
위 코드를 실행하면 다음과 같은 출력 결과를 확인할 수 있습니다.
Reading the CSV file...
Car Reg_Price Units
0 BMW 2500 100
1 Lexus 3500 80
2 Audi 2500 120
3 Jaguar 2000 70
4 Mustang 2500 110
Displaying only one column Car :
0 BMW
1 Lexus
2 Audi
3 Jaguar
4 Mustang
Name: Car, dtype: object
Displaying two columns :
Car Units
0 BMW 100
1 Lexus 80
2 Audi 120
3 Jaguar 70
4 Mustang 110정리
- 단일 열 선택:
df['열이름']— Series 형태로 반환됩니다. - 복수 열 선택:
df[['열1', '열2']]— 리스트로 감싸면 DataFrame 형태로 반환됩니다.
이처럼 대괄호 인덱싱만 익혀도 DataFrame에서 원하는 데이터를 손쉽게 추출할 수 있습니다. 더 복잡한 조건의 행·열 선택이 필요하다면 loc, iloc, filter() 등의 메서드도 함께 활용해 보시기 바랍니다.