列表解析式是将一个列表(实际上适用于任何可迭代对象(iterable))转换成另一个列表的工具。
numbers = [1,2,3,4,5]
doubled_ods = [] #1
for n in numbers: #3
if n % 2 == 1: #4
doubled_od = n *2 #2
doubled_ods.append(doubled_od)
结果:[2, 6, 10]
#等价于
#1 #2 #3 #4
doubled_odds = [n * 2 for n in numbers if n % 2 == 1]
结果:[2, 6, 10]
for循环转换列表解析,对应#1,2,3,4 这样就转换了
上述是有条件(if n % 2 == 1)的转换,无条件的转换去掉(if n % 2 == 1)这部分即可
嵌套循环
flattened = [] #1
for row in matrix:#3
for n in row: #4
flattened.append(n)#2
#1 #2 #3 #4
flattened = [n for row in matrix for n in row]
**请记住for循环子句的顺序与我们原来for循环的顺序是一致的。**
同样地原则也适用集合解析式(set comprehension)和字典解析式(dictionary comprehension)。
其他解析式
下面的代码提取单词序列中每个单词的首字母,创建了一个集合(set):
first_letters = set()
for w in words:
first_letters.add(w[0])
同样的代码可以改写为集合解析式:
first_letters = {w[0] for w in words}
下面的代码将原有字典的键和值互换,从而创建了一个新的字典:
flipped = {}
for key, value in original.items():
flipped[value] = key
同样的代码可以改写为字典解析式:
flipped = {value: key for key, value in original.items()}
还要注意可读性
你有没有发现上面的列表解析式读起来很困难?我经常发现,如果较长的列表解析式写成一行代码,那么阅读起来就非常困难。
不过,还好Python支持在括号和花括号之间断行。
列表解析式 List comprehension
断行前:
doubled_odds = [n * 2 for n in numbers if n % 2 == 1]
断行后:
doubled_odds = [
n * 2
for n in numbers
if n % 2 == 1
]
带嵌套循环的列表解析式
断行前:
flattened = [n for n in row for row in matrix]
断行后:
flattened = [
n
for row in matrix
for n in row
]
字典解析式
断行前:
flipped = {value: key for key, value in original.items()}
断行后:
flipped = {
value: key
for key, value in original.items()
}
请注意,我们并不是随意进行断行:我们是在每一行复制过来的代码之后断行的。
总结
纠结于写不出列表解析式吗?不要担心。先写一个for循环,能后按照本文说的顺序复制粘贴,就可以写出解析式了。
任何类似下面代码形式的for循环:
new_things = []
for ITEM in old_things:
if condition_based_on(ITEM):
new_things.append("something with " + ITEM)
都可以被改写为下面这种列表解析式:
new_things = ["something with " + ITEM for ITEM in old_things if condition_based_on(ITEM)]
转载自:http://codingpy.com/article/python-list-comprehensions-explained-visually/
例子:https://www.cnblogs.com/liu-shuai/p/6098227.html