파이썬 Pandas DataFrame에서 열(컬럼) 선택하는 방법

Pandas DataFrame에서 열 선택하기

Pandas DataFrame에서 특정 열(Column)을 선택하는 것은 데이터 분석 작업에서 가장 기본적이면서도 자주 사용되는 기능입니다. 가장 간단한 방법은 대괄호([])를 사용하여 원하는 열 이름을 지정하는 것입니다.

기본 문법은 다음과 같습니다:

dataFrame['ColumnName']

대괄호 안에 선택하고자 하는 열의 이름만 넣어주면 해당 열이 Series 형태로 반환됩니다.

1단계: 라이브러리 임포트

먼저 pandas 라이브러리를 임포트합니다:

import pandas as pd

2단계: DataFrame 생성

예제로 사용할 DataFrame을 만듭니다. 이 DataFrame에는 'Car'와 'Units' 두 개의 열이 포함되어 있습니다:

dataFrame = pd.DataFrame(
   {
      "Car": ['BMW', 'Lexus', 'Audi', 'Mustang', 'Bentley', 'Jaguar'],
      "Units": [100, 150, 110, 80, 110, 90]
   }
)

3단계: 단일 열 선택

하나의 열만 선택하려면 아래와 같이 대괄호 안에 열 이름을 지정합니다. 여기서는 'Car' 열을 선택합니다:

dataFrame['Car']

전체 예제 코드

다음은 위 과정을 모두 포함한 완성된 코드입니다:

import pandas as pd

# DataFrame 생성
dataFrame = pd.DataFrame(
   {
      "Car": ['BMW', 'Lexus', 'Audi', 'Mustang', 'Bentley', 'Jaguar'],
      "Units": [100, 150, 110, 80, 110, 90]
   }
)

print("DataFrame ...\n", dataFrame)

# 단일 열 선택 및 출력
print("\n선택한 단일 열 출력 = \n", dataFrame['Car'])

실행 결과

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

DataFrame ...
        Car   Units
0      BMW     100
1    Lexus     150
2     Audi     110
3  Mustang      80
4  Bentley     110
5   Jaguar      90

선택한 단일 열 출력 =
0      BMW
1    Lexus
2     Audi
3  Mustang
4  Bentley
5   Jaguar
Name: Car, dtype: object

출력 결과를 보면 선택한 열이 인덱스와 함께 Series 객체로 반환되며, 하단에 열 이름(Name)과 데이터 타입(dtype)이 함께 표시됩니다.

추가 팁: 여러 열 한 번에 선택하기

여러 개의 열을 동시에 선택하고 싶다면 대괄호 안에 열 이름들을 리스트 형태로 전달하면 됩니다. 이 경우 결과는 DataFrame으로 반환됩니다:

# 여러 열 선택
dataFrame[['Car', 'Units']]

또한 .loc[](라벨 기반 인덱싱)과 .iloc[](정수 위치 기반 인덱싱)을 활용하면 행과 열을 더욱 유연하게 조합하여 선택할 수 있으므로, 실무에서 데이터를 다룰 때 함께 익혀두면 매우 유용합니다.