将对象转换为另一个类

Question

我搜索过有关转换课程的类似问题，但没有一个对我有真正帮助。

我有一个 Fraction 类，如果可能，我想将它简化为一个 int。

class Fraction:
    def __init__(self, numerator, denominator):
        self.numerator = numerator
        self.denominator = denominator
        self.simplify()

        if self.denominator == 1:
            # convert self to int of value numerator

我在 Fraction 类中有一个函数，它可以在创建分数时简化分数（2/6 变为 1/3，等等）

    def simplify(self):
        div = gcd(self.numerator, self.denominator)
        self.numerator //= div
        self.denominator //= div

我目前对这个问题的解决方法如下：

def convert(fraction):
    if fraction.denominator == 1:
        return fraction.numerator  # int
    else:
        return fraction            # Fraction


f = Fraction(3, 1)
f = convert(f)

我在各种代数表达式中使用这些分数，所以我希望它们尽可能简单。例如。 1/2 + 1/2 = 1（而不是 1/1）

有什么方法可以将 Fraction 对象转换为 int 而不必每次都通过这个外部函数？

谢谢

Answer 1

你可以做几件事：

• 拥有返回两种不同类型的函数并不是最佳选择，因此您可以使用类方法处理转换。
• 如果 print() 函数也支持转换，这将有所帮助

考虑到这一点，我只是将 convert 移到类中，并添加了一个 str 方法。

from fractions import gcd

class Fraction:
    def __init__(self, numerator, denominator):
        self.numerator = numerator
        self.denominator = denominator
        self.simplify()

    def __str__(self):
        if self.denominator == 1:
            return str(self.numerator)
        return '{}/{}'.format(self.numerator,self.denominator)

    def simplify(self):
        div = gcd(self.numerator, self.denominator)
        self.numerator //= div
        self.denominator //= div
    def convert(fraction):
        if fraction.denominator == 1:
            return fraction.numerator  # int
        return fraction # Fraction

if __name__ == '__main__':
    f = Fraction(6, 2)
    c = Fraction.convert(f)
    print(f)
    print(c)
    f2 = Fraction(3, 4)
    print(f2)

看看能不能帮到你。