Python-列表推导式

简介: Python-列表推导式

列表推导式


要生成list [0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10]可以用list(range(0, 11))


list(range(0, 11))


如果要生成`[0, 1x1, 2x2, 3x3, …, 10x10]该如何呢?


# Part1L = []for i in range(0, 11):
  L.append(i * i)
L = [0, 1, 4, 9, 16, 25, 36, 49, 64, 81, 100]# 循环太繁琐,而列表生成式则可以用一行语句代替循环生成上面的list,如下# Part2L1 = [_ * _  for _ in range(11)]
L2 = [0, 1, 4, 9, 16, 25, 36, 49, 64, 81, 100]


for循环后面还可以加上if判断,这样我们就可以筛选出仅偶数的平方:


>>> [x * x for x in range(1, 11) if x % 2 == 0]
[4, 16, 36, 64, 100]


还可以使用两层循环,可以生成全排列:


>>> [m + n for m in 'ABC' for n in 'XYZ']
['AX', 'AY', 'AZ', 'BX', 'BY', 'BZ', 'CX', 'CY', 'CZ']


三层和三层以上的循环就很少用到了。


运用列表生成式,可以写出非常简洁的代码。例如,列出当前目录下的所有文件和目录名,可以通过一行代码实现:


>>> import os
>>> [d for d in os.listdir('.')] # os.listdir可以列出文件和目录['a.py', '列表生成式.md']


for循环其实可以同时使用两个甚至多个变量,比如dictitems()可以同时迭代key和value:


>>> d = {'x': 'A', 'y': 'B', 'z': 'C' }
>>> for k, v in d.items():
...     print(k, '=', v)
...
y = B
x = A
z = C


因此,列表生成式也可以使用两个变量来生成list:


>>> d = {'x': 'A', 'y': 'B', 'z': 'C' }
>>> [k + '=' + v for k, v in d.items()]
['y=B', 'x=A', 'z=C']


最后把一个list中所有的字符串变成小写:


>>> L = ['Hello', 'World', 'IBM', 'Apple']
>>> [s.lower() for s in L]
['hello', 'world', 'ibm', 'apple']


if … else


例如,以下代码正常输出偶数:


>>> [x for x in range(1, 11) if x % 2 == 0]
[2, 4, 6, 8, 10]


但是,我们不能在最后的if加上else


>>> [x for x in range(1, 11) if x % 2 == 0 else 0]
  File "<stdin>", line 1
    [x for x in range(1, 11) if x % 2 == 0 else 0]
                                              ^
SyntaxError: invalid syntax


这是因为跟在for后面的if是一个筛选条件,不能带else,否则如何筛选?


另一些童鞋发现把if写在for前面必须加else,否则报错:


>>> [x if x % 2 == 0 for x in range(1, 11)]
  File "<stdin>", line 1
    [x if x % 2 == 0 for x in range(1, 11)]
SyntaxError: invalid syntax


这是因为for前面的部分是一个表达式,它必须根据x计算出一个结果。因此,考察表达式:x if x % 2 == 0,它无法根据x计算出结果,因为缺少else,必须加上else


>>> [x if x % 2 == 0 else -x for x in range(1, 11)]
[-1, 2, -3, 4, -5, 6, -7, 8, -9, 10]


上述for前面的表达式x if x % 2 == 0 else -x才能根据x计算出确定的结果。


该如何生成0-10偶数的三次方列表呢?


# 一种方法是循环L = []for i in range(11):   if i % 2 == 0:
        L.append(i * i * i)# [0, 8, 64, 216, 512, 1000]# 使用位运算判断奇偶性,使用 ** 来进行乘方操作L = []for i in range(1, 11):    if (i & 1) == 0:
        L.append(i ** 3)# [0, 8, 64, 216, 512, 1000]# 列表推导式L = [_ ** 3 for _ in range(11) if _ % 2 == 0]# 或者L3 = [_ ** 3 for _ in range(11) if (_ & 1) == 0]# [0, 8, 64, 216, 512, 1000]


写列表生成式时,把要生成的元素x * x放到前面,后面跟for循环,就可以把list创建出来,十分有用,多写几次,很快就可以熟悉这种语法。注意此处的if并不能像三目运算符一样,追加else。有且只允许if并写在最后

目录
相关文章
|
1月前
|
存储 JavaScript Java
(Python基础)新时代语言!一起学习Python吧!(四):dict字典和set类型;切片类型、列表生成式;map和reduce迭代器;filter过滤函数、sorted排序函数;lambda函数
dict字典 Python内置了字典:dict的支持,dict全称dictionary,在其他语言中也称为map,使用键-值(key-value)存储,具有极快的查找速度。 我们可以通过声明JS对象一样的方式声明dict
134 1
|
1月前
|
开发者 Python
Python列表推导式:优雅与效率的完美结合
Python列表推导式:优雅与效率的完美结合
338 116
|
1月前
|
Python
Python列表推导式:简洁与高效的艺术
Python列表推导式:简洁与高效的艺术
326 119
|
1月前
|
开发者 Python
Python列表推导式:优雅与效率的完美融合
Python列表推导式:优雅与效率的完美融合
302 104
|
1月前
|
Python
Python列表推导式:优雅与效率的艺术
Python列表推导式:优雅与效率的艺术
236 99
|
1月前
|
数据处理 Python
解锁Python列表推导式:优雅与效率的完美融合
解锁Python列表推导式:优雅与效率的完美融合
253 99
|
1月前
|
开发者 Python
Python列表推导式:一行代码的艺术与力量
Python列表推导式:一行代码的艺术与力量
318 95
|
2月前
|
开发者 Python
Python神技:用列表推导式让你的代码更优雅
Python神技:用列表推导式让你的代码更优雅
413 99
|
2月前
|
程序员 Python
Python列表推导式:简洁与高效的艺术
Python列表推导式:简洁与高效的艺术
265 99
|
1月前
|
Python
Python列表推导式:简洁与高效的艺术
Python列表推导式:简洁与高效的艺术

推荐镜像

更多