将numpy数组保存到txt文件行wis

Question

我有一个数字数组

a = [1,2,3]

我想将其保存到一个.txt文件中，使该文件看起来像：

1 2 3

如果我使用numpy.savetxt，则会得到如下文件：

1
2
3

我想应该有一个简单的解决办法，有什么建议吗？

Answer 1

import numpy
a = numpy.array([1,2,3])

with open(r'test.txt', 'w') as f:
    f.write(" ".join(map(str, a)))

Answer 2

如果numpy >= 1.5，您可以：

# note that the filename is enclosed with double quotes,
# example "filename.txt"

numpy.savetxt("filename", a, newline=" ")

编辑

几个长度相同的一维阵列

a = numpy.array([1,2,3])
b = numpy.array([4,5,6])
numpy.savetxt(filename, (a,b), fmt="%d")

# gives:
# 1 2 3
# 4 5 6

几种变长一维阵列

a = numpy.array([1,2,3])
b = numpy.array([4,5])

with open(filename,"w") as f:
    f.write("\n".join(" ".join(map(str, x)) for x in (a,b)))

# gives:
# 1 2 3
# 4 5

Answer 3

另一种方法是重塑数组的形状，使其具有像这样的维度(1, N)：

savetext(filename, a.reshape(1, a.shape[0]))