python - numpy 将二维数组保存到文本文件

Question

我用

np.savetxt('file.txt', array, delimiter=',')

将数组保存到以逗号分隔的文件中。看起来像:

1, 2, 3
4, 5, 6
7, 8, 9

如何将数组保存到以 numpy 格式显示的文件中。换句话说，它看起来像:

[[1, 2, 3],
[4, 5, 6],
[7, 8, 9]]

Answer 1

In [38]: x = np.arange(1,10).reshape(3,3)    

In [40]: print(np.array2string(x, separator=', '))
[[1, 2, 3],
 [4, 5, 6],
 [7, 8, 9]]

---

将 NumPy 数组 x 保存到文件中:

np.set_printoptions(threshold=np.inf, linewidth=np.inf)  # turn off summarization, line-wrapping
with open(path, 'w') as f:
    f.write(np.array2string(x, separator=', '))