简单理解Python列表推导式

简单来说，列表推导式提供了一种更方便的方式来创建一个列表，也就是将原来几行才能完成的代码替换为一行搞定。其结构如下所示：

newlist = [expression for member in iterable (if conditional)]

翻译过来就是，对于一个可迭代对象iterable，我们每次从中取出一个member，假如其满足条件conditional(没有条件的话相当于直接为True)，那么则将其放入expression函数进行处理，并添加入newlist中。

注意，如果将外层的中括号去掉的话，推导式本质返回的是一个generator对象，而非一个list：

print(type(i for i in range(1, 100)))
# <class 'generator'>

print(type([i for i in range(1, 100)]))
# <class 'list'>

---

例子1

假设要创建一个从1到99的整数数组，原始写法如下：

ret = []
for i in range(1, 100):
	ret.append(i)
print(ret)

列表推导式的写法如下：

ret = [i for i in range(1, 100)]
print(ret)

---

例子2

假设要创建一个从1到99的奇数数组，原始写法如下：

ret = []
for i in range(1, 100):
	if i % 2 == 1:
		ret.append(i)
print(ret)

列表推导式的写法如下：

ret = [i for i in range(1, 100) if i % 2 == 1]
print(ret)

---

例子3

假设要创建一个包含从1到9的平方数的数组，原始写法如下：

ret = []
for i in range(1, 10):
	ret.append(i ** 2)
print(ret)

列表推导式的写法如下：

ret = [i ** 2 for i in range(1, 10)]
print(ret)

---

例子4

假设要从数组中移除元素K，一种原始写法如下：

lst = [1, 2, 3, 4, 5, 3, 6]
K = 3

ret = []
for x in lst:
    if x != K:
        ret.append(x)
print(ret)

列表推导式的写法如下：

lst = [1, 2, 3, 4, 5, 3, 6]
K = 3

ret = [x for x in lst if x != K]
print(ret)

从这里可以发现，列表推导式相当于filter函数的平替(甚至有可能更简洁)。使用filter的写法如下：

lst = [1, 2, 3, 4, 5, 3, 6]
K = 3

ret = list(filter(lambda x: x != K, lst))
print(ret)