在 Matplotlib 中绘制功率谱密度
matplotlibpythondata visualization
要在 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()
输出
