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