python 列表推导式

自学python，总结一下，云储存

1,举个普通例子

# 列表推导式置于一对方括号之中

[x**2 for x in range(10)]  #计算range(10)中每个数的平方，推导出新列表
# 得到新列表[0, 1, 4, 9, 16, 25, 36, 49, 64, 81]

2，可自定义函数

# 自定义函数f(x)
def fun(x):
    return x + x ** 2 + x ** 3     # 返回 x + x ** 2 + x ** 3

y = [fun(i) for i in range(10)]    # 列表推导式，按函数fun(x)，推导出新列表
print(y)                           # 输出列表[0, 3, 14, 39, 84, 155, 258, 399, 584, 819]

3，加上if从句

def fun(x):
    return x + x**2 + x ** 3  # 返回x + x ** 2 + x ** 3

# 列表推导式，根据原列表中的偶数，推导新列表
y = [fun(i) for i in range(10) if i%2 is 0]  
print(y)                                     # 输出列表[0, 14, 84, 258, 584]

4，多个For从句

ls = [(x, y) for x in [1,2,3] for y in [3,1,4] if x != y]
print(ls)      # 输出[(1, 3), (1, 4), (2, 3), (2, 1), (2, 4), (3, 1), (3, 4)]

5，多个例子

ls = [-4, -2, 0, 2, 4]
print([x*2 for x in ls])# 将原列表每个数字乘2，推导出新列表 [-8, -4, 0, 4, 8]
print([x for x in ls if x >= 0])   # 过滤列表，返回只包含正数的列表[0, 2, 4]
print([abs(x) for x in ls])        # 应用abs()函数推导新列表[4, 2, 0, 2, 4]
# 调用strip()方法去除每个元素前后的空字符,返回['banana', 'apple', 'pear']
freshfruit = ['  banana', '  apple ', 'pear  ']
print([fruit.strip() for fruit in freshfruit]) #
# 生成一个每个元素及其平方(number, square)构成的元组组成的列表
print([(x, x**2) for x in range(6)]) 
# [(0, 0), (1, 1), (2, 4), (3, 9), (4, 16), (5, 25)]

6，字典推导

dic = {'李明': '20191202001', '张凡':'20191202002'} # 姓名：学号
dic_update = {v: k for k, v in dic.items()}
print(dic_update)
# 输出交换了键和值的字典 {'20191202001': '李明', '20191202002': '张凡'}

7，集合推导

print({x ** 3 for x in range(10)})
# 输出集合 {0, 1, 64, 512, 8, 343, 216, 729, 27, 125}