python: 不同类型的重载运算符 >

Question

告诉我如何重载运算符 > (lt) 或 < (gt) 等以便它可以处理不同的类型（尤其是 0）

例如：

class CRational(object):

    def __init__(self, a = 0, b = 1):
        self.a = a
        self.b = b
        
    def __neg__(self):
        return CRational(-self.a, self.b)

    def __add__(self, other):
        return CRational(self.a * other.b + other.a * self.b, self.b * other.b)

    def __sub__(self, other):
        return CRational(self.a * other.b - other.a * self.b, self.b * other.b)

    def __mul__(self, other):
        return CRational(self.a * other.a, self.b * other.b)

    def __truediv__(self, other):
        return CRational(self.a * other.b, self.b * other.a)

    def __eq__(self, other):
        return self.a * other.b == other.a * self.b

    def __ne__(self, other):
        return not self.__eq__(other)

    def __lt__(self, other):
        return self.a * other.b < other.a * self.b

    def __le__(self, other):
        return self.a * other.b <= other.a * self.b

    def __gt__(self, other):
        return self.a * other.b > other.a * self.b

    def __ge__(self, other):
        return self.a * other.b >= other.a * self.b

错误输出：

File "lessons/python/square_eq.py", line 66, in __gt __ return self.a * other.b > other.a * self.b AttributeError: 'int' object has no attribute 'b'

在代码中：

s_a = "" if a > 0 else "-"

你是怎么解决的？

s_a = "" if a > CRational(0) else "-"

上面描述的方法有帮助，但并不漂亮:)

Answer 1

关于您的错误输出：您收到错误是因为您的“其他”是一个 int 并且 CRational.__gt__ 正在尝试访问它没有 b 的属性：

def __gt__(self, other):  # passing in an int here
        return self.a * other.b > other.a * self.b

关于您的代码示例，假设 a 是另一个 CRational 对象，那么 a > 0 将导致您在上面看到的 AttributeError 并且唯一的方法通过将它与另一个 CRational 对象或具有属性 a 和 b.

的其他对象进行比较来修复它

Answer 2

听起来您想测试您正在使用的对象类型。也许是这样的：

import numbers

def __gt__(self, other):
    if isinstance(other, type(self)):
        return self.a * other.b > other.a * self.b
    elif isinstance(other, numbers.Number):
        return self.a > other
    else:
        err_str = "Unknown type in comparison with %s object: %s" % (type(self).__name__, type(other).__name__)
        raise(TypeError(err_str))

这里，type(self) 是获取 CRational class 的通用方法。这样，如果您以后只需更改名称，就不必修改 class 的代码。

isinstance 检查给定对象是给定类型还是子类型。 Python documentation.

我编造了数字大小写，因为我不知道你想如何定义它。

Answer 3

如果您想将 CRational 对象与 int 进行比较，那么您的 __gt__ 方法也应该适用于整数。也就是说，如果 other 是一个整数，你显然不能做像 other.b 这样的事情。这是一个可能的解决方案：

class CRational:
    def __init__(self, a = 0, b = 1):
        self.a = a
        self.b = b
    def __gt__(self, other):
        if isinstance(other, CRational):
            return self.a * other.b > other.a * self.b
        elif isinstance(other, int):
            # Compare self (a CRational object) with other (an int)
            # TODO
        else:
            raise NotImplemented()

现在你可以这样做：

a = CRational()
if a > 3:
    ...

不过要小心！即使你把所有的方法都正确实现了，你还是做不到3 > a。顺序很重要！！ 3 > a 会调用 int class 的 __gt__ 方法。您只能做 a > 3、a < 3、a >= 3 等

python: 不同类型的重载运算符 >

python: overload operator > with different types

python

class

operator-overloading

operators

python-3.x