推导式是Python中提供的一个非常方便的功能,可以通过一行代码实现创建list、dict、tuple、set 的同时初始化一些值。
num_list = [ i for i in range(5)] # 列表
print(num_list) # 结果为:[0, 1, 2, 3, 4]
num_set = { i for i in range(5)} # 集合
print(num_set) # 结果为:{0, 1, 2, 3, 4}
num_dict = { i:i for i in range(5)} # 字典
print(num_dict) # 结果为:{0: 0, 1: 1, 2: 2, 3: 3, 4: 4}
data = (i for i in range(5)) # 元组
print(data) # 结果为:<generator object <genexpr> at 0x00000>
for item in data:
print(item)
元组不同于其他类型,不会立即执行内部循环去生成数据,而是得到一个生成器。
嵌套
## 生成一副扑克牌
poker_list = [[color, num] for num in range(1, 14) for color in ["红桃", "黑桃", "方片", "梅花"]]
print(poker_list)
烧脑面试题
def num():
return [lambda x: i * x for i in range(4)]
## 1. 执行num()并获取返回值 [函数,函数,函数,函数] i=3
## 2. for循环返回值
## 3. 返回值的每个元素(2)
result = [m(2) for m in num()] # [6,6,6,6]
def num():
return (lambda x: i * x for i in range(4))
## 1. 执行num()并获取返回值,得到一个生成器对象
## 2. for循环返回值
## 3. 返回值的每个元素(2)
result = [m(2) for m in num()] # [0,2,4,6]
首字母配对
girls = ['Python', 'Java', 'C']
boys = ['Java', 'Python', 'C']
letterGirls = {}
for girl in girls:
letterGirls.setdefault(girl[0], []).append(girl)
print([b + ' + ' + g for b in boys for g in letterGirls[b[0]]])
输出结果如下:
['Java + Java', 'Python + Python', 'C + C']
letterGirls.setdefault(girl[0], [])
是在letterGirls字典中寻找girl[0]元素(即girl变量的首字母),如果没找到,就返回默认值空列表。
那么,letterGirls.setdefault(girl[0], [])
这部分可看作[],即可理解为[].append(girl)。