python将数组保存至文件_python – 将numpy数组保存到文件(尽可能小的大小)

由于你有很多零,你只能写出表格中的非零元素(索引,数字).

假设您有一个包含少量非零数字的数组：

In [5]: a = np.zeros((10, 10))

In [6]: a

Out[6]:

array([[ 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.],

[ 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.],

[ 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.],

[ 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.],

[ 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.],

[ 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.],

[ 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.],

[ 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.],

[ 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.],

[ 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.]])

In [7]: a[3,1] = 2.0

In [8]: a[7,4] = 17.0

In [9]: a[9,0] = 1.5

首先,隔离有趣的数字及其指数：

In [11]: x, y = a.nonzero()

In [12]: zip(x,y)

Out[12]: [(3, 1), (7, 4), (9, 0)]

In [13]: nonzero = zip(x,y)

现在您只剩下少量数据元素.最简单的方法是将它们写入文本文件：

In [17]: with open('numbers.txt', 'w+') as outf:

....: for r, k in nonzero:

....: outf.write('{:d} {:d} {:g}\n'.format(r, k, a[r,k]))

....:

In [18]: cat numbers.txt

3 1 2

7 4 17

9 0 1.5

这也让您有机会了解数据.在C程序中,您可以使用fscanf读取此数据.

但是你可以使用struct编写二进制数据来进一步减小尺寸：

In [17]: import struct

In [19]: c = struct.Struct('=IId')

In [20]: with open('numbers.bin', 'w+') as outf:

....: for r, k in nonzero:

....: outf.write(c.pack(r, k, a[r,k]))

Struct构造函数的参数意味着;使用原生日期格式’=’.第一个和第二个数据元素是无符号整数“I”,第三个元素是双“d”.

在您的C程序中,这些数据可能最好作为二进制数据读入打包结构.

编辑：为2D阵列更新了答案.