如何使用 numpy.savetxt 保存和加载复数数组？

Question

我想使用 numpy.savetxt() 将复数数组保存到文本文件中。问题：

• 如果使用默认格式字符串保存复数数组，则虚部将被丢弃。
• 如果您使用 fmt='%s'，则 numpy.loadtxt() 无法加载它，除非您指定 dtype=complex, converters={0: lambda s：复数}。即使如此，如果数组中存在 NaN，加载仍然会失败。

看起来有人询问过这个多个 Numpy 上的时间邮件列表，甚至提交了 bug，但尚未得到回复。在我自己组装一些东西之前，有没有一种规范的方法可以做到这一点？

Answer 1

这是我的解决方案，以防有人从谷歌上遇到这个问题。

保存：

numpy.savetxt('outfile.txt', numpy.column_stack([array.real, array.imag]))

加载：

array_real, array_imag = numpy.loadtxt('outfile.txt', unpack=True)
array = array_real + 1j * array_imag

我仍然会将复选标记授予更好的解决方案！

Answer 2

它更容易并保存一些临时数组，以便将数组重新解释为真正的数组。

保存：

numpy.savetxt('outfile.txt', array.view(float))

加载：

array = numpy.loadtxt('outfile.txt').view(complex)

如果您希望实部和虚部位于文件的同一行，可以

numpy.savetxt('outfile.txt', array.view(float).reshape(-1, 2))

使用或

array = numpy.loadtxt('outfile.txt').view(complex).reshape(-1)

分别

。 （请注意，view() 和 reshape() 都不会复制数组 - 它只会以不同的方式重新解释相同的数据。）

问题的附录提问者：

如果你想在同一个文件中保存多个复杂数组，你可以这样做：

numpy.savetxt('outfile.txt', numpy.column_stack([
    array1.view(float).reshape(-1, 2),
    array2.view(float).reshape(-1, 2),
]))

array1, array2 = numpy.loadtxt('outfile.txt', unpack=True).view(complex)

重塑是必要的，因为 numpy.view() 不操作跨步数​​组。