码迷,mamicode.com
首页 > 其他好文 > 详细

列表生成式

时间:2018-05-04 20:23:32      阅读:172      评论:0      收藏:0      [点我收藏+]

标签:筛选   list   range   if语句   apple   就是   color   需要   class   

 

你可以通过list(range(0,11))生成一个列表,但是如果让你生成[1, 4, 9, 16, 25, 36, 49, 64, 81, 100],那就比较困难,但是聪明的你想到了循环:

L = []
for i in range(0,11):
    L.append( i * i )

 简单的列表生成式

这样也可以,很棒。但是Python提供了一种更简单的方式,一行代码就解决:

>>> [i * i for i in range(0, 11)]
[1, 4, 9, 16, 25, 36, 49, 64, 81, 100]

 

这就是列表生成式。把需要生成的元素 i * i 放在前面,后面紧跟一个for循环,非常简单。
再看一个例子,将列表中所有的元素变成大写:

>>> L = [Hello, World, IBM, Apple]
>>> [s.upper() for s in L]
[HELLO, WORLD, IBM, APPLE]

 if语句进行筛选

你还可以加上if语句来进行筛选:

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

 

这就相当于:

L = []
for i in range(0, 11):
    if (i * i) % 2 == 0:
        L.append(i * i)

 多层循环

 不仅如此,列表生成式还可以实现多层循环:

 

>>> [m + n for m in 123 for n in abc]
[1a, 1b, 1c, 2a, 2b, 2c, 3a, 3b, 3c]

 

 相当于:

L = []
for m in 123:
    for n in abc:
        L.append(m + n)

 




 

列表生成式

标签:筛选   list   range   if语句   apple   就是   color   需要   class   

原文地址:https://www.cnblogs.com/c-x-m/p/8992259.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!