导入/导出高等级numpy数组到文本文件

Question

我有一个numpy.ndarray具有很高的排名 - 比如说，形状(100,100,100,2)。将其写入文本文件的最佳方法是什么，以便于在其他程序和numpy中导入？

Answer 1

如果您扁平化阵列并在标题中包含阵列形状（第一行），则可以使用np.savetxt）。你知道如何当你在其他程序中打开它重塑数组方式：

import numpy as np 
a = np.random.rand(100,100,100,2) 
np.savetxt('test.txt', a.flatten(), header=str(a.shape)) 

要导入数据，重塑它，你能做到以下几点，例如：

import re 
with open('test.txt') as f: 
    shape = tuple(int(num) for num in re.findall(r'\d+', f.readline())) 
data = np.loadtxt('test.txt').reshape(shape) 

Answer 2

你可以使用tofile，也是numpy（link to doc）的一部分。要读取二进制文件，只需使用fromfile,doc here。

我不确定可移植性，但如果速度很重要，写入二进制文件是更好的选择。

import numpy as np 
a = np.random.rand(100,100,100,2) 
a.tofile('test.txt')