python - 相当于MATLAB中bandpower()的Python numpy

我正在使用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伪代码，那也将有所帮助。

最佳答案

以下用于计算[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])