本文旨在收集记录分享简洁漂亮的Python写法,欢迎补充
字典/集合推导式(dict/set comprehension)
类似于列表推导式:
my_dict = {i: i * i for i in xrange(100)}
my_set = {i * 15 for i in xrange(100)}
#这俩在写法上的区别只是字典推导式多了一个':'
Caution:
区别一下三种写法:
>>>my_set={i*15 for i in xrange(100)}
>>>my_list=[i*15 for i in xrange(100)]
>>>my_generator=(i*15 for i in xrange(100))
>>>print "my_set", type(my_set)
my_set <type 'set'>
>>>print "my_list", type(my_list)
my_list <type 'list'>
>>>print "my_generator", type(my_generator)
my_generator <type 'generator'>
枚举 (enumerate)
要在枚举过程中使用序号,可以使用enumerate
优雅地实现:
#i从0开始,每次加1
for i, item in enumerate(iterable):
print i, item
#i从2开始
for i, item in enumerate(iterable, 2):
print i, item
强制浮点数除法
如果我们对两个整数做/
除法,Python 2 返回的是一个整数,如需返回浮点数的结果,在Python 2.x中通常会这么做:
>>>1.0/2
0.5
可以这样更漂亮地实现:
>>>from __future__ import division
>>>1/2
0.5
而在Python 3中,/
是真除,默认精确到小数点后一位。
使用literal_eval()
我们都知道eval:
expr = "[1, 2, 3]"
my_list = eval(expr)
但估计没几个人知道literal_eval( ):
import ast
my_list = ast.literal_eval(expr)
(看起来好像并没有变漂亮呢,但ast.literal_eval()更安全。)
用dir( )查看一个Python对象
>>> foo = [1, 2, 3, 4]
>>> dir(foo)
['__add__', '__class__', '__contains__',
'__delattr__', '__delitem__', '__delslice__', ... ,
'extend', 'index', 'insert', 'pop', 'remove',
'reverse', 'sort']
调试脚本
利用pdb模块可以轻松地在脚本中设置断点:
import pdb
pdb.set_trace()
你可以在脚本的任何位置加上pdb.set_trace()来设置断点。
简化if 结构
当有多个值需要检验的时候,通常会这么写:
if n==1 or n==4 or n==5 or n==6:
更漂亮的写法是:
if n in [1,4,5,6]:
反转列表/字符串
你可以这样快速地反转一个列表:
>>>a=[1,2,3]
>>>a[::-1]
[3,2,1]
这样会生成一个新的列表,原列表不会变。如果你想在原地反转列表,可以这样:
>>> a.reverse()
>>> a
[3, 2, 1]
字符串类型string是不可变类型,因此也就没有reverse( )咯。
行内if 声明
a=1 if b>c else -1
print a if b>c else d
交换两个变量
a, b=b, a
连续赋值
a, b, c=1, 2, 3
串联
>>>a=[1,2]
>>>b=[3,4]
>>>a+b
[1,2,3,4]
>>>print str(1) + " a"
1 a
>>>print `1` + " b"
1 b
>>>print 1, " c"
1 c
地板除 (Python 2.2以后)
>>>5//2
2
>>>5.0//2
2.0
乘方
#a**b, a的b次方
>>>2**5
32
级联比较
if a<b<c:
blabla...
if a<b>c:
blabla...
同时遍历两个列表:
names=['Tom','Jerry']
ages=[23,17]
for name, age in zip(names, ages):
print name, age
建立并初始化列表
#一维
a=[0]*4
#二维, 3行4列
b=[[0]*4 for i in range(3)]
join( )
>>>a=['my', 'life', 'is', 'awesome']
>>>print '!'.join(a)
my!life!is!awesome
获取字典内的元素
#bad implementation:
dix={'Tom':23}
try:
name=dix['Tom']
except KeyError:
name='Default'
#good implementation:
name=dix.get('Tom', 'Default')
截取列表的一部分
>>>a = [1,2,3,4,5,6]
>>>a[:3]
[1,2,3]
>>>a[1:5]
[2,3,4,5]
>>>a[-3:]
[4,5,6]
>>>a[::2]
[1,3,5]
>>>a[1::2]
[2,4,6]
这里的a[x:y:z], x和y分别是下标起点,下标终点+1,z是间隔值。于是就能理解为啥a[::-1]能得到一个反转的列表了。
向列表添加元素
my_list=[1,2,3]
my_list+=4,
#my_list变成了[1,2,3,4]
my_list+=[5]
#my_list变成了[1,2,3,4,5]
参考: