打印numpy数组时仅删除零值的科学记数法？

Question

现在我的数组打印如下：

  [[ 1.5531e-01 -4.6594e-02  0.0000e+00  0.0000e+00  0.0000e+00  0.0000e+00]
 [-4.6594e-02  1.5531e-01  0.0000e+00  0.0000e+00  0.0000e+00  0.0000e+00]
 [ 0.0000e+00  0.0000e+00  4.0381e-01  0.0000e+00  0.0000e+00  0.0000e+00]
 [-0.0000e+00 -0.0000e+00 -0.0000e+00  1.9826e+03 -5.9479e+02 -0.0000e+00]
 [ 0.0000e+00  0.0000e+00  0.0000e+00 -5.9479e+02  1.9826e+03  0.0000e+00]
 [ 0.0000e+00  0.0000e+00  0.0000e+00  0.0000e+00  0.0000e+00  5.1549e+03]]

我想要的是：

[[ 1.4531e-01 -4.6594e-02  0           0           0           0         ]
 [-4.6594e-02  1.4531e-01  0           0           0           0         ]
 [ 0           0           4.0381e-01  0           0           0         ]
 [ 0           0           0           1.9826e+03 -5.9479e+02  0         ]
 [ 0           0           0          -5.9479e+02  1.9826e+03  0         ]
 [ 0           0           0           0           0           5.1549e+03]]

主要区别：

• 顶行对齐
• 零值显示为“0”，没有负号。

我打印了很多数组，所以我希望它是您可以设置的东西，而不是必须通过函数传递每个数组才能对其进行格式化。 np.set_printoptions 中是否有具有此功能的东西？

Answer 1

当然，您可以根据docs 在np.set_printoptions 中指定formatter。 下面的代码行将右填充设置为等于 0 的数字，否则以科学计数法打印它们，并保留符号。

np.set_printoptions(formatter={'all': lambda x: '{:<11d}'.format(int(x)) if x == 0 else "{:+.4e}".format(x)})

这是你得到的输出：

[[+1.5531e-01 -4.6594e-02 0           0           0           0          ]
 [-4.6594e-02 +1.5531e-01 0           0           0           0          ]
 [0           0           +4.0381e-01 0           0           0          ]
 [0           0           0           +1.9826e+03 -5.9479e+02 0          ]
 [0           0           0           -5.9479e+02 +1.9826e+03 0          ]
 [0           0           0           0           0           +5.1549e+03]]