Question

我正在寻找将数组保存在文件中的最简单方法。为此，我想使用numpy.savetxt，但问题是我的数组由n列组成（数字取决于我要做什么），它包含复杂元素（x + yj）。我知道如果有一列和真实元素但是我不知道该怎么做才能保存它。

有人有想法吗？

Answer 1

你可以pickle他们：

>>> A = np.array([[1,2],[3,4+2j]])
>>> pickle.dump(A, open("out.pkl", "wb"))
>>> pickle.load(open("out.pkl", "rb"))
array([[ 1.+0.j,  2.+0.j],
       [ 3.+0.j,  4.+2.j]])

但是，最好使用numpy.save和numpy.load，它们是为此设计的，并且会占用更少的空间。

>>> np.save("out.npy", A)
>>> np.load("out.npy")
array([[ 1.+0.j,  2.+0.j],
       [ 3.+0.j,  4.+2.j]])

用python写入文件

1 个答案: