如何在列表中找到匹配的元素？

Question

如何返回列表中匹配的元素？ 例如：

def search_closet(items, colour):
    >>> search_closet(['red summer jacket', 'red shoes', 'green hat'], 'red')
    ['red summer jacket', 'red shoes']
    >>> search_closet(['red shirt', 'green pants'], 'blue')
    []
    >>> search_closet([], 'mauve')
    []

我试过这个身体功能，但没用：

^{pr2}$

Answer 1

一个list comprehension应该这样做。在

def search_closet(items, colour):
    return [i for i in items if colour in i] 

演示：

^{pr2}$

输出：

['red summer jacket', 'red shoes']
[]
[] 

Answer 2

可以使用Python的filter方法匹配要查找的子字符串：

filter(lambda x: "red" in x, ['red summer jacket', 'red shoes', 'green hat'])

过滤器文件：

https://docs.python.org/3/library/functions.html#filter

输出：

^{pr2}$

把它放到一个方法中：

def search_closet(items, colour):
    return filter(lambda x: colour in x, items)