python列表推导式

来源:互联网 发布:人工智能失控 编辑:程序博客网 时间:2024/04/28 01:16

列表推导式(list comprehension)是利用其他列表创建新列表(类似于数学术语中的集合推导式)的一种方法。它的工作方式类似于for循环,也很简单:

In [0]: [x*x for x in range(10)]Out[0]: [0, 1, 4, 9, 16, 25, 36, 49, 64, 81]


如果只想打印出那些能被3整除的平方数,只需要通过添加一个if部分在推导式中就可以完成:

In [1]: [x*x for x in range(10) if x % 3 == 0]Out[1]: [0, 9, 36, 81]

也可以增加更多的for语句的部分:

In [2]: [(x,y) for x in range(3) for y in range(3)]Out[2]: [(0, 0), (0, 1), (0, 2), (1, 0), (1, 1), (1, 2), (2, 0), (2, 1), (2, 2)]In [3]: [[x,y] for x in range(2) for y in range(2)]Out[3]: [[0, 0], [0, 1], [1, 0], [1, 1]]


0 0
原创粉丝点击