numpy.savetxt：将1D数组写入多行和多列

Question

我有像

这样的1D 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