python--推导式、匿名函数

推导式

• 列表推倒式 [结果 for 变量 in 可迭代对象 if 筛选]
• 字典推倒式 {结果 for 变量 in 可迭代对象 if 筛选} 结果=>key:value
• 集合推倒式 {结果 for 变量 in 可迭代对象 if 筛选} 结果=>key

列表生成式

生成一个列表，里面有python1-python14的数据，平常我们都是这样生成的

lst = []
for i in range(1, 15):
    lst.append("python%s" % i)
print(lst)

结果：

['python1', 'python2', 'python3', 'python4', 'python5', 'python6', 'python7', 'python8', 'python9', 'python10', 'python11', 'python12', 'python13', 'python14']

列表推导式，最终给你的是列表

语法： [最终结果(变量) for 变量 in 可迭代对象]

lst = ["python%s" % i for i in range(1, 15)]
print(lst)

结果和上面的一样

[最终结果 for 变量 in 可迭代对象 if 条件]

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

结果：

[2, 4, 6, 8, 10, 12, 14, 16, 18]

给定一个列表，找出名字中有两个e的

names = [['Tom', 'Billy', 'Jefferson', 'Andrew', 'Wesley', 'Steven', 'Joe'],
         ['Alice', 'Jill', 'Ana', 'Wendy', 'Jennifer', 'Sherry', 'Eva']]
lst = [name for first in names for name in first if name.count("e") == 2]
print(lst)

结果：

['Jefferson', 'Wesley', 'Steven', 'Jennifer']

生成器表达式

要生成list[1, 2, 3, 4, 5, 6, 7, 8, 9, 10]可以用list(range(1, 11))

print(list(range(1, 11)))

结果：

[1, 2, 3, 4, 5, 6, 7, 8, 9, 10]

生成[1x1, 2x2, 3x3, ..., 10x10]怎么做？方法一是循环a=[]

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

结果：

[1, 4, 9, 16, 25, 36, 49, 64, 81, 100]

但是循环太繁琐，而列表生成式则可以用一行语句代替循环生成上面的list：

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

结果：

[1, 4, 9, 16, 25, 36, 49, 64, 81, 100]

for循环后面还可以加上if判断，这样我们就可以筛选出仅偶数的平方

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

结果：

[4, 16, 36, 64, 100]

还可以使用两层循环，可以生成全排列：

print([m + n for m in 'ABC' for n in 'XYZ'])

结果：

['AX', 'AY', 'AZ', 'BX', 'BY', 'BZ', 'CX', 'CY', 'CZ']

字典推导式

dic = {"a": "b", "c": "d"}
# 把字典中的key:value互换 .{"b":"a", "d":"c"}
new_dic = {dic[key]: key for key in dic}
print(new_dic)

结果：

{'b': 'a', 'd': 'c'}

lst1 = ["aaa", "bb", "cc", "dd"]
lst2 = ["AA", "BB", "CC", "DD"]
# {"aa":"AA", "bb":"BB","cc":"CC","dd":"DD"}
dic = {lst1[i]: lst2[i] for i in range(len(lst2))}
print(dic)

结果：

{'aaa': 'AA', 'bb': 'BB', 'cc': 'CC', 'dd': 'DD'}

集合推导式

lst = ["马化腾", "马化腾", "王建忠", "张建忠", "张建忠", "张雪峰", "张雪峰"]
s = {i for i in lst}  # 集合推倒式
print(s)

结果：

{'张雪峰', '王建忠', '张建忠', '马化腾'}

匿名函数

语法: 变量 = lambda 参数: 返回值

• 函数的参数可以有多个. 多个参数之间用逗号隔开
• 匿名函数不管多复杂. 只能写一行, 且逻辑结束后直接返回数据
• 返回值和正常的函数一样, 可以是任意数据类型

def func(n):
    return n * n


print(func(3))
a = func
a(3)
print(a.__name__)  # 查看函数的函数名

结果：

9
func

lambda 匿名函数
 x 参数
 : 后面是函数体(直接return的内容)

a = lambda x: x * x  # 一行搞定一个函数. 但是, 不能完成复杂的函数操作
print(a)
print(a(6))
print(a.__name__)

结果：

<function <lambda> at 0x0351B6A8>
36
<lambda>

b = lambda x, y: x + y
print(b(1, 3))
print(b.__name__)

结果：

4
<lambda>