帮我写一份QAM调制python代码

以下是一个简单的QAM调制Python代码，其中使用了numpy和matplotlib库：

import numpy as np
import matplotlib.pyplot as plt

# QAM调制函数
def qam_modulation(data, m, n):
    # 将原始数据分为实部和虚部
    x = data.real
    y = data.imag

    # 计算每个符号的幅度和相位
    amp = np.sqrt(x**2 + y**2)
    phase = np.arctan2(y, x)

    # 根据幅度和相位计算调制后的信号
    mod_signal = np.zeros((m, n))
    for i in range(m):
        for j in range(n):
            index = i*n + j
            if index < len(data):
                mod_signal[i][j] = amp[index] * np.exp(1j*phase[index])
    
    return mod_signal

# 生成随机数据
data = np.random.randint(0, 4, size=16) * np.exp(1j*np.random.rand(16)*2*np.pi)

# 调制后的信号
mod_signal = qam_modulation(data, 4, 4)

# 绘制调制后的信号
plt.scatter(mod_signal.real, mod_signal.imag)
plt.show()

在这个示例代码中，我们生成了一个16个符号的QAM调制信号，其中每个符号的幅度和相位是随机生成的。然后我们使用`qam_modulation`函数将这些数据进行调制，生成一个4x4的QAM调制信号矩阵。最后，我们使用matplotlib库绘制了调制后的信号。