Python 程序员需要知道的 30 个技巧

Posted 2021-07-17 修罗神天道

如果你让一个 Python 程序员说一下 Python 的优势，他会说简洁以及高可读是最有影响力的优势。为证明上述两点，本文聊聊许多基本的 Python 建议和技巧。



我们从开始使用 Python 便收集这些有用的捷径（贴士与技巧）。分享一些我们知道，同时又能造福于人的知识，有什么事情比这更棒吗？



过去我们分享过 一些给初学者的 Python 编程贴士 旨在优化代码并且减少编码工作，我们的读者仍旧很乐意阅读它。



所以今天我们带来另一些基本的 Python 贴士与技巧，所有的这些技巧都能帮助你压缩代码并且优化运行。此外，你可以在日常工作中很容易地在真实项目中使用他们。



每一个技巧都有一个示例并且给出简短的解释，如果要测试这些代码片段，你可以看一下这些 在线虚拟 Python 运行终端。



最近我们发布的另两个必看的 Python 资源：



9 中优化 Python 代码的主要方式

发现在编程时应该避免的最常见的错误



使用下面的 TOC 来快速地浏览这些 Python 贴士与技巧。



给程序员的 30 个基本 Python 贴士与技巧



贴士#1. 原地交换两个数字



Python 提供了一个直观的在一行代码中赋值与交换（变量值）的方法，请参见下面的示例：



x, y = 10, 20

print(x, y)

 

x, y = y, x

print(x, y)

 

#1 (10, 20)

#2 (20, 10)



赋值的右侧形成了一个新的元组，左侧立即解析（unpack）那个（未被引用的）元组到变量 <a> 和 <b>。



一旦赋值完成，新的元组变成了未被引用状态并且被标记为可被垃圾回收，最终也完成了变量的交换。



贴士#2. 链状比较操作符



比较操作符的聚合是另一个有时很方便的技巧：



n = 10

result = 1 < n < 20

print(result)

 

# True

 

result = 1 > n <= 9

print(result)

 

# False



贴士#3. 使用三元操作符来进行条件赋值



三元操作符是 if-else 语句也就是条件操作符的一个快捷方式：



[表达式为真的返回值] if [表达式] else [表达式为假的返回值]



这里给出几个你可以用来使代码紧凑简洁的例子。下面的语句是说“如果 y 是 9，给 x 赋值 10，不然赋值为 20”。如果需要的话我们也可以延长这条操作链。



x = 10 if (y == 9) else 20



同样地，我们可以对类做这种操作：



x = (classA if y == 1 else classB)(param1, param2)



在上面的例子里 classA 与 classB 是两个类，其中一个类的构造函数会被调用。



下面是另一个多个条件表达式链接起来用以计算最小值的例子：



def small(a, b, c):

    return a if a <= b and a <= c else (b if b <= a and b <= c else c)

 

print(small(1, 0, 1))

print(small(1, 2, 2))

print(small(2, 2, 3))

print(small(5, 4, 3))

 

#Output

#0 #1 #2 #3



我们甚至可以在列表推导中使用三元运算符：



[m**2 if m > 10 else m**4 for m in range(50)]

 

#=> [0, 1, 16, 81, 256, 625, 1296, 2401, 4096, 6561, 10000, 121, 144, 169, 196, 225, 256, 289, 324, 361, 400, 441, 484, 529, 576, 625, 676, 729, 784, 841, 900, 961, 1024, 1089, 1156, 1225, 1296, 1369, 1444, 1521, 1600, 1681, 1764, 1849, 1936, 2025, 2116, 2209, 2304, 2401]



贴士#4. 多行字符串



基本的方式是使用源于 C 语言的反斜杠：



multiStr = "select * from multi_row

where row_id < 5"

print(multiStr)

 

# select * from multi_row where row_id < 5



另一个技巧是使用三引号：



multiStr = """select * from multi_row

where row_id < 5"""

print(multiStr)

 

#select * from multi_row

#where row_id < 5



上面方法共有的问题是缺少合适的缩进，如果我们尝试缩进会在字符串中插入空格。所以最后的解决方案是将字符串分为多行并且将整个字符串包含在括号中：



multiStr= ("select * from multi_row "

"where row_id < 5 "

"order by age")

print(multiStr)

 

#select * from multi_row where row_id < 5 order by age



贴士#5. 存储列表元素到新的变量中



我们可以使用列表来初始化多个变量，在解析列表时，变量的数目不应该超过列表中的元素个数：【译者注：元素个数与列表长度应该严格相同，不然会报错】



testList = [1,2,3]

x, y, z = testList

 

print(x, y, z)

 

#-> 1 2 3



贴士#6. 打印引入模块的文件路径



如果你想知道引用到代码中模块的绝对路径，可以使用下面的技巧：



import threading

import socket

 

print(threading)

print(socket)

 

#1- <module 'threading' from '/usr/lib/python2.7/threading.py'>

#2- <module 'socket' from '/usr/lib/python2.7/socket.py'>



贴士#7. 交互环境下的 “_” 操作符



这是一个我们大多数人不知道的有用特性，在 Python 控制台，不论何时我们测试一个表达式或者调用一个方法，结果都会分配给一个临时变量： _（一个下划线）。



>>> 2 + 1

3

>>> _

3

>>> print _

3



“_” 是上一个执行的表达式的输出。



贴士#8. 字典/集合推导



与我们使用的列表推导相似，我们也可以使用字典/集合推导，它们使用起来简单且有效，下面是一个例子：



testDict = {i: i * i for i in xrange(10)}

testSet = {i * 2 for i in xrange(10)}

 

print(testSet)

print(testDict)

 

#set([0, 2, 4, 6, 8, 10, 12, 14, 16, 18])

#{0: 0, 1: 1, 2: 4, 3: 9, 4: 16, 5: 25, 6: 36, 7: 49, 8: 64, 9: 81}



注：两个语句中只有一个 <:> 的不同，另，在 Python3 中运行上述代码时，将 <xrange> 改为 <range>。



贴士#9. 调试脚本



我们可以在 <pdb> 模块的帮助下在 Python 脚本中设置断点，下面是一个例子：



import pdb

pdb.set_trace()



我们可以在脚本中任何位置指定 <pdb.set_trace()> 并且在那里设置一个断点，相当简便。



贴士#10. 开启文件分享



Python 允许运行一个 HTTP 服务器来从根路径共享文件，下面是开启服务器的命令：



# Python 2



python -m SimpleHTTPServer



# Python 3



python3 -m http.server



上面的命令会在默认端口也就是 8000 开启一个服务器，你可以将一个自定义的端口号以最后一个参数的方式传递到上面的命令中。



贴士#11. 检查 Python 中的对象



我们可以通过调用 dir() 方法来检查 Python 中的对象，下面是一个简单的例子：



test = [1, 3, 5, 7]

print( dir(test) )



['__add__', '__class__', '__contains__', '__delattr__', '__delitem__', '__delslice__', '__doc__', '__eq__', '__format__', '__ge__', '__getattribute__', '__getitem__', '__getslice__', '__gt__', '__hash__', '__iadd__', '__imul__', '__init__', '__iter__', '__le__', '__len__', '__lt__', '__mul__', '__ne__', '__new__', '__reduce__', '__reduce_ex__', '__repr__', '__reversed__', '__rmul__', '__setattr__', '__setitem__', '__setslice__', '__sizeof__', '__str__', '__subclasshook__', 'append', 'count', 'extend', 'index', 'insert', 'pop', 'remove', 'reverse', 'sort']



贴士#12. 简化 if 语句



我们可以使用下面的方式来验证多个值：



if m in [1,3,5,7]:



而不是：



if m==1 or m==3 or m==5 or m==7:



或者，对于 in 操作符我们也可以使用 '{1,3,5,7}' 而不是 '[1,3,5,7]'，因为 set 中取元素是 O(1) 操作。



贴士#13. 运行时检测 Python 版本



当正在运行的 Python 低于支持的版本时，有时我们也许不想运行我们的程序。为达到这个目标，你可以使用下面的代码片段，它也以可读的方式输出当前 Python 版本：



import sys

 

#Detect the Python version currently in use.

if not hasattr(sys, "hexversion") or sys.hexversion != 50660080:

    print("Sorry, you aren't running on Python 3.5n")

    print("Please upgrade to 3.5.n")

    sys.exit(1)

 

#Print Python version in a readable format.

print("Current Python version: ", sys.version)



或者你可以使用 sys.version_info >= (3, 5) 来替换上面代码中的 sys.hexversion != 50660080，这是一个读者的建议。



在 Python 2.7 上运行的结果：



Python 2.7.10 (default, Jul 14 2015, 19:46:27)

[GCC 4.8.2] on linux

 

Sorry, you aren't running on Python 3.5

 

Please upgrade to 3.5.



在 Python 3.5 上运行的结果：



Python 3.5.1 (default, Dec 2015, 13:05:11)

[GCC 4.8.2] on linux

 

Current Python version:  3.5.2 (default, Aug 22 2016, 21:11:05)

[GCC 5.3.0]



贴士#14. 组合多个字符串



如果你想拼接列表中的所有记号，比如下面的例子：



>>> test = ['I', 'Like', 'Python', 'automation']



现在，让我们从上面给出的列表元素新建一个字符串：



>>> print ''.join(test)



贴士#15. 四种翻转字符串/列表的方式



# 翻转列表本身



testList = [1, 3, 5]

testList.reverse()

print(testList)

 

#-> [5, 3, 1]



# 在一个循环中翻转并迭代输出



for element in reversed([1,3,5]):

    print(element)

 

#1-> 5

#2-> 3

#3-> 1



# 一行代码翻转字符串



"Test Python"[::-1]



输出为 “nohtyP tseT”



# 使用切片翻转列表



[1, 3, 5][::-1]



上面的命令将会给出输出 [5,3,1]。



贴士#16. 玩转枚举



使用枚举可以在循环中方便地找到（当前的）索引：



testlist = [10, 20, 30]

for i, value in enumerate(testlist):

    print(i, ': ', value)

 

#1-> 0 : 10

#2-> 1 : 20

#3-> 2 : 30



贴士#17. 在 Python 中使用枚举量



我们可以使用下面的方式来定义枚举量：



class Shapes:

    Circle, Square, Triangle, Quadrangle = range(4)

 

print(Shapes.Circle)

print(Shapes.Square)

print(Shapes.Triangle)

print(Shapes.Quadrangle)

 

#1-> 0

#2-> 1

#3-> 2

#4-> 3



贴士#18. 从方法中返回多个值



并没有太多编程语言支持这个特性，然而 Python 中的方法确实（可以）返回多个值，请参见下面的例子来看看这是如何工作的：



# function returning multiple values.

def x():

    return 1, 2, 3, 4

 

# Calling the above function.

a, b, c, d = x()

 

print(a, b, c, d)

 

#-> 1 2 3 4



贴士#19. 使用 * 运算符（splat operator）来 unpack 函数参数



* 运算符（splat operator）提供了一个艺术化的方法来 unpack 参数列表，为清楚起见请参见下面的例子：



def test(x, y, z):

    print(x, y, z)

 

testDict = {'x': 1, 'y': 2, 'z': 3}

testList = [10, 20, 30]

 

test(*testDict)

test(**testDict)

test(*testList)

 

#1-> x y z

#2-> 1 2 3

#3-> 10 20 30



贴士#20. 使用字典来存储选择操作



我们能构造一个字典来存储表达式：



stdcalc = {

    'sum': lambda x, y: x + y,

    'subtract': lambda x, y: x - y

}

 

print(stdcalc['sum'](9,3))

print(stdcalc['subtract'](9,3))

 

#1-> 12

#2-> 6



贴士#21. 一行代码计算任何数的阶乘



Python 2.x.



result = (lambda k: reduce(int.__mul__, range(1,k+1),1))(3)

print(result)

 

#-> 6



Python 3.x.



import functools

result = (lambda k: functools.reduce(int.__mul__, range(1,k+1),1))(3)

print(result)

 

#-> 6