numpy.savetxt: Написать 1D массив в несколько строк и colums

Question

У меня есть 1D NumPy массив как

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

Что я хочу написать в текстовый файл в несколько строк и столбцов, а также различных пробелов между ними , например:

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