如何使用 Python 和 matplotlib 标注行末尾？

要使用 Python 和 Matplotlib 标注行末尾，我们可以采取以下步骤 −

• 设置图形大小并调整子图之间以及周围的填充。
• 初始化变量，rows，以获取数据行数。
• 在矩形表格数据中获取 Pandas 数据框。
• 计算数据框的 cumsum（累积和）。
• 使用 plot() 方法绘制数据框。
• 迭代 line 和 name 以标注行末尾。
• 对列的名称、xy 坐标、线条颜色、大小等使用 annotate() 方法。
• 在图形上放置一个图例。
• 要显示图形，请使用 show() 方法。

示例

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()

输出

Rishikesh Kumar Rishi

更新时间：2021 年 6 月 16 日

1K+ 浏览量

开启您的 职业生涯

通过完成课程获得认证

立即开始