Question

我有一个列表如下：

>>> list1 = ['Mary','had','a','little','lamb','which','was','very','naughty']

我需要在list2之外创建另一个列表（list1），该列表是通过迭代list1并找到元素 little 然后创建的将小以及小后面的所有元素附加到list2。

读了一下，发现来自itertools的丢失只是为了我的要求。

>>> from itertools import dropwhile
>>> list2 = list(dropwhile(lambda l: 'little' not in l, list1))
>>> list2
['little', 'lamb', 'which', 'was', 'very', 'naughty']

完全符合我的要求。

但是，现在我还需要指定结尾元素。所以我想要一个从小到非常的元素列表，如下所示：

>>>list3
['little', 'lamb', 'which', 'was', 'very']

我如何用类似的方法做到这一点？

Answer 1

您可以使用自定义函数返回生成器：

def drop(iterable,start,end):

    for i,x in enumerate(iterable):
        if x==start :
            for y in iterable[i:]:
                if y!=end:
                    yield y
                else:
                    yield end
                    break

演示：

list1 = ['Mary','had','a','little','lamb','which','was','very','naughty']
print list(drop(list1,'little','very'))
['little', 'lamb', 'which', 'was', 'very']

注意这个解决方案比使用list.index更快，因为它的顺序是O（n），你使用它2次，但对于前面的函数，你迭代你的所有列表另外还有一次iterable[i:]中的一个索引，它的顺序小于O（n），所以它的速度更快。

为了更好地理解，请查看以下基准测试：

from timeit import timeit

s1="""
list1 = ['Mary','had','a','little','lamb','which','was','very','naughty']
list1[list1.index("little"):list1.index("very")+1]
"""
s2="""
def drop(iterable,start,end):

    for i,x in enumerate(iterable):
        if x==start :
            for y in iterable[i:]:
                if y!=end:
                    yield y
                else:
                    yield end
                    break
list1 = ['Mary','had','a','little','lamb','which','was','very','naughty']
drop(list1,'little','very')
    """


print ' first: ' ,timeit(stmt=s1, number=1000000)
print 'second : ',timeit(stmt=s2, number=1000000)

结果：

 first:  5.87736296654
second :  5.01044201851

Answer 2

不使用itertools这样做的简单方法可能看起来像：

start = list1.index("little")
end = list1.index("very")
list3 = list1[start:end+1]

或者只是你可以把它写成：

list3 = list1[list1.index("little"):list1.index("very")+1]

Answer 3

这应该可以解决问题：

 list1[list1.index('little'):-list1[::-1].index('very')]

结果：

['little', 'lamb', 'which', 'was', 'very']

我们在反转列表中搜索第二个关键字，然后在原始列表中使用否定索引。

Answer 4

如果您希望继续使用itertools（不关心表现）：

from itertools import dropwhile
list1 = ['Mary','had','a','little','lamb','which','was','very','naughty']

list2 = list(dropwhile(lambda x: x!="little", list1))
list3 = list(dropwhile(lambda x: x!="very", reversed(list2)))
print list(reversed(list3))

输出：

['little', 'lamb', 'which', 'was', 'very']

如果你关心性能，你应该定义一个自定义逻辑（感谢@Kasra）：

list1 = ['Mary','had','a','little','lamb','which','was','very','naughty']

def drop(sequence,start,end):
    keep = False
    for item in sequence:
        if item == start:
            keep = True
        if keep and item == end:
            yield item
            break
        if keep:
            yield item

list3 = drop(list1, 'little', 'very')

print list(list3)

Answer 5

不假设“小”在“非常”之前的解决方案：

def list_range(ls, a, b):
    a_i, b_i = ls.index(a), ls.index(b)
    if a_i < b_i:
        return ls[a_i:b_i+1]
    else:
        return ls[a_i:b_i-1:-1]

list1 = ['Mary','had','a','little','lamb','which','was','very','naughty']
print list_range(list1, 'little', 'very')

Answer 6

你可以列出切片。

>>> list1 = ['Mary','had','a','little','lamb','which','was','very','naughty']
>>> list1[list1.index('little'):list1.index('very')+1]
['little', 'lamb', 'which', 'was', 'very']

从现有列表创建较小的列表

6 个答案: