来自MATLAB的

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