如何使用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()