1

分光ピークを見つけるためにいくつかのスペクトルを分析しようとしています。見つけたいピークの前後をクリックして、2 つの 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)

これは、2回目のクリック後にしたいことです: ここに画像の説明を入力

4

1 に答える 1

2

plt.drawonclick メソッドにa を追加するだけで、図を再描画する必要があります。
さらに、図を表示する前にイベントを接続する必要があります。もちろん、ここでは注釈テキストは「5」ではなく「picco」です。

試す：

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.draw()

cid = fig.canvas.mpl_connect('button_press_event', onclick)
plt.show()

視覚的にもっと魅力的かもしれません (spanselector を使用):

import matplotlib.pyplot as plt 
from matplotlib.widgets import SpanSelector
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(xmin, xmax):
    print 'Interval: {0} - {1}'.format(xmin,xmax)
    peakY=0 #max Y value
    piccoX=0 #value of the X associate to the peak
    for i in [ j for j in X if xmin<j<xmax] :
        if Diz[i]>peakY:
            peakY=Diz[i]
            piccoX=i
    print "Peak at: %f " % piccoX
    PeaksList.append([piccoX,peakY])
    ax.annotate("picco", xy=(piccoX,peakY),  xycoords='data',
            xytext=(-50, 30), textcoords='offset points',
            arrowprops=dict(arrowstyle="->"))
    plt.draw()

span = SpanSelector(ax, onclick, 'horizontal', useblit=True,
                    rectprops=dict(alpha=0.5, facecolor='blue') )
plt.show()
于 2013-10-16T10:46:52.617 に答える