NumPy：将小数转换为分数

Question

我计算矩阵 A 的反转，例如，

import numpy as np

A = np.diag([1, 2, 3])
A_inv = np.linalg.pinv(A)
print(A_inv)

我知道了，

[[ 1.          0.          0.        ]
 [ 0.          0.5         0.        ]
 [ 0.          0.          0.33333333]]

但是，我想要这个，

[[ 1.          0.          0. ]
 [ 0.          1/2         0. ]
 [ 0.          0.          1/3]]

我尝试了np.set_printoptions，

import fractions
np.set_printoptions(formatter={'all':lambda x: str(fractions.Fraction(x))})
print(A_inv)

但我得到了这个，

[[1 0 0]
 [0 1/2 0]
 [0 0 6004799503160661/18014398509481984]]

如何在NumPy中将小数转换为分数？

Answer 1

这是一个浮点问题 - 回想一下2/3在Pythons表示中并不完全是2/3。

Fraction类有一个内置方法limit_denominator()来处理这个问题：

import fractions
np.set_printoptions(formatter={'all':lambda x: str(fractions.Fraction(x).limit_denominator())})
print(A_inv)

这给出了期望的答案：

[[1 0 0]
 [0 1/2 0]
 [0 0 1/3]]