集合生成式相关示例
一、例如生成集合{1,4,9,16,25}
# for in 循环实现
S = set()
for i in range(1,7):
S.add(i * i)
print(S) # {1, 4, 36, 9, 16, 25}
# 集合生成式
S = { i * i for i in range(1,7)}
print(S) # {1, 4, 36, 9, 16, 25}
二、 嵌套if
# for in 循环嵌套if
S = set()
for i in range(1,7):
if i % 2:
S.add(i * i)
print(S)
# 集合生成式
S = { i * i for i in range(1,7) if i % 2}
print(S)
三、循环嵌套
# for in 循环
S = set()
for i in range(1,4):
for j in range(1,4):
S.add((i,j))
print(S)
# 集合生成式
S = { (i,j) for i in range(1,4) for j in range(1,4)}
print(S)