Python中FFT峰值下的面积

import numpy as np
from scipy.fft import fft

signal = [1, 2, 3, 4, 5, 4, 3, 2, 1]  # 示例输入信号数据

fft_result = fft(signal)

amplitude_spectrum = np.abs(fft_result)

peak_index = np.argmax(amplitude_spectrum)

sampling_rate = 1  # 采样率，根据实际情况设置
frequency_range = np.fft.fftfreq(len(signal), d=1/sampling_rate)
peak_frequency = frequency_range[peak_index]

area_under_peak = np.sum(amplitude_spectrum[(frequency_range >= peak_frequency - delta) & (frequency_range <= peak_frequency + delta)])