如何生成numpy矩阵的漂亮输出？

Question

我目前有以下代码片段：

#!/usr/bin/python
# -*- coding: utf-8 -*-

import numpy
from numpy import linalg

A = [[1,2,47,11],[3,2,8,15],[0,0,3,1],[0,0,8,1]]
S = [[113,49,2,283],[-113,0,3,359],[0,5,0,6],[0,20,0,12]]

A = numpy.matrix(A)
S = numpy.matrix(S)

numpy.set_printoptions(precision=2, suppress=True, linewidth=120)
print("S^{-1} * A * S")
print(linalg.inv(S) * A * S)

以下是输出的效果：

是否有一种标准方法可以产生类似于上图的输出？我该如何获得这个输出？

[[ -1    -0.33  0  0]
 [  0     1     0  0]
 [  0  -648     4  0]
 [  0     6.67  0  5]]

有什么不同之处？

• 第i列的最后一个字符与第i+1列的第一个字符之间至少有两个空格，但如果需要更多则可能会有更多（NumPy输出使其为两个空格）
• 点是对齐的（是对齐的，但BetterPythonConsole的字体设置搞砸了它）
• 没有 -0 而是 0
• 没有 0. 而是 0

编辑：当我从终端启动Python时，似乎使用gEdit的BetterPythonConsole插件启动的Python控制台与Python做了一些不同的事情。

这是上面脚本的文本输出结果

moose@pc07:~/Desktop$ python matrixScript.py 
S^{-1} * A * S
[[  -1.     -0.33    0.     -0.  ]
 [   0.     -1.     -0.      0.  ]
 [   0.   -648.      4.     -0.  ]
 [   0.      6.67    0.      5.  ]]

使用prettyprint:

S^{-1} * A * S
matrix([[  -1.  ,   -0.33,    0.  ,   -0.  ],
        [   0.  ,   -1.  ,   -0.  ,    0.  ],
        [   0.  , -648.  ,    4.  ,   -0.  ],
        [   0.  ,    6.67,    0.  ,    5.  ]])

这肯定更糟糕，但值得一试。

- Martin Thoma

---

可能是NumPy数组的漂亮打印的重复问题。 - Chris

这不是重复的。我已经阅读了这个问题（和答案），但它没有提供足够的格式。 - Martin Thoma

你尝试过使用pprint吗？https://dev59.com/aHI_5IYBdhLWcg3wHfKr - Dmitry Zagorulkin

@ZagorulkinDmitry 感谢你的提示。因为你的提示，我已经从终端执行了它。我已经编辑了我的帖子。我没有考虑到 BetterPythonConsole 的错误字体设置。 - Martin Thoma

Answer 1

如果您使用的是numpy 1.8.x版本，您可以使用formatter参数自定义格式。例如，设置:

numpy.set_printoptions(formatter={'float': lambda x: 'float: ' + str(x)})

所有的浮点数都会被打印成 float: 3.0 或者 float: 12.6666666666。

不幸的是，我还安装了numpy 1.6.1版本，无法使用此选项来获取您想要的输出结果。

- Bakuriu

---

我猜更新我的系统可能是最好的选择，因为格式化中的所有其他问题似乎都与BetterPythonConsole有关。 - Martin Thoma

是的，另一个选项可能是编写自定义的“format”函数以与numpy数组一起使用。这可能不太难写，但如果您想使其稍微通用以供将来可能使用，那么这不是您可以在2分钟内编写的东西。 希望此升级除了这个之外还能给您其他好处。 - Bakuriu