如何让Python的range()接受自定义类作为参数?
让自定义类被range()接受的解决办法
range()并不认__int__方法,它判断参数是否符合要求时,会检查对象是否实现了__index__()这个特殊方法——这是Python专门用来获取对象整数索引值的方法,range、slice这类内置结构都会依赖它。
修改你的MyInteger类,添加__index__方法即可:
class MyInteger: def __init__(self, a:int): self._a = a def __int__(self): return self._a def __index__(self): # 直接返回内部的整数值,或者调用__int__的结果都可以 return self._a # 也可以写成 return int(self)
现在再运行测试代码就正常了:
n = MyInteger(5) for i in range(n): print(i)
输出会是0、1、2、3、4。
补充说明:__int__方法的作用是将对象转换为int类型实例,而range需要的是一个能被直接解析为整数的"索引兼容"对象,所以必须实现__index__才能被range直接接受。
内容的提问来源于stack exchange,提问作者Hotege
相关产品推荐
相关产品推荐

