for语句创建列表、字典、集合、生成器
一、列表
1、基本语法
[表达式 for 循环计数器 in 可迭代对象]
举例
a = [ i + i for i in range(0,5) ]
print (a) # 打印 [0, 2, 4, 6, 8]
上面的例子中，对range(0,5)中的每个元素 i 取出来，执行 i+i 之后放入新列表。
2、for+if
for表达式还可以在可迭代对象后面加if条件，这样就可以筛选出符合特定条件的元素放入新列表中。
例：
a = [ i + i for i in range(0,5) if i % 2 == 0]
print (a) # 打印[0, 4, 8]
上面的例子中的 if i % 2 == 0 筛选了range(0,5)中的偶数，只对符合条件的3个元素（0，2，4）进行了迭代。
3、嵌套for
for既然能和if结合，当然也能和for结合，即嵌套：
a = [[i, j] for i in range(1, 3) for j in range(3, 5)]
print(a) # 结果 [[1, 3], [1, 4], [2, 3], [2, 4]]
上述代码等价于:
a = []
for i in range(1, 3):
for j in range(3, 5):
a.append([i, j])
print(a)
当然啦，嵌套for里可以依然可以嵌入循环或者判断，我们再来一层：
a = [[i, j] for i in range(5) for j in range(5) if i == j]
print(a) # 结果[[0, 0], [1, 1], [2, 2], [3, 3], [4, 4]]
到这里，用这种语法创建列表的方式你已经学会了，这三个例子应该彻底打开了你的思路，让你可以放胆去写了。其实和传统方式没啥区别，只不过可以写在一行，看起来相对简洁、优雅~
二、生成器
上面我们把结果和推导式放在了一个方括号里，所以生成的是列表，如果放在圆括号里捏？会不会是元组呢？我们来试一下：
a = ([i, j] for i in range(5) for j in range(5) if i == j)
print(a)
结果却是酱紫的：
原来是一个生成器。
不过我们照样可以用for循环遍历它！
a = ([i, j] for i in range(5) for j in range(5) if i == j)
for item in a:
print(item, end=" ")
# 结果：[0, 0] [1, 1] [2, 2] [3, 3] [4, 4]
如果想要生成集合的话，只需要用内置函数 tuple 转换一下就好了。
三、字典与集合
既然试过了方括号和圆括号，怎么好意思不试试花括号呢？我们都知道，花括号在Python中可以表示两种数据类型：字典&集合。那我们就分别试一下：
a = {i: j for i in range(5) for j in range(5) if i == j}
print(a)
b = {i*j for i in range(5) for j in range(5) if i == j}
print(b)
结果如下：
果不其然，字典和集合均支持这种方式创建~
参考文章
转载：https://blog.csdn.net/qq_43617936/article/details/106171781
查看评论