python - np.savetxt 和多个数组

Question

我有三个数组:

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

我想做这样的事情:

np.savetxt('data.txt',np.array(a,b,c))

或者只是

np.savetxt('data.txt',(a,b,c))

但是我明白了

 TypeError: float argument required, not numpy.ndarray

我已经通过类似

的方式规避了这个问题

np.savetxt('data.txt',np.array([a[0],a[1],...,c[2]]))

但这不是很令人满意(特别是因为数组大小可以更改)并且还将所有值绘制在单列中而不是单行和多列中(我想要它)。

问题似乎是数组的维度不同。我最喜欢使用上面的示例的输出类型是:

   1 2 3 4 5 6 7 8

有谁知道更好的方法吗？

Answer 1

您也可以尝试将它们与 numpy.r_ 连接起来

np.savetxt('data.txt',np.r_[a,b,c])