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

Pandas 시리즈 데이터로 Matplotlib 막대 그래프 그리는 방법

Pandas 시리즈 데이터를 Matplotlib에서 막대 그래프(bar graph)로 시각화하려면 다음 단계를 따르면 됩니다.

  • 1부터 10 사이의 값들을 가지는 여러 키로 구성된 딕셔너리를 생성합니다.

  • Pandas의 DataFrame을 사용하여 데이터프레임을 만듭니다.

  • plot() 메서드에 kind="bar" 옵션을 지정하여 막대 그래프를 그립니다.

  • show() 메서드를 호출하여 그래프를 화면에 표시합니다.

예제 코드

import pandas as pd
from matplotlib import pyplot as plt
plt.rcParams["figure.figsize"] = [7.00, 3.50]
plt.rcParams["figure.autolayout"] = True
d = {'y=1/x': [1 / i for i in range(1, 10)],
    'y=x': [i for i in range(1, 10)],
    'y=x^2': [i * i for i in range(1, 10)],
    'y=x^3': [i * i * i for i in range(1, 10)]}
df = pd.DataFrame(d)
df.plot(kind='bar')
plt.show()

코드 설명

위 예제에서는 y=1/x, y=x, y=x², y=x³ 네 가지 수열을 각각 리스트 컴프리헨션으로 계산하여 하나의 딕셔너리에 담습니다. 이 딕셔너리를 pd.DataFrame()으로 변환하면 각 키가 열(column)이 되어 하나의 데이터프레임이 구성됩니다.

이후 df.plot(kind='bar')를 호출하면 네 개의 수열이 한 그래프에서 막대 형태로 나란히 비교되어 표시됩니다. figure.figsize 설정으로 그래프 크기를 지정하고, figure.autolayout을 활성화하면 레이블이나 제목이 잘리지 않도록 레이아웃이 자동으로 조절됩니다.

출력 결과

Pandas 시리즈 데이터로 Matplotlib 막대 그래프 그리는 방법