如何使 Julia output 向量与 Numpy 一样漂亮？

Question

如果我执行以下操作：

A = [
  2   1   3   0   0;
  1   1   2   0   0;
  0   6   0   2   1;
  6   0   0   1   1;
  0   0 -20   3   2
]
b = [10; 8; 0; 0; 0]
println(A\b)

output 是：

[8.000000000000002, 12.0, -6.000000000000001, -23.999999999999975, -24.000000000000043]

但是，我更希望它看起来类似于 Numpy 输出相同问题结果的方式（编辑：最好保留尾随零和逗号）：

[  8.  12.  -6. -24. -24.]

是否有捷径可寻？ 当然，我可以编写自己的 function 来执行此操作，但如果我可以设置一些格式化标志，那就太好了。

谢谢！

Answer 1

标准方法是更改IOContext ：

julia> println(IOContext(stdout, :compact=>true), A\b)
[8.0, 12.0, -6.0, -24.0, -24.0]

你可以写你的 function 例如（我不想在这里完全笼统，而是向你展示这个想法）：

printlnc(x) = println(IOContext(stdout, :compact=>true), x)

然后只需在您的代码中调用prinlnc 。

Answer 2

您可以通过重写浮点数的Base.show方法来更改 Julia 中的 REPL 行为。 例如：

Base.show(io::IO, f::Float64) = print(io, rstrip(string(round(f, digits=7)),'0') )

现在你有：

julia> println(A\b)
[8., 12., -6., -24., -24.]

正如@DNF Julia 所指出的那样，在向量中使用逗号。 如果你想要一个水平向量（实际上是一个 1xn 矩阵）你需要转置：

julia> (A\b)'
1×5 adjoint(::Vector{Float64}) with eltype Float64:
 8.  12.  -6.  -24.  -24.

julia> println((A\b)')
[8. 12. -6. -24. -24.]

Answer 3

Numpy 骗你。 它只是在打印时隐藏数字。 要检查它是否仅操纵 output 的打印，请执行print(A @ X - b)并查看结果。

print(A @ X - b)
[ 0.00000000e+00  0.00000000e+00  0.00000000e+00 -3.55271368e-15  0.00000000e+00]

另一方面，Julia 预先明确了这一点。 如果你在 Julia 中做同样的事情，你会得到相同的结果（我使用 Foat64 和 Numpy 一样）：

julia> X = A \ b;

julia> Float64.(A) * X - b
5-element Vector{Float64}:
  0.0
  0.0
  0.0
 -3.552713678800501e-15
  0.0

但是，您可以使用类似于 arrays 的压缩打印来删除不需要的数字。

julia> round.(X, digits=7)
5-element Vector{Float64}:
   8.0
  12.0
  -6.0
 -24.0
 -24.0

这个，比那个“丑”的8. 12. -6. -24. -24. 8. 12. -6. -24. -24.