python:浮点数的最大公约数(gcd),最好是numpy

时间:2017-07-26 09:59:59

标签: python numpy

我正在寻找一种有效的方法来确定两个使用python的浮点数的最大公约数。例程应具有以下布局

gcd(a, b, rtol=1e-05, atol=1e-08)
"""
Returns the greatest common divisor of a and b

Parameters
----------
a,b : float
    two floats for gcd
rtol, atol : float, optional
    relative and absolute tolerance

Returns
-------
gcd : float
    Greatest common divisor such that for x in [a,b]:
    np.mod(x,gcd) < rtol*x + atol 

.. _PEP 484:
    https://www.python.org/dev/peps/pep-0484/

"""

示例:理性和无理数的gcd

gcd(1., np.pi, rtol=0, atol=1e-5)应该返回(大致)1e-5,如

In [1]: np.mod(np.pi,1e-5)
Out[1]: 2.6535897928590063e-06

In [2]: np.mod(1.,1e-5)
Out[2]: 9.9999999999181978e-06

我更喜欢使用库实现而不是自己编写。 fractions.gcd函数在这里似乎不适合我,因为我不想使用分数,它(显然)没有公差参数。

2 个答案:

答案 0 :(得分:4)

似乎您可以修改fractions.gcd的代码以包含公差:

def float_gcd(a, b, rtol = 1e-05, atol = 1e-08):
    t = min(abs(a), abs(b))
    while abs(b) > rtol * t + atol:
        a, b = b, a % b
    return a

答案 1 :(得分:1)

以下代码可能有用。调用的函数是float_gdc(a,b)。

from math import gcd, log10, pow

def float_scale(x):
    max_digits = 14
    int_part = int(abs(x))
    magnitude = 1 if int_part == 0 else int(log10(int_part)) + 1
    if magnitude >= max_digits:
        return 0
    frac_part = abs(x) - int_part
    multiplier = 10 ** (max_digits - magnitude)
    frac_digits = multiplier + int(multiplier * frac_part + 0.5)
    while frac_digits % 10 == 0:
        frac_digits /= 10
    return int(log10(frac_digits))


def float_gcd(a, b):
    sc = float_scale(a)
    sc_b = float_scale(b)
    sc = sc_b if sc_b > sc else sc
    fac = pow(10, sc)

    a = int(round(a*fac))
    b = int(round(b*fac))

    return round(gcd(a, b)/fac, sc)

部分代码摘自这里:Determine precision and scale of particular number in Python