python使用列表推导

来源:互联网 发布:淘宝怎么查看购物车 编辑:程序博客网 时间:2024/06/05 12:45

列表推导List comprehensions:在编程的时候,我们常常想要将一种数据类型转换为另一种。下面是一个简单例子,将列表中的每个元素变成它的平方。

nums = [0, 1, 2, 3, 4]squares = []for x in nums:    squares.append(x ** 2)print squares   # Prints [0, 1, 4, 9, 16]

使用列表推导,你就可以让代码简化很多:

nums = [0, 1, 2, 3, 4]squares = [x ** 2 for x in nums]print squares   # Prints [0, 1, 4, 9, 16]

列表推导还可以包含条件:

nums = [0, 1, 2, 3, 4]even_squares = [x ** 2 for x in nums if x % 2 == 0]print even_squares  # Prints "[0, 4, 16]"
0 0