如何使用python将x轴从样本转换为时间(s)和频率(Hz)？

import numpy as np
import matplotlib.pyplot as plt

# 假设我们有一个采样率为1000Hz的信号，长度为1000个样本
sampling_rate = 1000  # 采样率 (Hz)
signal_length = 1000  # 信号长度 (样本数)

# 生成一个简单的正弦波信号
t = np.linspace(0, signal_length / sampling_rate, signal_length, endpoint=False)
signal = np.sin(2 * np.pi * 50 * t) + np.sin(2 * np.pi * 120 * t)

# 将x轴从样本转换为时间 (s)
time_axis = t

# 将x轴从样本转换为频率 (Hz)
frequency_axis = np.fft.fftfreq(signal_length, d=1/sampling_rate)

# 计算信号的FFT（快速傅里叶变换）
fft_signal = np.fft.fft(signal)

# 绘制时域信号
plt.figure(figsize=(12, 6))
plt.subplot(2, 1, 1)
plt.plot(time_axis, signal)
plt.title('Time Domain Signal')
plt.xlabel('Time (s)')
plt.ylabel('Amplitude')

# 绘制频域信号
plt.subplot(2, 1, 2)
plt.plot(frequency_axis[:signal_length//2], np.abs(fft_signal[:signal_length//2]))
plt.title('Frequency Domain Signal')
plt.xlabel('Frequency (Hz)')
plt.ylabel('Amplitude')

plt.tight_layout()
plt.show()