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

时间:2013-07-25 23:36:49

标签: python list search sublist

我有一个清单:

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

我想定义一个函数,它将在此列表中找到子列表的第一个和最后一个索引。因此:

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

应该返回:

1, 3

建议?

4 个答案:

答案 0 :(得分:8)

如果你想要多个匹配,这可行:

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)

答案 1 :(得分:0)

切片列表:

>>> 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):
    ...

答案 2 :(得分:-1)

如果您确定您的列表将始终位于您的子列表中,您可以这样做:

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。

答案 3 :(得分:-1)

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

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]