numpy.savetxt：将一维数组写入多行多列

Posted 2023-05-07

【中文标题】numpy.savetxt：将一维数组写入多行多列

【英文标题】：numpy.savetxt: Write 1D array into multiple rows and colums

【发布时间】：2015-01-30 17:54:00

【问题描述】：

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

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

我想在多行和多列中写入 .txt 文件，以及中间的各种空格，例如：

HEADER
 1   2   3
 4   5   6
 7   8

应指定列数，调整行数。

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

【问题讨论】：

可能与***.com/questions/15192847/…有关

【参考方案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

【讨论】：

谢谢！但是，我想指定列数，而不是行数。 （我只是将其添加到说明中。）