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

这是一个浮点问题 - 回想一下，在 Python 表示中 2/3 不完全是 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]]