迭代列表并在这些列表中查找连续的数字

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)]