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

Python Pandas – DataFrame의 여러 열을 하나의 그래프로 시각화하는 방법

DataFrame에서 여러 개의 열을 동시에 시각화할 때는 막대그래프(Bar Graph)를 활용하는 것이 효과적입니다. Pandas의 plot() 메서드를 호출한 뒤, kind 매개변수를 'bar'로 지정하면 손쉽게 막대그래프를 그릴 수 있습니다. 먼저 필요한 라이브러리를 임포트합니다.

import pandas as pd
import matplotlib.pyplot as mp

다음은 각 팀의 기록(Team Records)을 담고 있는 데이터입니다.

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

이 데이터를 Pandas DataFrame으로 변환하고 열 이름을 지정해 줍니다.

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

이제 여러 열을 하나의 막대그래프에 함께 표시합니다. 핵심은 y 매개변수에 원하는 열 이름들을 리스트 형태로 전달하는 것입니다. x축에는 "Team" 열을, y축에는 "Rank_Points"와 "Year" 두 개의 열을 지정했습니다.

dataFrame.plot(x="Team", y=["Rank_Points","Year" ], kind="bar", 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.plot(x="Team", y=["Rank_Points","Year" ], kind="bar", figsize=(10, 9))

# 그래프 화면에 출력
mp.show()

실행 결과

위 코드를 실행하면 다음과 같은 막대그래프가 출력됩니다. 각 팀별로 랭킹 점수(Rank_Points)와 연도(Year)가 나란히 표시되어, 여러 열의 값을 한눈에 비교할 수 있습니다.

Python Pandas – DataFrame의 여러 열을 하나의 그래프로 시각화하는 방법