查找元组中的元素并过滤它们

时间:2011-08-17 22:20:37

标签: python

假设我有一个像元组那样的元组:

[('text-1','xxx'), ('img-1','iii'), ('img-2','jjj'), ('text-2','xxx')]

我想过滤掉列表并生成一个包含以'img'开头的元素的新列表。所以我的新列表将如下所示:

[('img-1','iii'), ('img-2','jjj')]

谢谢!

5 个答案:

答案 0 :(得分:27)

一种方式:

>>> l = [('text-1','xxx'), ('img-1','iii'), ('img-2','jjj'), ('text-2','xxx')]
>>> [t for t in l if t[0].startswith('img')]
[('img-1', 'iii'), ('img-2', 'jjj')]

另一种方式:

>>> filter(lambda x: x[0].startswith('img'), l)
[('img-1', 'iii'), ('img-2', 'jjj')]

第一种称为列表理解。有关相关技术,请参阅F.C.'s answer。基本语法是[{expression} for {item_var_or_vars} in {iterable} if {boolean_expression}]。它在语义上等同于这样的东西:

new_list = []
for {item_var_or_vars} in {iterable}:
    if {boolean_expression}:
        new_list.append({expression})

if {boolean_expression}位是可选的,就像在for循环中一样。

第二个是内置函数filter,它接受​​一个测试函数和一个iterable,并返回一个包含“传递”测试函数的每个元素的列表。 lambda,如果您之前没有看过它,只是定义函数的快捷方法。你可以这样做:

def keep_this_element(element):
    return element[0].startswith('img')   # returns True for ('img...', '...')

new_list = filter(keep_this_element, l)   # keeps only elements that return True

答案 1 :(得分:8)

或许这样的事情?

l2 = [item for item in l1 if item[0].startswith('img')]

答案 2 :(得分:5)

使用generator expression的另一种方式(它可能会对大型列表产生一些影响)

>>> l = [('text-1','xxx'), ('img-1','iii'), ('img-2','jjj'), ('text-2','xxx')]
>>> (x for x in l if x[0].find('img') == 0)
0: <generator object <genexpr> at 0x917a00>
>>> gen = (x for x in l if x[0].find('img') == 0)
>>> list(gen)
1: [('img-1', 'iii'), ('img-2', 'jjj')]

答案 3 :(得分:1)

tuple_filter = lambda t, i, w: filter(lambda a: a[i].startswith(w), t)
newtuple = tuple_filter(thetuple, 0, 'img')

答案 4 :(得分:1)

具有正常循环

input = [('text-1','xxx'), ('img-1','iii'), ('img-2','jjj'), ('text-2','xxx')]
inp = []
for i in input:
    if 'img' in i[0]:
        inp.append(i)
print(inp)

或使用lambda函数

inp = list(filter(lambda x: 'img' in x[0],input))
 print(inp)

或通过使用列表理解

 inp = [i for i in input if 'img' in i[0]]
 print(inp)