用python写锁相放大器原理代码

下面是一个简单的锁相放大器原理的Python代码示例：

import numpy as np

# 输入信号
t = np.linspace(0, 1, 1000)
x = np.sin(2 * np.pi * 10 * t)

# 参考信号
ref = np.sin(2 * np.pi * 10 * t + np.pi/4)

# 相关函数
corr = np.correlate(x, ref, mode='same')

# 带通滤波器
fc = 10
bw = 2
b, a = signal.butter(3, [fc-bw, fc+bw], btype='band')
filt_corr = signal.filtfilt(b, a, corr)

# 输出信号
out = filt_corr / max(filt_corr)

# 绘图
import matplotlib.pyplot as plt
plt.figure(figsize=(12,6))
plt.subplot(2,1,1)
plt.plot(t, x, label='input')
plt.plot(t, ref, label='reference')
plt.legend()
plt.subplot(2,1,2)
plt.plot(t, filt_corr, label='correlation')
plt.plot(t, out, label='output')
plt.legend()
plt.show()

该代码演示了锁相放大器的基本原理，包括输入信号、参考信号、相关函数、带通滤波器和输出信号。其中，输入信号是一个正弦波，参考信号是另一个相位不同的正弦波，相关函数是输入信号和参考信号的乘积，带通滤波器用于滤除高频噪声，输出信号是经过滤波后的相关函数的归一化值。