Python numpy 等效于来自 MATLAB 的 bandpower()

Question

我正在使用 Python numpy 的 ftt.ftt() 方法来生成信号的傅立叶变换。但是我想计算一个频率范围内的带宽。MATLAB 有方法 bandpower(x,fs,freqrange)，我试图专门模拟这个函数的语法。来源：https : //www.mathworks.com/help/signal/ref/bandpower.html

看起来 numpy 没有等效的功能，但是有谁知道我可以用来模拟 bandpower(x,fs,freqrange) 的代码片段？我不清楚函数的幕后到底发生了什么。

注意：如果您知道一些可以实现 Matlab 功能的非 Python 伪代码，那也会有所帮助。

Answer 1

以下用于计算频段 [fmin, fmax] 中的功率的片段对我有用：

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])