home

หน้าแรก
 Computer >> บทช่วยสอนคอมพิวเตอร์ >  >> การเขียนโปรแกรม >> Python

การพลอต Power Spectral Density ใน Matplotlib

เพื่อพลอต Power Spectral Density (PSD) ใน Matplotlib

  • ตั้งค่าขนาดรูปและ padding ระหว่าง subplots
  • กำหนดตัวแปร dt
  • สร้างข้อมูล t, nse, r, cnse, s ด้วย numpy
  • สร้าง figure และ subplots
  • พลอต t และ s ด้วย plot()
  • พลอต power spectral density
  • แสดงรูปด้วย 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()

ผลลัพธ์

การพลอต Power Spectral Density ใน Matplotlib