Python 3 matplotlib添加多尺度轴水印

import matplotlib.pyplot as plt
import numpy as np

# 创建数据
t = np.linspace(0, 10, 1000)
signal = np.sin(2 * np.pi * 5 * t) + np.sin(2 * np.pi * 20 * t)

# 创建图表
fig, ax1 = plt.subplots()

# 绘制时域信号
color = 'tab:red'
ax1.set_xlabel('Time (s)')
ax1.set_ylabel('Amplitude', color=color)
ax1.plot(t, signal, color=color)
ax1.tick_params(axis='y', labelcolor=color)

# 创建第二个轴
ax2 = ax1.twinx()

# 计算频域数据
fft = np.fft.fft(signal)
freq = np.fft.fftfreq(len(signal), d=t[1]-t[0])
magnitude = np.abs(fft)

# 绘制频域信号
color = 'tab:blue'
ax2.set_ylabel('Magnitude', color=color)
ax2.plot(freq[:len(freq)//2], magnitude[:len(magnitude)//2], color=color)
ax2.tick_params(axis='y', labelcolor=color)

# 显示图表
plt.show()