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

Python - Pandas DataFrame을 선 그래프(Line Graph)로 시각화하는 방법

Pandas DataFrame 데이터를 선 그래프로 시각화하려면 plot() 메서드를 사용하고 kind 매개변수를 'line'으로 설정하면 됩니다. 이 방법은 시간의 흐름에 따른 데이터 변화나 추세를 한눈에 파악할 때 특히 유용합니다.

먼저 필요한 라이브러리를 임포트해 보겠습니다.

import pandas as pd
import matplotlib.pyplot as mp

1. 샘플 데이터 준비하기

이번 예제에서는 각 팀의 기록(Team Records)을 담은 데이터를 사용하겠습니다.

data = [["Australia", 2500, 2021],["Bangladesh", 1000, 2021],["England", 2000, 2021],["India", 3000, 2021],["Srilanka", 1500, 2021]]

준비한 데이터를 Pandas DataFrame으로 변환하고, 각 열(column)의 이름을 지정합니다.

dataFrame = pd.DataFrame(data, columns=["Team","Rank_Points", "Year"])

2. 선 그래프로 그리기

DataFrame을 선 그래프로 출력하려면 plot() 메서드에 x축과 y축에 사용할 열을 지정하고, kind 매개변수를 "line"으로 설정합니다. figsize 옵션을 사용하면 그래프의 크기도 조절할 수 있습니다.

dataFrame.plot(x="Team", y=["Rank_Points","Year" ], kind="line", figsize=(10, 9))

전체 예제 코드

지금까지 설명한 내용을 하나로 합친 전체 코드는 다음과 같습니다.

import pandas as pd
import matplotlib.pyplot as mp

# 샘플 데이터
data = [["Australia", 2500, 2021],["Bangladesh", 1000, 2021],["England", 2000, 2021],["India", 3000, 2021],["Srilanka", 1500, 2021]]

# DataFrame 생성
dataFrame = pd.DataFrame(data, columns=["Team","Rank_Points", "Year"])

# DataFrame을 선 그래프로 그리기
dataFrame.plot(x="Team", y=["Rank_Points","Year" ], kind="line", figsize=(10, 9))

# 선 그래프 화면에 표시
mp.show()

실행 결과

위 코드를 실행하면 x축에는 팀 이름(Team), y축에는 랭킹 포인트(Rank_Points)와 연도(Year)가 표시된 선 그래프가 출력됩니다. 각 열이 서로 다른 색상의 선으로 구분되어 나타나므로, 여러 데이터 계열을 동시에 비교하기에도 좋습니다.

Python - Pandas DataFrame을 선 그래프(Line Graph)로 시각화하는 방법