欢迎您访问 最编程 本站为您分享编程语言代码,编程技术文章!
您现在的位置是: 首页

完美解决 ValueError:shape=(20000,1,1024),这样的多维数据保存到 excel 中的细节!-第二,错误分析

最编程 2024-07-08 16:07:38
...

我试图将一个形状为(20000, 1, 1024)的数据列表转换为pandas DataFrame,然后保存为Excel文件。

因为to_excel需要2维输入,所以报错了。

经过分析发现我的data_list是一个三维的tensor,有20000个样本,每个样本是一个1x1024的向量

因此需要把它reshape成二维,有3种方法:

  1. data_list = data_list.reshape(-1,1024)

  2. data_list = data_list.squeeze()

  3. data_list = data_list.reshape(20000,1024)