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

Làm thế nào để vẽ quang phổ cường độ trong Matplotlib bằng Python?

Để vẽ quang phổ cường độ, 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ụ.
  • Nhận giá trị gốc ngẫu nhiên.
  • Khởi tạo dt để biết khoảng thời gian lấy mẫu và tìm tần suất lấy mẫu.
  • Tạo các điểm dữ liệu ngẫu nhiên cho t.
  • Để tạo ra tiếng ồn, get nse, r, cnse s sử dụng numpy
  • Tạo một hình và một tập hợp các ô con, sử dụng phương thức subplots ().
  • Đặt tiêu đề của cốt truyện.
  • Vẽ biểu đồ của quang phổ cường độ.
  • Để 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

np.random.seed(0)

dt = 0.01 # sampling interval
Fs = 1 / dt # sampling frequency
t = np.arange(0, 10, dt)

# generate noise:
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(4 * np.pi * t) + cnse

fig, axs = plt.subplots()

axs.set_title("Magnitude Spectrum")
axs.magnitude_spectrum(s, Fs=Fs, color='C1')

plt.show()

Đầu ra

Làm thế nào để vẽ quang phổ cường độ trong Matplotlib bằng Python?