最近在幫朋友做EMD變換 其中需要尋找變換波形的極大值 因?yàn)榫W(wǎng)上大多是尋找最大值，找不到合適的，所以自己寫(xiě)了段python代碼，本人學(xué)生小白，初學(xué)者，不足之處請(qǐng)多多指教。

原理就是 尋找二階導(dǎo)數(shù)小于0的點(diǎn)就是極大值，轉(zhuǎn)換到離散的中就是二階差分小于0的點(diǎn)（其中一階差分需要大于零）

返回的是給定數(shù)組極大值的下標(biāo)。

時(shí)間倉(cāng)促也沒(méi)有做優(yōu)化，寫(xiě)的也很粗糙。

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#返回極大值的數(shù)組下標(biāo)

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

print(find_max(X))

測(cè)試之后為4 7 10

總結(jié)

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

如果覺(jué)得生活随笔網(wǎng)站內(nèi)容還不錯(cuò)，歡迎將生活随笔推薦給好友。