Python Pandas 데이터프레임에서 여러 열(컬럼) 선택하는 방법

Pandas 데이터프레임에서 여러 개의 열을 동시에 선택하는 방법을 알아보겠습니다. 이 글에서는 CSV 파일의 데이터를 불러온 뒤, 원하는 특정 컬럼들만 추출하는 과정을 예제 코드와 실행 결과를 통해 단계별로 설명합니다.

1. CSV 파일 데이터 불러오기

먼저 Microsoft Excel로 열어본 CSV 파일이 아래와 같은 자동차 판매 데이터를 담고 있다고 가정해 보겠습니다.

다음과 같이 read_csv() 함수를 사용하면 CSV 파일의 데이터를 Pandas DataFrame으로 손쉽게 불러올 수 있습니다.

dataFrame = pd.read_csv("C:\\Users\\amit_\\Desktop\\SalesData.csv")

2. 여러 열 선택하기

데이터프레임에서 여러 개의 열 레코드를 선택하려면 대괄호([])를 사용합니다. 대괄호 안에 원하는 열 이름을 쉼표로 구분하여 나열하면, 전체 데이터셋에서 해당 열들을 한 번에 가져올 수 있습니다.

dataFrame[['Reg_Price','Units']]

참고: 열 이름 목록을 감싸는 대괄호가 하나 더 필요하므로, 실제로는 이중 대괄호([[]]) 형태로 작성해야 합니다. 이렇게 하면 결과가 Series가 아닌 DataFrame 형태로 반환됩니다.

3. 전체 예제 코드

지금까지의 내용을 종합한 전체 코드는 다음과 같습니다.

import pandas as pd

# CSV 파일의 데이터를 Pandas DataFrame으로 불러오기
dataFrame = pd.read_csv("C:\\Users\\amit_\\Desktop\\SalesData.csv")
print("\nReading the CSV file...\n",dataFrame)

# 두 개의 열(Reg_Price, Units)만 선택하여 출력하기
res = dataFrame[['Reg_Price','Units']]
print("\nDisplaying two columns : \n",res)

4. 실행 결과

위 코드를 실행하면 다음과 같은 결과가 출력됩니다.

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 two columns :
    Reg_Price   Units
0        2500     100
1        3500      80
2        2500     120
3        2000      70
4        2500     110

마무리 정리

Pandas에서 여러 열을 선택할 때는 dataFrame[['열1', '열2']]처럼 이중 대괄호 안에 열 이름을 리스트로 전달하면 됩니다. 반면 대괄호를 하나만 사용해 단일 열을 선택하면 Series 객체가 반환되므로, 상황에 맞게 활용하시기 바랍니다. 이 방법은 대용량 데이터에서 필요한 컬럼만 추출해 분석할 때 매우 유용하게 사용됩니다.