Computer >> Máy Tính >  >> Lập trình >> Python

Vẽ biểu đồ mật độ phổ công suất trong Matplotlib

Để vẽ biểu đồ Mật độ phổ công suất trong Matplotlib, chúng ta có thể thực hiện các bước sau -

  • Đặt kích thước hình và điều chỉnh phần đệm giữa và xung quanh các ô phụ.
  • Khởi tạo một biến, dt .
  • Tạo t, nse, r, cnse, s, r điểm dữ liệu bằng cách sử dụng numpy
  • Tạo một hình và một tập hợp các tình huống phụ.
  • t s dữ liệu sử dụng plot () phương pháp.
  • Vẽ biểu đồ mật độ phổ công suất.
  • Để hiển thị hình này, hãy sử dụng show () phương pháp.

Ví dụ

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()

Đầu ra

Vẽ biểu đồ mật độ phổ công suất trong Matplotlib