train_data.shape[0] #训练集样本数量
train_data.shape[1] - 1 #每个样本的采样点数量类似于img.sahpe[0]就是图像的高度
在这里的test_data
这个矩阵是2040*401
每一行是一个振动点的数据
2040列就是2040个test的数据