Làm cách nào để chú thích cuối dòng bằng Python và Matplotlib?

Để chú thích cuối dòng bằng Python và Matplotlib, chúng ta có thể thực hiện các bước sau -

• Đặt kích thước hình và điều chỉnh phần đệm giữa và xung quanh các ô phụ.
• Tạo ra một biến, hàng , để nhận dữ liệu về số lượng hàng.
• Nhận khung dữ liệu Pandas trong dữ liệu dạng bảng hình chữ nhật.
• Tính toán cumsum (tổng tích lũy) của khung dữ liệu.
• Vẽ đồ thị khung dữ liệu bằng cách sử dụng plot () phương pháp.
• Lặp lại dòng và tên để chú thích cuối dòng.
• Sử dụng annotate () phương thức với tên của cột, tọa độ xy, màu của các dòng, kích thước, v.v.
• Đặt chú giải vào hình vẽ.
• Để hiển thị hình này, hãy sử dụng show () phương pháp.

Ví dụ

import pandas as pd
import numpy as np
import matplotlib.pyplot as plt

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

rows = 75

df = pd.DataFrame(np.random.randint(-5, 5, size=(rows, 3)),
columns=['A', 'B', 'C'])
df = df.cumsum()
ax = df.plot()

for line, name in zip(ax.lines, df.columns):
      y = line.get_ydata()[-1]
      ax.annotate(name, xy=(1, y), xytext=(6, 0),
                  color=line.get_color(), xycoords=ax.get_yaxis_transform(),
                  textcoords="offset points", size=14, va="center")

plt.legend(loc='lower left')
plt.show()

Đầu ra