Python--推导式（列表、字典、集合、元祖）

1、列表推导式

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

[out_exp_res for out_exp in input_list if condition]

• out_exp_res：列表生成元素表达式，可以是有返回值的函数。
• for out_exp in input_list：迭代 input_list 将 out_exp 传入到 out_exp_res 表达式中。
• if condition：条件语句，可以过滤列表中不符合条件的值。

eg：

计算 30 以内可以被 3 整除的整数：

multiples = [i for i in range(30) if i % 3 == 0]
print(multiples)
# [0, 3, 6, 9, 12, 15, 18, 21, 24, 27]

计算列表中大于2的数字的平方：

list1 = [1, 2, 3, 4]
list2 = [i * i for i in list1 if i > 2]
print(list2)
# [9, 16]

把字典中所有的key打印出来：

my_json = {"三三": 10,"四四": 20,"六六": 30}
list4 = [key for key, value in my_json.items()]
print(list4)
# ['三三', '四四', '六六']

把key对应的值打印出来：

my_json = {"三三": 10,"四四": 20,"六六": 30}
list5 = [value for key, value in my_json.items() if key == "三三"]  # 把key1的值打印出来
print(list5)
# [10]

2、字典推导式

{ key_expr: value_expr for value in collection if condition }

eg：

提供三个数字，以三个数字为键，三个数字的平方为值来创建字典：

dic = {x: x ** 2 for x in (2, 4, 6)}
print(dic)
# {2: 4, 4: 16, 6: 36}

3、集合推导式

{ expression for item in Sequence if conditional }

eg：

计算数字 1,2,3 的平方数：

setnew = {i ** 2 for i in (1, 2, 3)}
print(setnew)
# {1, 4, 9}

4、元祖推导式（生成器表达式）

(expression for item in Sequence )

元组推导式可以利用 range 区间、元组、列表、字典和集合等数据类型，快速生成一个满足指定需求的元组。

元组推导式和列表推导式的用法也完全相同，只是元组推导式是用 () 圆括号将各部分括起来，而列表推导式用的是中括号 []，另外元组推导式返回的结果是一个生成器对象。

eg：

生成一个包含数字 1~9 的元组:

a = (x for x in range(1, 10))
print(tuple(a))  #生成的a是一个生成器对象，需要用tuple函数转换成元祖
# (1, 2, 3, 4, 5, 6, 7, 8, 9)