使用列表生成式生成列表
>>> lst1 = [x**2 for x in range(10)]
>>> lst1
[0, 1, 4, 9, 16, 25, 36, 49, 64, 81]
>>> mybag = [' glass','apple ','green leaf ']
>>> mybag2 =[one.strip() for one in mybag] # 去除空格
>>> mybag2
['glass', 'apple', 'green leaf']
>>>
条件过滤
列表生成式的for循环后面还可以加上if判断。
# 将list中的所有字符串变成大写,非字符串元素忽略
def toUppers(L):
return [i.upper() for i in L if isinstance(i,str)]
L = toUppers(['Hello', 'world', 101])
print (L)
多层表达式
for循环可嵌套,在列表生成式中,可以使用多层for循环来生成列表。
b = [100 * n1 + 10 * n2 + n3 for n1 in range(1, 10) for n2 in range(10) for n3 in range(10) if n1==n3]
print (len(b))
print (b)
# 翻译成循环代码如下:
L = []
for n1 in range(1,10):
for n2 in range(10):
for n3 in range(10):
if n1==n3:
L.append(100 * n1 + 10 * n2 + n3)
print (len(L))
print (L)
可以看出,使用python列表生成式来生成列表,十分的简洁。在工作中也会频繁的使用它。所以,让我们一起pick它吧~~~