2021-11-24 python互相關(guān)函數(shù)xcorr

def xcorr(a,b,mode ='coef'):
    fa = np.fft.fft(a,n=len(a)*2-1)
    fb = np.fft.fft(b,n=len(a)*2-1)
    xx = fa*np.conj(fb)
    xcc = np.fft.fftshift(np.fft.ifft(xx))
    aa = fa*np.conj(fa)
    xaa = np.fft.fftshift(np.fft.ifft(aa))
    bb = fb*np.conj(fb)
    xbb = np.fft.fftshift(np.fft.ifft(bb))
    if mode == 'coef':
        xcc = xcc/np.sqrt(xaa[len(a)-1]*xbb[len(b)-1])
    return np.real(xcc)

以上為政宏老哥寫的xcorr函數(shù),對(duì)應(yīng)numpy或者scipy的為

cc = np.correlate(lope1, lope, mode='full')/np.sqrt(np.correlate(lope,lope,mode='valid')[0]
                                                                *np.correlate(lope1,lope1,mode='valid')[0])

速度提升巨大,lgnb,感謝??

?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請(qǐng)聯(lián)系作者
【社區(qū)內(nèi)容提示】社區(qū)部分內(nèi)容疑似由AI輔助生成,瀏覽時(shí)請(qǐng)結(jié)合常識(shí)與多方信息審慎甄別。
平臺(tái)聲明:文章內(nèi)容(如有圖片或視頻亦包括在內(nèi))由作者上傳并發(fā)布,文章內(nèi)容僅代表作者本人觀點(diǎn),簡(jiǎn)書系信息發(fā)布平臺(tái),僅提供信息存儲(chǔ)服務(wù)。

相關(guān)閱讀更多精彩內(nèi)容

友情鏈接更多精彩內(nèi)容