最近在幫朋友做EMD變換 其中需要尋找變換波形的極大值 因為網上大多是尋找最大值，找不到合適的，所以自己寫了段python代碼，本人學生小白，初學者，不足之處請多多指教。

原理就是 尋找二階導數小于0的點就是極大值，轉換到離散的中就是二階差分小于0的點（其中一階差分需要大于零）

返回的是給定數組極大值的下標。

時間倉促也沒有做優化，寫的也很粗糙。

import numpy as np


def find_max(temp):
? ? i = 0
? ? diff1_num = np.zeros(np.size(temp)-1)
? ? while i<np.size(temp)-1 :?
? ? ? ? diff1_num[i] = temp[i+1] - temp[i] ??
? ? ? ? if diff1_num[i] > 0:
? ? ? ? ? ? diff1_num[i] = 1?
? ? ? ? else:
? ? ? ? ? ? diff1_num[i] = 0?
? ? ? ? i = i + 1?
? ??
? ? j = 0?
? ? diff2_num = np.zeros(np.size(diff1_num)-1)
? ? count_1 =[]
? ? k = 0
? ? while j<np.size(diff1_num)-1:
? ? ? ? diff2_num[j] = diff1_num[j+1] - diff1_num[j]?
? ? ? ? if diff2_num[j] < 0:
? ? ? ? ? ? count_1.append(j) ?#n
? ? ? ? ? ? k = k + 1?
? ? ? ? j = j + 1
? ? count_1 = np.array(count_1)
? ? K = 0
? ? u = np.zeros(np.size(count_1))
? ? un = 0
? ? count_1 = count_1.astype(np.int)
? ? while K<np.size(count_1):
? ? ? ? if temp[count_1[K]+1] > temp[count_1[K]] :
? ? ? ? ? ? u[un]=K
? ? ? ? ? ? un = un + 1?
? ? ? ? K = K + 1

? ? return count_1 + 1#返回極大值的數組下標

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

print(find_max(X))

測試之后為4 7 10

總結

以上是生活随笔為你收集整理的寻找波形中的极大值的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。