如何在 Pyplot 中自动添加注释，表示最大值？

要为 Pyplot 中的最大值添加注释，我们可以执行以下步骤 -

• 设置图形大小并调整子图之间和周围的填充。
• 创建一个新图形或激活一个现有的图形。
• 创建一个列表，其中包含 x 和 y 数据点。
• 使用 numpy 绘制 x 和 y 数据点。
• 找出 Y 数组中的最大值及数组中对应于该 最大值 元素的位置
• 使用局部最大值对该点添加注释。
• 要显示图形，请使用 show() 方法。

示例

import numpy as np
import matplotlib.pyplot as plt

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

fig = plt.figure()
ax = fig.add_subplot(111)

x = np.array([1, 3, 5, 3, 1])
y = np.array([2, 1, 3, 1, 2])
line, = ax.plot(x, y)

ymax = max(y)
xpos = np.where(y == ymax)
xmax = x[xpos]

ax.annotate('local max', xy=(xmax, ymax), xytext=(xmax, ymax + 5), arrowprops=dict(facecolor='black'),)

plt.show()

输出

Rishikesh Kumar Rishi

更新时间： 2021 年 7 月 9 日

7K+ 查看次数

开启你的 职业旅程

完成课程，获得认证

开始