在元组中查找元素并过滤

Question

假设我有一个元组，内容是：

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

我想从这个列表中筛选出以'img'开头的元素，生成一个新的列表。所以我新的列表会是这样的：

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

谢谢！

Answer 1

这是一种使用 生成器表达式 的方法（在处理大列表时可能会有一些不同的效果）

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

Answer 2

也许可以这样做？

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

Answer 3

一种方法：

>>> 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.的回答，里面有相关的技巧。它的基本语法是 [{表达式} for {项变量} in {可迭代对象} if {布尔表达式}]。这其实和下面的写法是一样的：

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

这里的 if {布尔表达式} 是可选的，和 for 循环中的用法一样。

第二种方法就是内置的 filter 函数，它接受一个测试函数和一个可迭代对象，然后返回一个包含所有“通过”测试函数的元素的列表。如果你之前没见过 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