python - numpy.savetxt 在新行上保存每个元素？

Question

with file("features.txt", "w") as outfile:
    # temp_array is an array of shape (500L,)
    np.savetxt(outfile, X=np.array(temp_array, dtype='uint8'), delimiter=",")

我使用上述语法在文本文件中存储了大约 10,000 个数组，但我注意到所有元素都存储在不同的行中。

如果我使用 numpy.loadtxt 加载文本文件，我将得到一个数组而不是矩阵。

我知道可以在加载的数组上使用 numpy.reshape 将其转换为矩阵，但在我的应用程序中，加载它的用户不知道行数。

如何解决这个问题？

Answer 1

如果您给 np.savetxt 一个一维数组，它会将所述数组视为一列并将每个条目写入新行。我可以想到两种解决方法。第一种是向数组添加一个新轴，使其成为行向量:

x = np.arange(5)
np.savetxt('test.txt', x[np.newaxis], fmt='%d', delimiter=',')

第二种方法是告诉 np.savetxt 使用不同于 \n 的不同字符作为换行符。比如一个空格:

np.savetxt('test.txt', x, fmt='%d', newline=' ', delimiter=',')

这两者都会产生一个看起来像这样的文件

0 1 2 3 4