以ASCII格式保存numpy数组

Question

我有一些数据，格式是一个二维数组。

my_array = np.array([[1,2,3],[4,5,6])

我该如何把这些数据保存成一个ascii文件呢？

另外，如果这个数组里有很多数据，会不会有什么问题呢？

我知道怎么把数据保存成txt文件，但那不是我想要的，或者说我需要以某种方式更改设置。

Answer 1

你可以使用 np.savetxt 这个函数来保存数据到文本文件。如果你想加载这个文本文件，可以用 np.loadtxt。

import numpy as np

a = np.array([[1, 2, 3, 4], [5, 6, 7, 8]])
np.savetxt('saved_array.txt', a)

b = np.loadtxt('saved_array.txt')
print(a == b) # validation step

另外，你也可以直接用Python自带的输入输出功能，把可打印的字符串写入文本文件。

import numpy as np

a = np.array([[1, 2, 3, 4], [5, 6, 7, 8]])

with open('saved_array.txt', 'w') as fl:
    fl.write(repr(a))