welch算法计算功率谱 python

使用Python的SciPy库可以很方便地实现Welch算法来计算功率谱。

以下是一个简单的示例代码，其中x为输入信号，fs为采样率，N为每个段的长度，noverlap为相邻段之间的重叠长度，window为窗函数：

from scipy import signal
import matplotlib.pyplot as plt
import numpy as np

# 输入信号
x = # your input signal
fs = # your sampling frequency

# Welch算法计算功率谱
f, Pxx = signal.welch(x, fs, nperseg=N, noverlap=noverlap, window=window)

# 绘制功率谱
plt.semilogy(f, Pxx)
plt.xlabel('Frequency [Hz]')
plt.ylabel('PSD [V**2/Hz]')
plt.show()

其中，f为频率向量，Pxx为对应的功率谱估计值。可以根据需要进行相应的参数调整，例如窗函数类型、段长、重叠长度等。