推导式
推导式包括:1.列表(list)推导式 2.字典(dict)推导式 3.集合(set)推导式。
1.列表推导式
列表推导式(list comprehension)是利用其它列表创建新列表的一种方式,
工作原理类似for循环,即可对得到的元素进行转换或者筛选。
列表推导式的基本格式:
变量名=[表达式 for 变量 in 列表] 或者
变量名= [表达式 for 变量 in 列表 if 条件]
**练习:**将lst中每一个元素进行平方后放入到一个新列表中
lst = [1, 2, 3, 4, 5, 6, 7, 8]
第一种方式:
lst_new = [i * i for i in lst]
print(lst_new) # [1, 4, 9, 16, 25, 36, 49, 64]
第二种方式:使用 函数
def square(num):
return num * num
new_lst = [square(i) for i in lst]
print(new_lst)
练习:求出lst 中是奇数的值,然后放入到一个新列表中
lst = [1, 2, 3, 4, 5, 6, 7, 8]
lst1 = [i for i in lst if i % 2 == 1]
print(lst1) # [1, 3, 5, 7]