python - 在列表中查找子列表的开始和结束索引

Question

我有一个清单：

greeting = ['hello','my','name','is','bob','how','are','you']

我想定义一个函数来查找此列表中子列表的第一个和最后一个索引。因此：

find_sub_list(['my','name','is'], greeting)

应该返回：

1, 3

建议？

Answer 1

如果您想要多个匹配项，则可以使用：

greeting = ['hello','my','name','is','bob','how','are','you','my','name','is']

def find_sub_list(sl,l):
    results=[]
    sll=len(sl)
    for ind in (i for i,e in enumerate(l) if e==sl[0]):
        if l[ind:ind+sll]==sl:
            results.append((ind,ind+sll-1))

    return results

print find_sub_list(['my','name','is'], greeting) 
# [(1, 3), (8, 10)]

或者，如果您只想要第一场比赛：

greeting = ['hello','my','name','is','bob','how','are','you','my','name','is']

def find_sub_list(sl,l):
    sll=len(sl)
    for ind in (i for i,e in enumerate(l) if e==sl[0]):
        if l[ind:ind+sll]==sl:
            return ind,ind+sll-1

print find_sub_list(['my','name','is'], greeting)    
# (1, 3)

Answer 2

切片列表：

>>> greeting[0:3]
['hello', 'my', 'name']
>>> greeting[1:4]
['my', 'name', 'is']
>>> greeting[1:4] == ['my','name','is']
True

这应该让你开始：

for n in range(len(greeting) - len(sub_list) + 1):
    ...

Answer 3

如果您确定您的列表将始终在您的子列表中，您可以这样做：

def find_sub_list(sub_list,this_list):
    return (this_list.index(sub_list[0]),len(sub_list))

如果要检查列表中存在的子列表中的项目，请使用：

def find_sub_list(sub_list,this_list):
    if set(sub_list).issubset(set(this_list)): 
        return(this_list.index(sub_list[0]),len(sub_list))
    else:
        return False

最后，如果 sub_list 中项目的顺序也是未知的，那么使用这个：

def find_sub_list(sub_list,this_list):
    if sub_list[0] in this_list:
        for i,item in enumerate(sub_list[1:]):
            if item not in this_list[this_list.index(sub_list[i]):]:
                return False
        return(this_list.index(sub_list[0]),len(sub_list))

现在，项目必须按正确的顺序排列，函数才不会返回 false。

Answer 4

如果只返回第一个和最后一个条目的索引，则以下是一个解决方案：

def find_sub_list(subl, l):
    ind_subl = [i for i in range(len(l)) if l[i] in subl]
    return [ind_subl[0], ind_subl[-1]]

print find_sub_list(['my', 'name', 'is'], greeting)
# [1, 3]