發電機本身的產品

Question

我需要迭代發電機與自身的乘積，不包括對角線。我試圖用itertools.tee消耗相同的發電機兩次

def pairs_exclude_diagonal(it): 
    i1, i2 = itertools.tee(it, 2) 
    for x in i1: 
     for y in i2: 
      if x != y: 
       yield (x, y) 

這不起作用

In [1]: for (x, y) in pairs_exclude_diagonal(range(3)): 
    ...:  print(x, y) 
0 1 
0 2 

的documentation for tee狀態：從單一的迭代

返回n個獨立的迭代器。

這樣做的正確方法是什麼？

（我使用python3.6.1）

Answer 1

的問題是，你想重用i2迭代器。迭代一次後，它已經耗盡，所以你將無法再次迭代它。當你嘗試時，它什麼都沒有。

我想，而不是tee（這是不是爲了這個目的反正非常有效的），你應該使用itertools.product生成所有對（過濾掉你想跳過那些前）：

def pairs_exclude_diagonal(it): 
    for x, y in itertools.product(it, repeat=2): 
     if x != y: 
      yield (x, y) 

Answer 2

它看起來就像你想用itertools.permutations一樣。

In [1]: import itertools 

In [2]: for x, y in itertools.permutations(range(3), 2): 
    ...:  print(x, y) 
    ...:  
0 1 
0 2 
1 0 
1 2 
2 0 
2 1 

如果你真的想用tee它做，你將不得不把第二迭代爲list，使其不通過外for循環耗盡在第二時間：

In [14]: def pairs_exclude_diagonal(it): 
    ...:  i1, i2 = itertools.tee(it, 2) 
    ...:  l2 = list(i2) 
    ...:  for x in i1: 
    ...:   for y in l2: 
    ...:    if x != y: 
    ...:     yield (x, y) 
    ...:     

In [15]: for (x, y) in pairs_exclude_diagonal(range(3)): 
    ...:  print(x, y) 
    ...:  
0 1 
0 2 
1 0 
1 2 
2 0 
2 1 

請注意，這是非常沒有意義的，因爲在迭代器上調用list將它加載到內存中並且首先破壞了具有迭代器的目的。