python中实现1000个数据点，经过stft变换变为（250，250）的数组的代码

以下是Python代码实现：

import numpy as np
from scipy import signal

# 生成1000个随机数数据点
data = np.random.rand(1000)

# 设置STFT参数
window_length = 25  # 窗口长度
step = 10  # 步长
nfft = 250  # 快速傅里叶变换点数

# 利用scipy库的signal模块计算STFT
f, t, zxx = signal.stft(data, fs=1.0, nperseg=window_length,
                        noverlap=window_length-step, nfft=nfft)

# zxx为(frequency, time)复数矩阵，取它的振幅得到(时间,频率)矩阵
amplitude = np.abs(zxx)

# 缩放振幅矩阵到[0,1]之间
amplitude = (amplitude - amplitude.min()) / (amplitude.max() - amplitude.min())

# 将振幅矩阵重新调整为(250,250)形状
amplitude_resized = np.zeros((nfft//2, nfft//2))
amplitude_resized[:, :125] = amplitude[:, 125:]
amplitude_resized[:, 125:] = amplitude[:, :125]

# 打印调整后的振幅矩阵形状
print(amplitude_resized.shape)

运行上述代码后，将输出形状为(250,250)的振幅矩阵。