商城首页欢迎来到中国正版软件门户

您的位置: 首页 > 文章列表 > 编程开发 > Python类如何支持位运算操作_通过__and__与__or__重载符号

Python类如何支持位运算操作_通过__and__与__or__重载符号

  发布于2026-07-20 阅读(0)

扫一扫,手机访问

在Python中,让类支持&|运算符,需要定义__and____or__方法。但有个关键点常被忽略:必须显式返回一个非None的值,比如返回一个新实例。否则,像if obj1 & obj2这样的判断会直接报错,因为None没有__bool__方法。另外,如果想让代码支持链式运算,比如a & b & c,每次调用都得返回同类实例。

Python类如何支持位运算操作_通过__and__与__or__重载符号

Python类怎么实现&|运算符重载

直接在类中定义__and____or__方法就行,这两个是Python内置的反射方法,对应&|操作符。调用时Python会自动触发,不需要手动调用。

需要注意的是,只有当左操作数是该类实例时,这两个方法才会被调用。如果左操作数不支持,而右操作数实现了对应的__rand____ror__,才可能回退调用——不过在大多数自定义场景中,只需要实现左边版本就够了。

  • __and__(self, other)必须返回一个值,不能只写逻辑不return,否则默认返回None,导致表达式结果为None
  • 参数other类型不做限制,可以加类型检查,比如只接受同类型或int,也可以宽松处理
  • 若想支持链式运算,比如a & b & c,每个__and__的返回值也得是同类实例,否则后续的&会失败

为什么__and__返回None会导致bool()报错

常见错误现象:if obj1 & obj2:TypeError: __bool__ returned non-bool (type NoneType)。这是因为__and__没有显式return,隐式返回None,而if语句需要对结果调用__bool____len__,但None不支持这些协议。

  • 务必在__and__/__or__中return一个有明确真值的对象,比如新实例、int、bool
  • 不要只写self.flag &= other.flag就结束——这行代码本身不产生返回值
  • 若逻辑上只需“就地修改”,也应显式return self,前提是允许原地变更

实际例子:用位运算模拟权限组合

假设你有一个Permission类,内部用整数存储权限位,希望支持p1 & p2取交集、p1 | p2取并集:

class Permission:    def __init__(self, bits=0):        self.bits = bits & 0xFF  # 限制为8位
def __and__(self, other):    if not isinstance(other, Permission):        return NotImplemented    return Permission(self.bits & other.bits)def __or__(self, other):    if not isinstance(other, Permission):        return NotImplemented    return Permission(self.bits | other.bits)def __repr__(self):    return f"Permission({self.bits:#010b})"

这样就能写read | writeadmin & user,返回仍是Permission实例,可继续参与运算。

  • 返回NotImplemented而非抛异常,让Python尝试调用other.__rand__等回退逻辑,更符合Python惯例
  • & 0xFF截断高位,避免无意义的符号扩展,尤其在右移或取反后
  • 不建议在__and__中直接修改self.bitsreturn self,除非你明确要“可变权限对象”,否则易引发共享状态问题

和内置整数位运算行为一致吗

不完全一致。Python整数的&/|是纯函数式、不可变的,且对任意整数都成立;而你自定义的类只在显式定义了对应方法时才支持,且行为完全由你控制。

  • 整数支持1 & 2 & 4是因为int.__and__返回新int,你的类也要保持同样契约
  • 整数还支持与布尔值运算,比如1 & True1,但你的类若没处理bool类型,就会走到NotImplemented分支,最终报TypeError
  • 如果你的类实例被用于集合或字典键,还需额外实现__hash____eq__,否则无法去重或查找

真正容易被忽略的是:位运算重载不是“语法糖”,它不改变底层数据表示,也不自动提供掩码、测试某位等辅助逻辑——那些都得你自己基于self.bits手动实现。

本文转载于:https://www.php.cn/faq/2324180.html 如有侵犯,请联系zhengruancom@outlook.com删除。
免责声明:正软商城发布此文仅为传递信息,不代表正软商城认同其观点或证实其描述。

热门关注