python - 使用 numpy 将数组写入标准输出

Question

将 Numpy 二维数组写入标准输出的惯用方法是什么？例如我有一个数组

a = numpy.array([[2., 0., 0.], [0., 2., 0.], [0., 0., 4.]])

[[ 2.  0.  0.]
 [ 0.  2.  0.]
 [ 0.  0.  4.]]

我希望输出为:

2.0 0.0 0.0
0.0 2.0 0.0
0.0 0.0 4.0

我可以通过转换为嵌套列表，然后加入列表元素来做到这一点:

print( '\n'.join( [ ' '.join( [ str(e) for e in row ] ) for row in a.tolist() ] ) )

但是想要这样的东西:

a.tofile( sys.stdout )

(除非这给出了语法错误)。

Answer 1

下面的代码怎么样？

>>> a = numpy.array([[2., 0., 0.], [0., 2., 0.], [0., 0., 4.]])
>>> numpy.savetxt(sys.stdout, a, fmt='%.4f')
1.0000 2.0000 3.0000
0.0000 2.0000 0.0000
0.0000 0.0000 4.0000

在 Python 3+ 中，使用 numpy.savetxt(sys.stdout.buffer, ...)。