python - 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

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