在 Matplotlib 中绘制功率谱密度
要在 Matplotlib 中绘制功率谱密度,我们可以采取以下步骤 -
- 设置图形大小并调整子图之间和周围的填充。
- 初始化变量dt。
- 使用 numpy 创建t、nse、r、cnse、s和r数据点
- 创建图形和一组子图。
- 使用plot()方法绘制t和s数据。
- 绘制功率谱密度。
- 要显示图形,请使用show()方法。
示例
import matplotlib.pyplot as plt import numpy as np plt.rcParams["figure.figsize"] = [7.50, 3.50] plt.rcParams["figure.autolayout"] = True dt = 0.01 t = np.arange(0, 10, dt) nse = np.random.randn(len(t)) r = np.exp(-t / 0.05) cnse = np.convolve(nse, r) * dt cnse = cnse[:len(t)] s = 0.1 * np.sin(2 * np.pi * t) + cnse fig, (ax0, ax1) = plt.subplots(2, 1) ax0.plot(t, s) ax1.psd(s, 512, 1 / dt) plt.show()
输出
广告