導入/導出高等級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')