numpy.savetxt:将一维数组写入多个行和列

Question

我有一个一维numpy数组，像这样

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

我希望以多行和多列的形式将其写入到.txt文件中，并在其间添加各种空格，例如：

HEADER
 1   2   3
 4   5   6
 7   8

应指定列数，调整行数。

有没有一种简单的方法可以使用numpy.savetxt或类似的东西来做到这一点？

Answer 1

您可以使用array_split：

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

with open('out_file.txt','w') as f :
     for i in np.array_split(a,3) :
         f.write('  '.join(map(str,i)))

结果：

1  2  3
4  5  6
7  8