将3D numpy数组保存到.txt文件

时间:2018-07-07 00:11:24

标签: python numpy save

我只是问这个问题,因为它在两天前就可以在我的台式机上运行了,但是当我试图在笔记本电脑上运行它时,它不起作用。

array = numpy.zeros((3,3,1))
numpy.savetxt("nxx", array, fmt='%s')

这给了我以下错误:

“预期的1D或2D数组,取而代之的是%dD数组”%X.ndim) ValueError:预期为1D或2D数组,取而代之的是3D数组

1 个答案:

答案 0 :(得分:0)

the docs所述,savetxt只能保存1D和2D数组。

您将如何保存savetxt保存到CSV文件中:用空格隔开的列,用换行符隔开的行,在文本编辑器中看起来很漂亮,是2D的。您如何将其扩展到3D?

如果您想使用某种特定的格式,则可能有一个格式化程序(或者可能只是reshape-将数组降为2D的问题?),但是您我将不得不解释你想要什么。

如果您不想在文本编辑器中编辑输出,为什么不只将其保存为二进制格式,例如numpy.save("nxx.npy", array)?可以处理任何形状。


IIRC,NumPy的旧版本通过将除一个维度以外的所有维度展平为一个维度来解决此问题。因此,要使用您的数据,您必须先加载它,然后对其进行整形-仅当您从其他地方知道原始尺寸是什么时才可以这样做。直到整个星期都花了几TB的文件,而现在您不知道该如何使用,您甚至都不会意识到这个问题。 (我想这就是他们改变行为的原因-这种行为被偶然使用的次数多于有意使用的次数,并带来令人讨厌的后果。)

无论如何,如果您想要这种行为,那么将阵列自己修复为2D并保存,当然是微不足道的。

np.savetxt("nxx", array.reshape((3,-1)), fmt="%s")

或者,甚至更好:

np.savetxt("nxx", array.reshape((3,-1)), fmt="%s", header=str(array.shape))