python - matplotlib 中的信号特定点

Question

有一个如下所示的基本图表:

import matplotlib.pyplot as plt
import numpy as np

y = np.array([1,2,3,4,5,6])
f, ax = plt.subplots()
ax.plot(y)
plt.show()

...有没有办法创建一个小箭头，比如说，进入观测点“3”和“5”？我有这个需求，因为我处理金融系列，我想用小箭头突出显示某些事件发生的点(根据 bool 发生真/假列表)。因此，在我上面询问的情况下，出现次数列表将是 [False, False, True, False, True, False] 但我想为此将向量乘以原始值就足够了数组，以便我可以将标记置于正确的高度。 Here是一个更复杂的网络示例来给出视觉想法。提前致谢

Answer 1

在 matplotlib 中，您可以使用文本注释 annotate your graph suitably 。下面的代码显示了一个简单的示例，其中我注释了线性图中的第三点。显然，如果您要在自己的代码中执行此操作，您可以提供一种更智能的方式来选择箭头点和文本的 x,y 坐标。

import matplotlib.pyplot as plt
import numpy as np

y = np.array([1,2,3,4,5,6])

f, ax = plt.subplots()

ax.plot(y, linestyle='None', marker='o')


for i in [2, 3, 4]:
    ax.annotate('', xy=(i-1, i), xytext=(i-1,i+0.5),
                arrowprops=dict(facecolor='black',
                                width=1,
                                shrink=0.1))

plt.margins(0.1)
plt.show()

编辑我已修改代码以将文本注释删除为空字符串。