python列表生成式

1.生成列表

要生成 list[1,2,3,4,5]，可以用range(1,6)

要生成[1*1,2*2,3*3,4*4,5*5] 怎么办？

循环太繁琐，而列表生成式则可以用一行语句代替循环完成，这种写法就是python特有的列表生成式。

利用列表生成式，可以以非常简洁的代码生成 list。

 

练习：请利用列表生成式生成列表 [1x2, 3x4, 5x6, 7x8, ..., 29x30]

提示：range(1, 100, 2) 可以生成list [1, 3, 5, 7, 9,...]

代码参考：
要生成[1*1,2*2,3*3,4*4,5*5] 怎么办？
方法一，循坏
L = []
for x in range(1, 6):
    L.append(x * x)
print(L)
print("----------------------")

方法二，列表生成式
L1 = [x * x for x in range(1,6)]
print(L1)


'''
练习：请利用列表生成式生成列表 [1x2, 3x4, 5x6, 7x8, ..., 29x30]
提示：range(1, 100, 2) 可以生成list [1, 3, 5, 7, 9,...]
'''
y = [x * (x + 1) for x in range(1,30,2)]
print(y)

 

2.条件过滤

列表生成式的 for 循环后面还可以加上 if 判断。例如：

 

练习：请编写一个函数，它接受一个 list，然后把list中的所有字符串变成大写后返回，非字符串元素将被忽略。

提示：

1. isinstance(x, str) 可以判断变量 x 是否是字符串；

2. 字符串的 upper() 方法可以返回大写的字母。

代码参考：
L = [x * x for x in range(1, 11)]
print(L)

#如果我们只想要奇数的平方，不改动 range()的情况下，可以加上 if 来筛选
L1 = [x * x for x in range(1, 11) if x % 2 == 1]
print(L1)
'''有了 if 条件，只有 if 判断为 True 的时候，才把循环的当前元素添加到列表中'''


#编写一个函数，它接受一个 list，然后把list中的所有字符串变成大写后返回，非字符串元素将被忽略
def toUpper(L):
    return [x.upper() for x in L if isinstance(x, str)]
L1 = ['Hello', 0, 'RaoJuan', 317]
print(toUpper(L1))

 

3.多层表达式

for循环可以嵌套，因此，在列表生成式中，也可以用多层 for 循环来生成列表，例如：

 

练习：利用 3 层for循环的列表生成式，找出对称的 3 位数。例如，121 就是对称数，因为从右到左倒过来还是 121。

 

代码参考：
#coding:utf-8

#对于字符串 'ABC' 和 '123'，可以使用两层循环，生成全排列：
L = []
for x in 'ABC':
    for y in '123':
        L.append(x + y)
print("for循环结果：", L)

#使用多层表达式
L1 = [x + y for x in 'ABC' for y in '123']
print("列表生成式结果：", L1)


#利用 3 层for循环的列表生成式，找出对称的 3 位数。例如，121 就是对称数，因为从右到左倒过来还是 121

#方法一：多重for循环
L2=[]
for x in range(1, 10):
    for y in range(10):
        for z in range(1, 10):
            if x==z :
                L2.append(100*x+10*y+z)
print("方法一结果：", L2)

#方法二：多层列表生成式
L3 = [100 * x + 10 * y + z for x in range(1, 10) for y in range(10) for z in range(10) if x == z]
print("方法二结果：", L3)

#方法三：索引
L4 = [i for i in range(100, 1000) if str(i)[0] == str(i)[-1]]
print("方法三结果：", L4)

#方法四：切片
L5 = [i for i in range(100, 1000) if str(i)[:1] == str(i)[-1:]]
print("方法四结果：", L5)