Python——列表生成式

列表生成式

列表生成式 或叫列表解析
格式：

[expression for i in 序列 ]

题目一

生成一个列表，各项分别为1～n的平方[1**2,2**2,3**2,4**2,.....,n**2]
一般方式：

li = []
for i in range(1,11):
    li.append(i**2)
print(li)

列表生成式：

print([i ** 2 for i in range(1,11)])

加if判断条件的形式

格式：

[expression for i in 序列 if...]

题目一

生成一个列表，列表元素为1～10之间的所有偶数各自的平方
一般方式：

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

列表生成式：

print([i ** 2 for i in range(1,11) if i %2 == 0])

题目二

找出1—10之间所有偶数，并且返回一个列表(包含以这个偶数为半径的圆的面积)
pi值的表示：

import math
a = math.pi
print(a)

列表生成式：

import math
a = math.pi
print([math.pi * i * i for i in range(1,11) if i % 2 == 0])

题目三

找出1～num之间的所有质数

def isPrime(num):
    for i in range(2,num):
        if num % i == 0:    # 不是质数
            return False
    else:       # 跟for对齐，否则循环的不完全，会取出非质数
        return True
print([i for i in range(2,101) if isPrime(i)]) 

题目四

给定一串字符串，将下述代码用列表生成式表示出来

s = '51 5000 10000'
li = []
for item in s.split():
    li.append(int(item))
print(li)

列表生成式：

s = '51 5000 10000'
print([int(item) for item in s.split()])

题目五

找出日志/var/log目录下以.log结尾的文件并打印出来
导入虚拟命令行：

import os
print(os.listdir('/var/log'))

列表生成式实现如下：

import os
print([filename for filename in os.listdir('/var/log') if filename.endswith('.log')])

列表生成式变形

s1 = ‘ABC’
s2=‘123’
要求输出列表：A1 A2 A3 … C1 C2 C3
列表生成式：

s1 = 'ABC'
s2 = '123'
print([i + j for i in 'ABC' for j in '123'])

给定列表li，要求输出列表[1,2,3,4,5,6,7,8,9]
一般方式：

li = [
    [1,2,3],
    [4,5,6],
    [7,8,9]
]
resultli = []
for i in li:
    for j in i:
        resultli.append(j)
print(resultli)

列表生成式：

li = [
    [1,2,3],
    [4,5,6],
    [7,8,9]
]
print([j for i in li for j in i ])