列表生成式
[1]定义列表
num_list = [i for i in range(6)]
print(num_list) # [0, 1, 2, 3, 4, 5]
[2]对列表中的值进行遍历的处理
# 对列表里的每个元素都进行去除空格的操作.strip(),在赋值给原来的列表
animal_list = [' lion', 'dog ', ' cat ']
animal_list = [i.strip() for i in animal_list]
print(animal_list) # ['lion', 'dog', 'cat']
[3]还可以进行复杂一些的嵌套循环
list_1 = [1, 2, 3]
list_2 = [4, 5, 6]
# 使list_1 和 list_2 列表里的元素两两相乘，并放入列表
list_3 = [i * j for i in list_1 for j in list_2]
print(list_3) # [4, 5, 6, 8, 10, 12, 12, 15, 18]
#还可以在生成的列表中嵌套列表
list_4 = [[i, i * j] for i in list_1 for j in list_2]
print(list_4)
#[[1, 4], [1, 5], [1, 6], [2, 8], [2, 10], [2, 12], [3, 12], [3, 15], [3, 18]]
字典生成式
- 有了列表推导式的概念，字典推导式学起来就非常简单了，语法格式如下：
# {键:值 for 迭代变量 in 可迭代对象 [if 条件表达式]}
dict_1 = {key: value for key in range(3) for value in range(3)}
print(dict_1) # {0: 2, 1: 2, 2: 2}