python中numpy函数ftt_相当于MATLAB中bandpower（）的Python numpy

最新推荐文章于 2022-08-01 13:59:00 发布

法山叔

最新推荐文章于 2022-08-01 13:59:00 发布

阅读量760

点赞数 1

文章标签： python中numpy函数ftt

本文链接：https://blog.csdn.net/weixin_34972620/article/details/111956109

版权

I am using Python numpy's ftt.ftt() method to generate the fourier transform of a signal. However I want to calculate the bandpower over a range of frequencies. MATLAB has the method bandpower(x,fs,freqrange), I am trying to simulate specifically this syntax of the function. Source: https://www.mathworks.com/help/signal/ref/bandpower.html

It doesn't look like numpy has an equivalent function, but does anyone know a code snippet I can use to mimic bandpower(x,fs,freqrange)? It's not clear to me what exactly is going on behind the scenes in the function.

Note: If you know some non-Python pseudocode that would achieve the Matlab function, that would also be helpful.

解决方案

The following snippet for computing the power in the band [fmin, fmax] worked for me:

import scipy

def bandpower(x, fs, fmin, fmax):

f, Pxx = scipy.signal.periodogram(x, fs=fs)

ind_min = scipy.argmax(f > fmin) - 1

ind_max = scipy.argmax(f > fmax) - 1

return scipy.trapz(Pxx[ind_min: ind_max], f[ind_min: ind_max])