Vẽ biểu đồ từ DataFrame pandas
Biến một DataFrame pandas thành biểu đồ bằng df.plot và tùy chỉnh bằng Matplotlib.
Giới thiệu
pandas tích hợp sẵn Matplotlib: df.plot vẽ biểu đồ trực tiếp từ DataFrame và dùng tên cột làm chú giải. Hàm này trả về một Axes của Matplotlib, nên mọi thứ bạn đã học về tiêu đề, nhãn và kiểu dáng vẫn dùng được.
x="month" đặt các tháng lên trục x và y=[...] chọn các cột cần vẽ; kind chọn loại biểu đồ ("line", "bar", "barh", "area"...). df.plot trả về ax, một Axes bình thường của Matplotlib, nên ax.set_ylabel và ax.set_title dùng như trước. rot=0 giữ tên tháng nằm ngang, còn stacked=True sẽ xếp chồng các cột.
Trình mô phỏng tương tác
Đổi kind, chọn cột dữ liệu và xếp chồng chúng. pandas vẽ bằng Matplotlib và trả về một Axes để bạn tiếp tục tùy chỉnh.
Ví dụ Matplotlib chạy được
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()
