迭代列表並在這些列表中查找連續的數字

Question

當前我正在嘗試在嵌套列表中查找連續的數字。我的目標是下面的例子：

nested = [ [1,7,13], [2,5,8], [3, 6, 9] ] 

OUTPUT：(1,2,3), (7,8,9)

還應該工作，如果有三個以上的列表。列表的數量可能會有所不同。

（背景爲這個，我是建設學校和搜索引擎在用戶給出的查詢詞的嵌套的位置的數字。）

我用：

def is_coherent(x): 
    return all(np.diff(x) == 1) 

到看看整數是否連續。我也嘗試過itertools.izip。然而，這不允許我改變我想要迭代的列表數量。

任何幫助將不勝感激！

Answer 1

我想你想從內部列表中，其所有項目都是連續的笛卡爾積的項目，對，你可以嘗試這樣的事：

from itertools import izip, product 

def find_continuous_items(data): 
    for p in product(*data): 
     if all(b-a==1 for a, b in izip(p, p[1:])): 
      yield p 

nested = [[1,7,13], [2,5,8], [3, 6, 9]] 
print list(find_continuous_items(nested)) 
#[(1, 2, 3), (7, 8, 9)]