我正在尝试分析一些光谱以找到光谱峰值,我编写了这个简单的代码,通过在我想要查找的峰值之前和之后单击来找到两个 X 数据之间的最大 Y 值(峰值)。这可行,我可以获得峰值的坐标,但我想自动注释找到的峰值。
import matplotlib.pyplot as plt
X=[1,2,3,4,5,6,7,8,9,10]
Y=[1,1,1,2,10,2,1,1,1,1]
fig = plt.figure()
ax = fig.add_subplot(111)
ax.plot(X,Y,label="prova") #plot the function
#plt.legend(loc=1, ncol=1, shadow=True)
plt.xlim(min(X) * 0.9, max(X) * 1.1)
plt.ylim(min(Y) * 0.9, max(Y) * 1.1)
plt.ylabel(r'Y axis')
plt.xlabel(r'X axis')
Diz=dict(zip(X,Y)) #create a dictionary that associate X with Y
Interval=[] #interval where the user search for peaks
PeaksList=[] #list of peaks found
def onclick(event):
print 'First limit at =%f'%(event.xdata)
Interval.append(event.xdata)
if len(Interval)%2==0:
a=Interval[-2]
b=Interval[-1]
if b<a: #if the user select first the highest value these statements filp it!
A=b
B=a
else:
A=a
B=b
#find the max Y value: the peak!
peakY=0 #max Y value
piccoX=0 #value of the X associate to the peak
for i in [ j for j in X if A<j<B] :
if Diz[i]>peakY:
peakY=Diz[i]
piccoX=i
print "Interval: %f - %f Peak at: %f " %(a,b,piccoX)
PeaksList.append([piccoX,peakY])
ax.annotate("picco", xy=(piccoX,peakY), xycoords='data',
xytext=(-50, 30), textcoords='offset points',
arrowprops=dict(arrowstyle="->")
)
plt.show()
cid = fig.canvas.mpl_connect('button_press_event', onclick)
这是我第二次点击后想要的: