我有一个 Python 生成器lexg
,它在每次迭代时生成一个列表。该代码似乎在传统的for
-loop 意义上工作,即
for i in lexg(2,2): print(i)
产生:
[2, 0]
[1, 1]
[1, 0]
[0, 2]
[0, 1]
[0, 0]
但似乎打破了列表理解,也就是说,两者
list(lexg(2,2))
和
[i for i in lexg(2,2)]
生产
[[0, 0],
[0, 0],
[0, 0],
[0, 0],
[0, 0],
[0, 0]]
而,我希望list(lexg(2,2))
生产
[[2, 0]
[1, 1]
[1, 0]
[0, 2]
[0, 1]
[0, 0]]
的代码lexg
是:
def lexg( n, d ):
exponent = [0] * n;
def looper( m, totalDegree ):
r = reversed( range( 0, d - totalDegree + 1 ) );
for j in r:
exponent[n-m] = j;
if m == 1:
yield exponent;
else:
for x in looper( m-1, totalDegree+j ): yield x
return looper( n, 0 );
是什么导致空输出?
编辑/解决方案
如下所示,问题在于生成器的每一步都返回相同的列表。因此,一种解决方案是在返回之前复制列表。例如,我已将 的yield exponent;
行更改lexg
为yield list(exponent);
,从而解决了问题。