如何使用 Python 和 Matplotlib 注释行尾?
matplotlibpythondata visualization
要使用 Python 和 Matplotlib 注释行尾,我们可以采取以下步骤 −
- 设置图形大小并调整子图之间和周围的填充。
- 初始化变量 rows,以获取行数数据。
- 在矩形表格数据中获取 Pandas 数据框。
- 计算数据框的 cumsum(累计总和)。
- 使用 plot() 方法绘制数据框。
- 迭代 line 和 name 来注释行尾。
- 使用 annotate() 方法列的名称、xy 坐标、线条颜色、大小等。
- 在图形上放置图例。
- 要显示图形,请使用 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()