如何将Numpy的4D数组保存成CSV文件?

5
我正在尝试将一系列图片保存到CSV文件中,以便用作Alexnet Keras机器学习的训练数据集。其形状为(15,224,224,3)。
目前我遇到了问题,无法完成这个任务。我已经成功将所有数据存储到numpy数组中,但现在无法将它保存到文件中。
请提供帮助。
2个回答

22

我认为将其保存到csv文件并不是正确的方法,它用于创建类似表格的1d或2d数据结构,因此我会提供另一种解决方案。你可以使用np.save来将任何numpy数组保存到文件中;

np.save('file_name', your_array)

然后可以使用np.load加载;

loaded_array = np.load('file_name.npy')

希望这对你有用。


1
你可以尝试使用pickle来保存数据。与np.save相比,它更加多样化且易于处理。

请注意,pickle 被认为不安全 - Bentoy13

网页内容由stack overflow 提供, 点击上面的
可以查看英文原文,
原文链接