20个非常有用的单行Python代码片段
1. 写在前面
继上篇，继续在本文分享 20 个 Python 单行代码，可以在 30 秒或更短时间内轻松学会。这些单行代码不仅可以提高效率，同时使代码看起来更整洁、更易读。：）
个人博客： https://jianpengzhang.github.io/
CSDN博客： http://blog.csdn.net/u011521019
公众号： 滑翔的纸飞机
2. Python 单行代码
2.1. 单行 for 循环
for 循环是多行语句，但在 Python 中，可以使用列表推导在一行中编写 for 循环。例如，要过滤掉小于 250 的值，请看下面的代码示例。
常规方式：
In [3]: mylist = [200, 300, 400, 500]
In [4]: result = []
In [5]: for x in mylist:
...: if x > 250:
...: result.append(x)
...:
In [6]: print(result)
[300, 400, 500]
单行方式：
In [7]: result = [x for x in mylist if x > 250]
In [8]: print(result)
[300, 400, 500]
2.2. 单行 While 循环
在一行中使用 While 循环代码，演示两种不同的方法。
# Method 1 单一声明
In [9]: while True: print(1) # 无限 1
# Method 2 多重声明
In [10]: x = 0
...: while x < 5: print(x); x= x + 1
0
1
2
3
4
2.3. 单行 if else 语句
单行 if else 将使用三元运算符。三元运算符的语法是"[on true] if [expression] else [on false]"。
示例代码中展示 3 个示例，可以清楚地了解如何在单行 if-else 语句中使用三元运算符：
# Example 1: if else
In [11]: print("Yes") if 8 > 9 else print("No")
No
# Example 2: if elif else: 从右至左进行判断
In [15]: E = 2
In [16]: print("High") if E == 5 else print("数据STUDIO") if E == 2 else print("Low")
数据STUDIO
# Example 3: only if
In [17]: if 3 > 2: print("Exactly")
Exactly
2.4. 单行合并词典
如何使用一行代码将两个字典合并为一个字典。在这里，介绍两种合并词典的方法：
# 单行合并词典
In [18]: d1 = { 'A': 1, 'B': 2 }
...: d2 = { 'C': 3, 'D': 4 }
# Method1:
In [19]: d1.update(d2)
...: print(d1)
{'A': 1, 'B': 2, 'C': 3, 'D': 4}
# Method2:
In [20]: d3 = {**d1, **d2}
...: print(d3)
{'A': 1, 'B': 2, 'C': 3, 'D': 4}
2.5. 单行函数
第一种方法，使用与三元运算符或单行循环方法相同的函数定义。
第二种方法，使用 lambda 来定义函数。
请看下面的示例代码，以便更清楚地理解：
# 单行函数
# Method1：
In [21]: def fun(x): return True if x % 2 == 0 else False
...: print(fun(2))
True
# Method2：
In [22]: fun = lambda x : x % 2 == 0
In [23]: print(fun(2))
True
In [24]: print(fun(3))
False
2.6. 单行递归
如何在一行中使用递归。使用单行函数和单行 if-else 语句。下面是一个查找斐波那契数字的示例：
# 单行递归
#Fibonaci 单行递归示例
In [25]: def Fib(x): return 1 if x in {0, 1} else Fib(x-1) + Fib(x-2)
In [26]: print(Fib(5))
8
In [27]: print(Fib(15))
987
2.7. 单行筛选数组
使用 Python 列表推导方法可以在一行代码中完成对数组的筛选。下面是一个过滤偶数列表的示例:
# 单行筛选数组
# 常规方式：
In [29]: result = []
...: for x in mylist:
...: if x % 2 == 0:
...: result.append(x)
...: print(result)
[2, 8, 12]
# 单行方式：
In [30]: result = [x for x in mylist if x % 2 == 0]
...: print(result)
[2, 8, 12]
2.8. 单行异常处理
我们使用异常来处理 Python 中的运行时错误。你知道我们可以在一行中编写 Try except 语句吗？通过使用 exec() 语句，我们就可以做到这一点。
# 在一行中处理异常
# 常规方式
In [32]: try:
...: print(x)
...: except:
...: print("Error")
...:
Error
# 单行方式
In [34]: exec('try:print(x) \nexcept:print("Error")')
Error
【注意：】谨慎使用 exec、eval , 其中 eval() 尽量用ast 模块的 literal()替换。
2.9. 一行中将列表转换为字典
使用 Python enumerate() 函数在一行内将列表转换为字典。将列表传递给 enumerate()，然后使用 dict() 将最终输出转换为字典格式。
# 单行字典
In [36]: mydict = ["John", "Peter", "Mathew", "Tom"]
...: mydict = dict(enumerate(mydict))
...: print(mydict)
{0: 'John', 1: 'Peter', 2: 'Mathew', 3: 'Tom'}
2.10. 单行多变量
Python 允许在一行中进行多个变量赋值。下面的示例代码将向您展示如何做到这一点。
# 在一行中分配多个变量。
# 单行方法
In [37]: x = 5
...: y = 7
...: z = 10
...: print(x , y, z)
5 7 10
# 单行方式
In [38]: a, b, c = 5, 7, 10
...: print(a, b, c)
5 7 10
2.11. 单行数据交换
在一行中交换值。交换值是编程中一项有趣的任务，但它总是需要第三个变量名 temp 来保存交换值。这个单行代码片段将告诉你如何在一行中交换值，而不需要任何临时变量。
常规方式：
In [1]: v1 = 100
...: v2 = 200
...: temp = v1
...: v1 = v2
...: v2 = temp
...: print(v1, v2)
200 100
单行方式：
In [2]: v1, v2 = v2, v1
...: print(v1, v2)
100 200
2.12. 单行排序
排序是编程中常见的问题，Python 有许多内置方法来解决排序问题。下面的代码示例将展示如何用一行代码进行排序。
In [3]: mylist = [32, 22, 11, 4, 6, 8, 12]
# 常规方式
In [4]: mylist.sort()
...: print(mylist)
[4, 6, 8, 11, 12, 22, 32]
# 单行方式
In [5]: print(sorted(mylist))
[4, 6, 8, 11, 12, 22, 32]
2.13. 单行读取文件
不使用语句或普通读取方法，也能正确读取文件的某一行。
# 单行读取文件
# 常规方式
In [1]: with open("data.txt", "r") as file:
...: data = file.readline()
...: print(data)
...:
hello world
# 单行方式
In [3]: data = [line.strip() for line in open("data.txt","r")]
...: print(data)
['hello world', 'Hello Python']
2.14. 单行类
类总是多行工作。但在 Python 中，有一些方法可以在一行代码中使用类的特性。
# 单行类
# 常规方式
In [5]: class Emp:
...: def __init__(self, name, age):
...: self.name = name
...: self.age = age
...: emp1 = Emp("a44", 22)
...: print(emp1.name, emp1.age)
a44 22
# 当行方式
# Method 1： 带有动态属性的 Lambda
In [6]: Emp = lambda:None; Emp.name = "a44"; Emp.age = 22
...: print(Emp.name, Emp.age)
a44 22
# Method 2
In [7]: from collections import namedtuple
...: Emp = namedtuple('Emp', ["name", "age"]) ("a44", 22)
...: print(Emp.name, Emp.age)
a44 22
2.15. 单行分号
在本代码片段中，展示了如何使用分号在一行中编写多行代码。
In [8]: a = "Python"; b = "Programming"; c = "languages"; print(a, b, c)
Python Programming languages
2.16. 单行打印
这不是一个重要代码片段，但有时在不需要使用循环来执行任务时，它可能会很有用。
常规方式：
In [9]: for x in range(1, 5):
...: print(x)
...:
1
2
3
4
单行方式：
In [10]: print(*range(1, 5))
1 2 3 4
In [11]: print(*range(1, 6))
1 2 3 4 5
2.17. 单行map函数
map 函数是一个有用的高阶函数，它可以对每个元素应用一个函数。下面举例说明如何在一行代码中使用 map 函数。
# 使用list()函数，来将返回的 map 转换成 list 列表
In [12]: print(list(map(lambda a: a + 2, [5, 6, 7, 8, 9, 10])))
[7, 8, 9, 10, 11, 12]
# 示例： 字符串列表 转为大写字母
In [13]: directions = ["north", "east", "south", "west"]
...: def to_upper_case(s): return s.upper() # 单行函数
...: print(list(map(to_upper_case, directions))) # 单行map处理
['NORTH', 'EAST', 'SOUTH', 'WEST']
【备注】
map(function, iterable, ...)
- function - 针对每一个迭代调用的函数
- iterable - 支持迭代的一个或者多个对象。在 Python 中大部分内建对象，例如 lists, dictionaries, 和 tuples 都是可迭代的。
在 Python 3 中，map()返回一个与传入可迭代对象大小一样的 map 对象。
在 Python 2中，这个函数返回一个列表 list。
2.18. 删除列表多个元素
使用 del 方法在一行代码中删除 List 中的多个元素，而无需进行任何修改。当然这里还有其他解决方式；
In [19]: mylist = [100, 200, 300, 400, 500]
...: del mylist[1::2] # 下表1开始删除，步长2，mylist[1::2] =》[200, 400]
...: print(mylist)
[100, 300, 500]
2.19. print 打印重复内容
In [30]: print(':)' * 3)
...: print(':)' * 2)
...: print(':)' * 1)
:):):)
:):)
:)
2.20. 一行代码就能查找某一范围内的质数
如何编写一行代码来查找某一范围内的质数。
In [31]: print(list(filter(lambda a: all(a % b != 0 for b in range(2, a)),
...: range(2,20))))
[2, 3, 5, 7, 11, 13, 17, 19]