如何以len(...)
喜欢的方式将__len__(...)
添加到现有对象?在
用例是在第一次迭代之后以不需要在迭代站点之间编写/更改代码的方式记住生成器的长度。在
以下是我目前所掌握的情况:
def generator2iterator(generator):
class Iterator(object):
def __iter__(self):
i = 0
for e in generator():
i += 1
yield e
#self.__len__ = types.MethodType(lambda self: i, self)
self.__len__ = (lambda self: i).__get__(self)
return Iterator()
def test_generator():
yield 3
iterator = generator2iterator(test_generator)
print(list(iterator))
# [3]
print(iterator.__len__())
# 1
print(len(iterator))
# TypeError: object of type 'Foo' has no len()
定义一个包含长度的属性,并为
__len__
返回长度。在相关问题 更多 >
编程相关推荐