Python列表推导式

卧底西瓜

已于 2022-07-31 14:32:33 修改

阅读量484

点赞数

文章标签： python

于 2022-07-31 14:29:50 首次发布

本文链接：https://blog.csdn.net/weixin_43192888/article/details/126075050

版权

需求一：找出原列表中大于3的元素组成新的列表

实现思路：通过循环获取列表中的元素后做判断，满足条件的添加至新列表

"""
需求：找出原列表中大于3的元素组成新的列表
"""
old_list = [1, 2, 3, 4, 5]
new_list = []
for element in old_list:
    if element >3:
        new_list.append(element)
print(new_list)

通过列表推导式简化代码

old_list = [1, 2, 3, 4, 5]
new_list = [element for element in old_list if element>3]
"""
new_list: 新列表
element：新列表中存放的元素
for element in old_list：循环获取原列表中的元素
if element>3：循环获取到的元素需满足的条件
"""

需求二：找出原列表中大于3的元素并做+1处理后组成新的列表

实现思路：通过循环获取列表中满足大于3的元素后做+1操作，并将其添加至新列表

"""
需求：找出原列表中大于3的元素并做+1处理后组成新的列表
"""
old_list = [1, 2, 3, 4, 5]
new_list = []
for element in old_list:
    if element >3:
        element +=1
        new_list.append(element)
print(new_list)

通过列表推导式简化代码

old_list = [1, 2, 3, 4, 5]
new_list = [element+1 for element in old_list if element > 3]
"""
new_list: 新列表
element+1：从原列表中循环获取满足条件的元素后+1
for element in old_list：循环获取原列表中的元素
if element>3：循环获取到的元素需满足的条件
"""

需求三：将原列表中的元素作为参数传给函数后，将函数的返回值组成一个新列表

old_list = [1, 2, 3, 4, 5]
def func(data):
    data = str(data) + '修改'
    return data


new_list = [func(element) for element in old_list]
print(new_list)
"""
new_list: 新列表
func(element)：从原列表中循环获取元素后传入函数后获得的返回值
for element in old_list：循环获取原列表中的元素
if element>3：获取到的元素需满足的条件
"""

列表推导式： [表达式 for 变量 in 列表 if 条件]

表达式：生成列表中的元素，该表达式只要满足语法规则都可

for 变量 in 列表：循环列表，往表达式中传递变量；若表达式中未用到原列表中的变量，且在无if条件下，for循环的次数就是新列表中表达式的个数
old_list = [2, 3, 4, 5, 6]
new_list = [old_list[2:] for i in old_list]
if 条件：条件语句，可以过滤列表中不符合条件的值