Python 10 个习惯用法，第二期

Python 使用习惯第一期：Python 10 个习惯用法，看看你都知道不？

今天是 Python 使用习惯第二期：11-20

11 字典默认值

In[1]: d = {'a': 1, 'b': 3}

In[2]: d.get('b', [])  # 存在键 'b'
Out[2]: 3

In[3]: d.get('c', [])  # 不存在键 'c'，返回[]
Out[3]: []

12 lambda 函数

lambda 函数使用方便，主要由入参和返回值组成，被广泛使用在 max, map, reduce, filter 等函数的 key 参数中。

如下，求 x 中绝对值最大的元素，key 函数确定abs(x)作为比较大小的方法：

x = [1, 3, -5]
y = max(x, key=lambda x: abs(x))
print(y) # -5 

13 max

求 x 中绝对值最大的元素，key 函数确定abs(x)作为比较大小的方法：

x = [1, 3, -5]
y = max(x, key=lambda x: abs(x))
print(y) # -5 

14. map

map 函数映射 fun 到容器中每个元素，并返回迭代器 x

x = map(str, [1, 3, 5])
for e in x:
    print(e, type(e))

下面写法不够 Pythoner

for e in [1, 3, 5]:
    print(e, str(e)) # '1','3','5'

15. reduce

reduce 是在 functools 中，第一个参数是函数，其必须含有 2 个参数，最后归约为一个标量。

from functools import reduce
x = [1, 3, 5]
y = reduce(lambda p1, p2: p1*p2, x)
print(y) # 15

下面写法不够 Pythoner:

y = 1
for e in x:
    y *= e
print(y)

16. filter

使用 filter 找到满足 key 函数指定条件的元素，并返回迭代器

如下，使用 filter 找到所有奇数：

x = [1, 2, 3, 5]
odd = filter(lambda e: e % 2, x)
for e in odd:  # 找到奇数
    print(e)

还有另外一种方法，使用列表生成式，直接得到一个odd 容器，

odd = [e for e in x if e % 2]
print(odd) # [1,3,5]

下面写法最不符合 Python 习惯：

odd = []
for e in x:
    if e % 2:
        odd.append(e)
print(odd)  # [1,3,5]

17 sorted

sorted 是内置函数，拿来即用，用于排序，key 参数指定排序使用的方法。

如下 key 参数确定abs(x)作为排序的方法：

x = [4, -3, 1, -6, 2]
y = sorted(x, key=lambda x: abs(x))
print(y) # [1, 2, -3, 4, -6]

18 位置变参

In [4]: def f(*a):
   ...:     return a

In [5]: f(1,2,5) # 可传入多个参数
Out[5]: (1, 2, 5)

19  字典变参

In [6]: def g(**a):
   ...:     return a

In [7]: g(a=1,b=2)
Out[7]: {'a': 1, 'b': 2}

20 习惯 yield

遇到 yield 就中断返回，下次进入直接到 yield 下一句

In [12]: def fg(n):
    ...:     i = 0
    ...:     a,b=0,1
    ...:     while i<n:
    ...:         yield b
    ...:         a,b = b, a+b
    ...:         i+=1

In [13]: for i in fg(10):
    ...:     print(i)
1
1
2
3
5
8
13
21
34
55     

你还可以看：

从 0 学算法第一周总结：1 万字 30 多幅图

长按关注

你点的每个“在看”，我都认真当成了喜欢