Python Pandas로 CSV 파일에서 특정 열만 추출하는 방법

Python에서 CSV 파일의 특정 열만 골라 리스트 형태로 추출하고 싶다면, Pandas 라이브러리의 read_csv() 메서드를 활용하면 됩니다. 이 메서드는 usecols 파라미터와 함께 사용하면 원하는 열만 손쉽게 불러올 수 있어 매우 편리합니다.

처리 순서

• 추출하고자 하는 열 이름들의 목록(list)을 먼저 정의합니다.

• read_csv() 메서드에 usecols 옵션을 지정하여 CSV 파일을 데이터프레임(DataFrame)으로 불러옵니다.

• 추출된 데이터를 출력하여 확인합니다.

• plot() 메서드를 사용해 데이터프레임을 그래프로 시각화합니다.

• show() 메서드로 완성된 그래프를 화면에 표시합니다.

예제 코드

import pandas as pd
from matplotlib import pyplot as plt
plt.rcParams["figure.figsize"] = [7.00, 3.50]
plt.rcParams["figure.autolayout"] = True
columns = ["Name", "Marks"]
df = pd.read_csv("input.csv", usecols=columns)
print("CSV 파일 내용:\n", df)
plt.plot(df.Name, df.Marks)
plt.show()

입력 데이터 (input.csv)

위 예제에서 사용되는 CSV 파일에는 다음과 같은 데이터가 들어 있습니다.

실행 결과

코드를 실행하면 CSV 파일에서 'Name'과 'Marks' 두 개의 열만 추출되어 콘솔에 데이터프레임이 출력되고, 동시에 이름별 점수를 나타내는 다음과 같은 꺾은선 그래프가 화면에 표시됩니다.

핵심 포인트 정리

• usecols 파라미터에 열 이름 리스트를 전달하면 해당 열만 메모리에 로드되므로, 대용량 CSV 파일 처리 시 성능 면에서도 유리합니다.

• 열 이름 대신 열 인덱스 번호(0부터 시작)를 사용할 수도 있습니다. 예: usecols=[0, 1]

• matplotlib의 rcParams 설정을 통해 그래프 크기와 레이아웃을 자유롭게 조절할 수 있습니다.