如何将大小（m，n）转换为大小（m，1，n）数组？

Question

我有一个train_dataset，大小为（237738，25），请参见下图，它是使用pandas数据框显示的。您可以使用train_set = train_set.values将其转换为NumPy数组。我如何将这个数组转换为 （237738,1,25）数组的大小？

谢谢！

Answer 1

刚发现，实际上可以直接使用numpy.reshape函数。即train_set = np.reshape（train_set，（train_set.shape [0]，1，train_set.shape [1]））！