python - Numpy savetxt 将一维数组保存为列

Question

我使用loadtxt来初始化数组。

source = np.loadtxt('source.txt').astype(int)

之后我在函数中使用这个数组，其主体是:

file = open('johnson.txt', 'ab')
first = increase(np.argsort(source[0]))
np.savetxt(file, first, delimiter='-', fmt='%i')
file.close()

因此，在 txt 文件中我应该有这样的内容:

7-1-3-6-2-4-8-5

但是我有这个:

7
1
3
6
2
4
8
5

我必须以二进制模式打开文件，因为我需要将另一行附加到文件中。那么，我该如何解决这个问题呢？谢谢!

Answer 1

savetxt 迭代输入数组，并将每个“行”写入新行。对于典型的二维数组来说，这将是一行。但对于一维数组来说，这将是一个元素。

因此，将您的写入更改为保存二维数组:

np.savetxt('test.txt', [first], delimiter=..., fmt=...)

假设first是一个一维数组，那么np.array([first])是1行2d，first[None,:] 也可以(或 reshape )。

要追加行，请以“a”追加模式打开文件。 'wb'` 二进制文件没有帮助。