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