Python小技巧

Python小技巧
一、 计算代码执行时间

timeit 模块可以用来估算代码的执行时间：

>>> import timeit
>>> timeit.timeit('"-".join(str(n) for n in range(100))', number=10000)
0.5584494843667756
>>> timeit.timeit('"-".join([str(n) for n in range(100)])',number=10000)
0.48983852808862594
>>> timeit.timeit('"-".join(map(str, range(100)))',number=10000)
0.3763218479738555

函数释义

timeit(stmt='pass', setup='pass', timer=, number=1000000)

返回：返回执行 stmt 这段代码 number 遍所用的时间，单位为秒，float 型
参数：

 - stmt：要执行的那段代码 
 - setup：执行代码的准备工作，不计入时间，一般是import之类的 
 - timer：这个在 win32 下是time.clock()，linux 下是 time.time()，默认的，不用管 
 - number：要执行 stmt 多少遍

二、打印字典
标准的字典打印出来可读性不太友好：

>>> my_dict = {'a': 23, 'b': 42, 'c': 0xc0ffee}
>>> print(my_dict)
{'b': 42, 'c': 12648430, 'a': 23}

我的 JSON 模块更能胜任这个工作：

>>> import json
>>> print(json.dumps(my_dict, indent=4, sort_keys=True))    
{                                                           
    "a": 23,                                                
    "b": 42,                                                
    "c": 12648430                                           
}     

需要注意的是，这里只能是原始数据类型的字典：

>>> json.dumps({all: 'yup'})
TypeError: keys must be a string

三、字典默认值
字典的 get() 方法，及其参数 default

dict.get(key, default=None)

• key – 字典中要查找的键。
• default – 如果指定键的值不存在，则返回该默认值。

>>> dic = {1:'one', 2:'two'}
>>> dic.get(1)
'one'
>>> dic.get(3,'other')
'other'
>>> dic.get(3)
None

在使用字典的 get() 方法时，通过设置 default 可以避免一些异常的出现。
四、字典按值排序
1 使用 lambda 表达式

>>> dic = {'a': 4, 'b': 31, 'c': 2, 'd': 12}
>>> sorted(dic.items())
[('a', 4), ('b', 31), ('c', 2), ('d', 12)]

>>> sorted(dic.items(), key=lambda item:item[1])
[('c', 2), ('a', 4), ('d', 12), ('b', 31)]

首先，items()方法将字典的元素转化为了元组。

另外，lambda 表达式是一行函数，在其他语言中也被称为匿名函数。lambda x:y 中 x 表示输入参数，y 表示 lambda 函数的返回值。

如果写作 key=lambda item:item[0] 的话则是选取第一个元素作为比较对象，也就是 key 值作为比较对象。

2 使用 operator 模块

>>> import operator
>>> sorted(dic.items(), key=operator.itemgetter(1))
[('c', 2), ('a', 4), ('d', 12), ('b', 31)]

operator 模块提供的 itemgetter函数 用于获取对象的哪些维的数据，类似于 lambda 表达式。如：

>>> func = operator.itemgetter(1)
>>> func([1,2,3])
2
>>> func = operator.itemgetter(1,3)
>>> func([1,2,3,4])
(2, 4)
>>> operator.itemgetter('rank')(dict(rank='captain', name='dotterbart'))
'captain'

五、合并字典
如何合并字典？

也许你能说出 32 种合并字典的方法，但你知道如何优雅地合并吗？

# Python 3.5+ 环境

>>> x = {'a': 1, 'b': 2}
>>> y = {'b': 3, 'c': 4}

>>> z = {**x, **y}
>>> z
{'c': 4, 'a': 1, 'b': 3}

>>> a = {**y, **x}
>>> a
{'a': 1, 'c': 4, 'b': 2}

# Python 2.x 环境

>>> z = dict(x, **y)
>>> z
{'a': 1, 'c': 4, 'b': 3}

本例中，Python 按顺序（即 x --> y）将字典的 key 进行了合并，当 key 值重复时，后面的会覆盖前面的。
希望我的文章能够帮助你！