广义S变换 python代码，注释每一行代码的作用

好的，以下是添加注释后的广义S变换的示例代码：

import numpy as np
import matplotlib.pyplot as plt
from scipy.fftpack import fft2, fftshift

# 构造二维信号
N = 128
t = np.linspace(0, 2*np.pi, N)
X, Y = np.meshgrid(t, t)  # 生成网格坐标矩阵
Z = np.sin(16*X)*np.cos(16*Y)  # 构造二维正弦余弦信号

# 将信号按照规则排列成矩阵
M = np.zeros((2*N-1, N))  # 初始化矩阵
for i in range(N):
    M[i:i+N, i] = Z[:, i]  # 将列向量插入到矩阵的第i列

for i in range(1, N):
    M[N-1+i:2*N-1, i+N-1] = Z[i:, :N-i].diagonal()  # 将斜线向量插入到矩阵的第i+N-1列

# 进行广义S变换
Y = fft2(M)  # 对矩阵进行二维傅里叶变换
Y = fftshift(Y)

# 可视化输出
plt.imshow(np.abs(Y), cmap='gray')  # 可视化幅值大小
plt.show()

注释中解释了每个步骤的作用，方便理解代码的实现过程。