pandas DataFrame 그리기
df.plot으로 pandas DataFrame을 차트로 만들고 Matplotlib으로 꾸밉니다.
소개
pandas에는 Matplotlib이 내장되어 있습니다. df.plot은 DataFrame에서 바로 차트를 그리고 열 이름을 범례로 씁니다. Matplotlib의 Axes를 돌려주므로 제목, 레이블, 스타일에 대해 배운 내용을 그대로 쓸 수 있습니다.
x="month"는 월을 x축에 놓고, y=[...]는 그릴 열을 고르며, kind는 차트 종류("line", "bar", "barh", "area" 등)를 고릅니다. df.plot은 평소의 Matplotlib Axes인 ax를 돌려주므로 ax.set_ylabel과 ax.set_title을 그대로 쓸 수 있습니다. rot=0은 월 이름을 가로로 유지하고, stacked=True는 막대를 쌓습니다.
인터랙티브 시각화
kind를 바꾸고, 열을 고르고, 쌓아 보세요. pandas는 Matplotlib으로 그리고 계속 꾸밀 수 있는 Axes를 돌려줍니다.
실행 가능한 Matplotlib 예제
Matplotlib
main.py
import matplotlib.pyplot as plt
import pandas as pd
df = pd.DataFrame({
"month": ["Jan", "Feb", "Mar", "Apr", "May", "Jun"],
"coffee": [120, 110, 130, 150, 170, 160],
"tea": [80, 85, 90, 70, 60, 65],
})
ax = df.plot(x="month", y=["coffee", "tea"], kind="bar", rot=0, figsize=(7, 4))
ax.set_ylabel("Cups sold")
ax.set_title("Drinks sold per month")
plt.show()
