Python 常用技巧

最新推荐文章于 2024-10-11 10:04:38 发布

转载最新推荐文章于 2024-10-11 10:04:38 发布 · 521 阅读

文章标签：

#python #常用技巧

Python 专栏收录该内容

18 篇文章

订阅专栏

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

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. 简化 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) 操作。

11.组合多个字符串

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

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

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

>>> print ''.join(test)

12.四种翻转字符串/列表的方式

翻转列表本身

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]。

13.玩转枚举

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

testlist = [10, 20, 30]
for i, value in enumerate(testlist):
    print(i, ': ', value)

#1-> 0 : 10
#2-> 1 : 20
#3-> 2 : 30

14.从方法中返回多个值

并没有太多编程语言支持这个特性，然而 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

15.使用字典来存储选择操作

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

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

16.一行代码计算任何数的阶乘

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

17.找到列表中出现最频繁的数

test = [1,2,3,4,2,2,3,1,4,4,4]
print(max(set(test), key=test.count))

#-> 4

18.从两个相关的序列构建一个字典

t1 = (1, 2, 3)
t2 = (10, 20, 30)

print(dict (zip(t1,t2)))

#-> {1: 10, 2: 20, 3: 30}

19.不使用循环构造一个列表

import itertools
test = [[-1, -2], [30, 40], [25, 35]]
print(list(itertools.chain.from_iterable(test)))

#-> [-1, -2, 30, 40, 25, 35]

20.在 Python 中实现一个真正的 switch-case 语句

下面的代码使用一个字典来模拟构造一个 switch-case。

def xswitch(x):
    return xswitch._system_dict.get(x, None)

xswitch._system_dict = {'files': 10, 'folders': 5, 'devices': 2}

print(xswitch('default'))
print(xswitch('devices'))

#1-> None
#2-> 2