matplotlib 如何获得PSD的幅值

7lrncoxx 于 2022-12-19 发布在其他

关注(0)|答案(1)|浏览(95)

我在这里分享我提取的python代码：Plotting a fast Fourier transform in Python。我正在制作傅里叶频谱，然后我想“挑选”从图上的特定点获得值。在下一张图片中，您可以看到50 Hz频率，但我想知道下一个峰值的频率值。我如何找到它？

import numpy as np
import matplotlib.pyplot as plt
import scipy.fftpack

N = 600
# sample spacing
T = 1.0 / 800.0
x = np.linspace(0.0, N*T, N)
y = np.sin(50.0 * 2.0*np.pi*x) + 0.5*np.sin(80.0 * 2.0*np.pi*x)
yf = scipy.fftpack.fft(y)
xf = np.linspace(0.0, 1.0/(2.0*T), N//2)

fig, ax = plt.subplots()
ax.plot(xf, 2.0/N * np.abs(yf[:N//2]))
plt.show()

matplotlib

来源：https://stackoverflow.com/questions/74839276/how-to-get-the-amplitude-of-psd

1条答案

按热度按时间

txu3uszq1#

以下代码的结果将峰分离为50.836120401337794

import numpy as np
import matplotlib.pyplot as plt
import scipy.fftpack

# Set the parameters for the signal and the Fourier transform
N = 600
T = 1.0 / 800.0
x = np.linspace(0.0, N*T, N)
y = np.sin(50.0 * 2.0*np.pi*x) + 0.5*np.sin(80.0 * 2.0*np.pi*x)
yf = scipy.fftpack.fft(y)
xf = np.linspace(0.0, 1.0/(2.0*T), N//2)

# Find the indices of the peaks in the spectrum
peak_indices = np.argmax(np.abs(yf[:N//2]), axis=0)

# Get the corresponding frequency values from the xf array
peaks = xf[peak_indices]

# Print the frequency values of the peaks
print(peaks)

# Generate the Fourier spectrum plot
fig, ax = plt.subplots()
ax.plot(xf, 2.0/N * np.abs(yf[:N//2]))
plt.show()

赞(0）回复(0）举报 2022-12-19

我来回答

matplotlib 如何获得PSD的幅值

1条答案

相关问题

热门标签

最新问答