написать в файл с питоном

Question

Я ищу самый простой способ сохранить массив в файл.Для этого я хотел бы использовать numpy.savetxt, но проблема в том, что мой массив состоит из n столбцов (число зависит от того, что я хочу сделать) и содержит сложные элементы (x + yj).Я знаю, как сохранить его, если есть один столбец и реальные элементы, но я не знаю, как это сделать.

У кого-нибудь есть идея?

Answer 1

Вы могли бы мариновать их:

>>> 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]])