当前位置: 代码迷 >> python >> NumPy:将小数转换为分数
  详细解决方案

NumPy:将小数转换为分数

热度:122   发布时间:2023-06-13 15:18:32.0

我计算矩阵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中的分数?

这是一个浮点问题 - 回想一下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]]