Python生成器对象：__sizeof __（）

Question

这可能是一个愚蠢的问题，但是我还是会问。我有一个生成器对象：

>>> def gen():
...     for i in range(10):
...         yield i
...         
>>> obj=gen()

我可以测量它的大小：

>>> obj.__sizeof__()
24

据说发电机会被消耗掉：

>>> for i in obj:
...     print i
...     
0
1
2
3
4
5
6
7
8
9
>>> obj.__sizeof__()
24

…但obj.__sizeof__()保持不变。

使用字符串可以按我的预期工作：

>>> 'longstring'.__sizeof__()
34
>>> 'str'.__sizeof__()
27

如果有人能启发我，我将不胜感激。

Answer 1

__sizeof__()不按照您的想法去做。该方法返回给定对象的内部大小（以字节为单位），而不是生成器将要返回的项目数。

Python无法事先知道生成器的大小。以下面的无限生成器为例（例如，有更好的方法来创建计数器）：

def count():
    count = 0
    while True:
        yield count
        count += 1

那个发电机是无止境的。没有可分配的大小。但是生成器对象本身占用内存：

>>> count.__sizeof__()
88

通常，__sizeof__()您通常不会调用该sys.getsizeof()函数，而这会增加垃圾回收器的开销。

如果您 知道 生成器将是有限的，并且 必须 知道它会返回多少个项目，请使用：

sum(1 for item in generator)

但请注意，这会使发电机 耗尽 。