Computer >> Hướng Dẫn Máy Tính >  >> Lập Trình >> Python

Vẽ nhiều đường biểu đồ (multiple line graphs) với Pandas và Matplotlib

Để vẽ nhiều đường biểu đồ dùng Pandas và Matplotlib:

Các bước thực hiện

  • Thiết lập kích thước hình và khoảng đệm.
  • Tạo DataFrame 2D với cột x, y, equation.
  • Reshape DataFrame bằng pivot theo index x, columns equation, values y.
  • Dùng plot() vẽ các đường.
  • Hiển thị bằng show().

Ví dụ

import pandas as pd
from matplotlib import pyplot as plt

plt.rcParams["figure.figsize"] = [7.50, 3.50]
plt.rcParams["figure.autolayout"] = True

df = pd.DataFrame([
    ["y=x^3", 0, 0], ["y=x^3", 1, 1], ["y=x^3", 2, 8], ["y=x^3", 3, 27], ["y=x^3", 4, 64],
    ["y=x^2", 0, 0], ["y=x^2", 1, 1], ["y=x^2", 2, 4], ["y=x^2", 3, 9], ["y=x^2", 4, 16],
    ["y=mx", 0, 0], ["y=mx", 1, 1], ["y=mx", 2, 2], ["y=mx", 3, 3], ["y=mx", 4, 3],
], columns=['equation', 'x', 'y'])

df = df.pivot(index='x', columns='equation', values='y')
df.plot()
plt.show()

Kết quả

Vẽ nhiều đường biểu đồ (multiple line graphs) với Pandas và Matplotlib