首页 > 代码库 > Python的优雅写法

Python的优雅写法

枚举

之前我们这样操作:

 
 
 
 
 
Python
 
1
2
3
4
i = 0
for item in iterable:
    print i, item
    i += 1

现在我们这样操作:

 
 
 
 
 
Python
 
1
2
for i, item in enumerate(iterable):
    print i, item

enumerate函数还可以接收第二个参数。就像下面这样:

 
 
 
 
 
Python
 
1
2
3
4
5
>>> list(enumerate(‘abc‘))
[(0, ‘a‘), (1, ‘b‘), (2, ‘c‘)]
 
>>> list(enumerate(‘abc‘, 1))
[(1, ‘a‘), (2, ‘b‘), (3, ‘c‘)]

字典/集合 解析

你也许知道如何进行列表解析,但是可能不知道字典/集合解析。它们简单易用且高效。就像下面这个例子:

 
 
 
 
 
Python
 
1
2
3
4
5
6
my_dict = {i: i * i for i in xrange(100)}
my_set = {i * 15 for i in xrange(100)}
 
# There is only a difference of ‘:‘ in both
 
# 两者的区别在于字典推导中有冒号

强制浮点除法

如果我们除以一个整数,即使结果是一个浮点数,Python 2(校注,这里我添上了版本号)依旧会给我们一个整数。为了规避这个问题,我们需要这样做:

 
 
 
 
 
Python
 
1
result = 1.0/2

但是现在有一种别的方法可以解决这个问题,甚至在之前我都没有意识到有这种方法存在。你可以进行如下操作:

 
 
 
 
 
Python
 
1
2
3
4
from __future__ import division
result = 1/2
# print(result)
# 0.5

瞧,现在你不需要在数据上附件“.0” 来获得准确答案了。需要注意的是这个窍门只适用于Python 2。在Python 3 中就不需要进行import 操作了,因为它已经默认进行import了。

简单服务器

你是否想要快速方便的共享某个目录下的文件呢?你可以这么做:

 
 
 
 
 
Python
 
1
2
3
4
5
# Python2
python -m SimpleHTTPServer
 
# Python 3
python3 -m http.server

这样会为启动一个服务器。

对Python表达式求值

我们都知道eval函数,但是我们知道literal_eval函数么?也许很多人都不知道吧。可以用这种操作:

 
 
 
 
 
Python
 
1
2
import ast
my_list = ast.literal_eval(expr)

来代替以下这种操作:

 
 
 
 
 
Python
 
1
2
expr = "[1, 2, 3]"
my_list = eval(expr)

我相信对于大多数人来说这种形式是第一次看见,但是实际上这个在Python中已经存在很长时间了。

脚本分析

你可以很容易的通过运行以下代码进行脚本分析:

 
 
 
 
 
Python
 
1
python -m cProfile my_script.py

对象自检

在Python 中你可以通过dir() 函数来检查对象。正如下面这个例子:

 
 
 
 
 
Python
 
1
2
3
4
5
6
>>> foo = [1, 2, 3, 4]
>>> dir(foo)
[‘__add__‘, ‘__class__‘, ‘__contains__‘,
‘__delattr__‘, ‘__delitem__‘, ‘__delslice__‘, ... ,
‘extend‘, ‘index‘, ‘insert‘, ‘pop‘, ‘remove‘,
‘reverse‘, ‘sort‘]

调试脚本

你可以很方便的通过pdb模块在你的脚本中设置断点。正如下面这个例子:

 
 
 
 
 
Python
 
1
2
import pdb
pdb.set_trace()

你可以在脚本的任何地方加入pdb.set_trace(),该函数会在那个位置设置一个断点。超级方便。你应该多阅读pdb 函数的相关内容,因为在它里面还有很多鲜为人知的功能。

if 结构简化

如果你需要检查几个数值你可以用以下方法:

 
 
 
 
 
Python
 
1
if n in [1,4,5,6]:

来替代下面这个方式:

 
 
 
 
 
Python
 
1
if n==1 or n==4 or n==5 or n==6:

字符串/数列 逆序

你可以用以下方法快速逆序排列数列:

 
 
 
 
 
Python
 
1
2
3
4
5
6
7
8
>>> a = [1,2,3,4]
>>> a[::-1]
[4, 3, 2, 1]
 
# This creates a new reversed list.
# If you want to reverse a list in place you can do:
 
a.reverse()

这总方式也同样适用于字符串的逆序:

 
 
 
 
 
Python
 
1
2
3
>>> foo = "yasoob"
>>> foo[::-1]
‘boosay‘

优美地打印

你可以通过以下方式对字典和数列进行优美地打印:

 
 
 
 
 
Python
 
1
2
from pprint import pprint
pprint(my_dict)

这种方式对于字典打印更加高效。此外,如果你想要漂亮的将文件中的json文档打印出来,你可以用以下这种方式:

 
 
 
 
 
Python
 
1
cat file.json | python -m json.tools

三元运算

三元运算是if-else 语句的快捷操作,也被称为条件运算。这里有几个例子可以供你参考,它们可以让你的代码更加紧凑,更加美观。

 
 
 
 
 
Python
 
1
2
3
[on_true] if [expression] else [on_false]
x, y = 50, 25
small = x if x < y else y

Python的优雅写法