numpy.savetxt一维数组写入的问题

Question

我正在尝试使用numpy的savetxt函数来生成一堆文件作为另一个软件的输入.

我正在尝试编写一个表单数组:

a=np.array([1,2,3,4,...])
a.shape=>(1,n)

到格式为1,2,3,4的文本文件,...

当我输入命令

np.savetxt('test.csv',a,fmt='%d',delimiter=',')

我得到一个看起来像这样的文件:

1

2

3

4

...

savetxt 像我期望的2D阵列那样工作,但我不能将1D阵列的所有值都放在一行上

有什么建议？

谢谢

编辑:

我解决了这个问题.使用np.atleast_2d(a)savetxt作为输入强制savetxt将数组写为行而不是列

Answer 1

有不同的方法来解决这个问题.最接近您当前方法的是:

np.savetxt('test.csv', a[None], fmt='%d', delimiter=',')

[None]即将切片添加到您的数组,使其只有一行二维.