Computer >> 컴퓨터 >  >> 프로그래밍 >> Python

Matplotlib으로 CSV 파일에서 여러 줄 플롯 그리기

데이터 분석에서 CSV 파일에 저장된 여러 컬럼의 데이터를 한 번에 시각화해야 하는 경우가 자주 있습니다. Matplotlib과 Pandas를 함께 사용하면 CSV 파일의 데이터를 손쉽게 읽어와 여러 개의 선을 가진 플롯(multiline plot)을 만들 수 있습니다. 아래 단계를 따라 해 보세요.

구현 단계

  • 그림 크기(figure size)를 설정하고 서브플롯 주변 및 사이의 여백(padding)을 조정합니다.
  • CSV 파일에서 가져올 컬럼 이름 목록을 생성합니다. 이때 컬럼 이름은 CSV 파일 내 실제 컬럼명과 정확히 일치해야 합니다.
  • Pandas의 read_csv() 메서드를 사용해 CSV 파일에서 데이터를 읽어옵니다.
  • df.plot() 메서드를 호출하여 여러 개의 선을 한 번에 그립니다.
  • plt.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 = ['mpg', 'displ', 'hp', 'weight']

# CSV 파일 읽기
df = pd.read_csv("auto-mpg.csv", usecols=columns)

# 여러 줄 플롯 그리기
df.plot()

plt.show()

실행 결과

위 코드를 실행하면 'mpg', 'displ', 'hp', 'weight' 네 개의 컬럼 데이터가 각각 다른 색상의 선으로 표현된 하나의 그래프가 출력됩니다.

Matplotlib으로 CSV 파일에서 여러 줄 플롯 그리기Matplotlib으로 CSV 파일에서 여러 줄 플롯 그리기

추가 팁

usecols 매개변수를 사용하면 대용량 CSV 파일에서 필요한 컬럼만 선택적으로 로드할 수 있어 메모리 사용량을 줄이고 처리 속도를 높일 수 있습니다. 또한 df.plot()에는 linewidth, style, title 등 다양한 매개변수를 추가해 그래프의 선 두께, 스타일, 제목을 자유롭게 꾸밀 수 있습니다.