首页 热点资讯 义务教育 高等教育 出国留学 考研考公
您的当前位置:首页正文

Python拾珍:2. 列表理解

2024-12-16 来源:花图问答

下面的函数接收一个字符串列表,将每个元素通过字符串方法 capitalize 进行映射,并返回一个新的字符串列表:

def capitalize_all(t):
    res = []
    for s in t:
        res.append(s.capitalize())
    return res

我们可以用列表理解(list comprehension)把这个函数写得更紧凑:

def capitalize_all(t):
    # res = []
    # for s in t:
    #     res.append(s.capitalize())
    # return res
    return [s.capitalize() for s in t]

实例2,是一段使用过滤的代码:

def only_upper(t):
    res = []
    for s in t:
        if s.isupper():
            res.append(s)
    return res

我们可以用列表理解将其重写为:

def only_upper(t):
    # res = []
    # for s in t:
    #     if s.isupper():
    #         res.append(s)
    # return res
    return [s for s in t if s.isupper()]
显示全文