python推导式(comprehensions)
推导式是可以从一个数据序列构建另一个新的数据序列的结构体。 推导式在Python2和3中都有支持,有下列几种推导式:
- 列表(list)推导式
- 字典(dict)推导式
- 集合(set)推导式
- 生成器(generator)推导式
1. 列表推导式
列表推导式(又称列表解析式)提供了一种简明扼要的方法来创建列表。
它的结构是在一个中括号里包含一个表达式,然后是一个for语句,然后是0个或多个for或者if语句。那个表达式可以是任意的,意思是你可以在列表中放入任意类型的对象。返回结果将是一个新的列表。
multiples = [i for i in range(30) if i % 3 == 0]
print(multiples)
# Output: [0, 3, 6, 9, 12, 15, 18, 21, 24, 27]
当需要使用for循环来生成一个新列表时,列表推导式就会很有效:
squared = [x**2 for x in range(10)]
上述语句等效于:
squared = []
for x in range(10):
squared.append(x**2)
2. 字典推导式
类似于列表推展式,例如:
mcase = {'a': 10, 'b': 34, 'A': 7, 'Z': 3}
mcase_frequency = {
k.lower(): mcase.get(k.lower(), 0) + mcase.get(k.upper(), 0)
for k in mcase.keys()
}
# mcase_frequency == {'a': 17, 'z': 3, 'b': 34}
在上面的例子中我们把同一个字母但不同大小写的值合并起来了。
还可以快速对换一个字典的键和值:
{v: k for k, v in some_dict.items()}
3. 集合推导式
它们跟列表推导式也是类似的。 唯一的区别在于它们使用大括号{}。 举个例子:
squared = {x**2 for x in [1, 1, 2]}
print(squared)
# Output: {1, 4}
4. 生成器推导式
类似于列表推导式,不同的是生成器推导式不给整个列表分配内存,而是一次生成一个,这样的话更节约内存:
multiples_gen = (i for i in range(30) if i % 3 == 0)
print(multiples_gen)
# Output: <generator object at 0x7fdaa8e407d8>
for x in multiples_gen:
print(x)
# Outputs numbers