Python推导式

编程 · 2023-08-17 · 183 人浏览

推导式是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)。

Python
Theme Jasmine by Kent Liao