Python列表和字典推导及列表推导式和生成器推导式

什么是列表和字典推导？

List Comprehension 或 Dict Comprehension

比如 [i for i in range(10) if i % 2 == 0]

一种快速生成list/dict/set的方式。 用来替代map/filter等

(i for i in range(10) if i % 2 == 0) 返回生成器

把方括号改成圆括号就会返回生成器从而去大大节省内存

代码一：

a = ['a', 'b', 'c']
b = [4, 5, 6]

# 要求将上面两个列表合成d = {'a':4, 'b':5, 'c':6}

# 方法一：普通方法
d = {}
for i in range(len(a)):
	d[a[i]] = b[i]

print(d)  # 输出结果:{'a': 4, 'b': 5, 'c': 6}

# 方法二：使用字典推导
# 使用zip函数成对迭代a和b
d = {k: v for k,v in zip(a, b)}
print(d) # 输出结果:{'a': 4, 'b': 5, 'c': 6}

代码二：

# 列表推导式
l = [i for i in range(10)]
print(l) # 输出结果:[0, 1, 2, 3, 4, 5, 6, 7, 8, 9]

# 生成器推导式
l1 = (i for i in range(10))
print(type(l1)) # 输出结果：<class 'generator'>

for i in l1:
	print(i) 
# 输出结果： 
0
1
2
3
4
5
6
7
8
9