如何在 matplotlib 中的曲线末端放置箭头？

Question

我想要曲线旁边的箭头。例如：

import numpy as np
import matplotlib.pyplot as plt

X = np.linspace(0,4*np.pi,10000)
Y = np.sin(X)

shift = 0.1
seg_size = 300

i = 0
plt.plot(X,Y,color='blue')
while i +seg_size < len(X):
    x = X[i:i+seg_size]
    y = Y[i:i+seg_size]+shift
    plt.plot(x,y,color='black')
    #input here command for arrow head
    i += seg_size*2

plt.show()

我试图计算曲线末端线的下一个角度并绘制箭头线，但我做错了，箭头变形了。任何提示？

Answer 1

怎么样使用

numpy.annotate() 

功能？

请参阅此处 pyplot.annotate() 函数创建的箭头示例：

https://matplotlib.org/examples/pylab_examples/annotation_demo.html