今天主要学习了函数作为变量及生成器的相关知识。
一、函数作为变量
在python中,函数就是一种特殊的类型。声明函数的时候,其实就是在声明类似时function的变量。变量能做的事,函数都可以做。
1、函数给其他变量赋值
def func1():
print('hello python')
c = func1
c() #用c()就等同于func1()调用函数
2、函数作为列表的元素
list2 = []
list3 = []
for x in range(10):
def func2(y):
print(x+y)
list2.append(func2)
list3.append(func2(x))
# list2中每个元素的值都是函数
print(list2)
print(list3)
# list2[0]就是一个函数
func = list2[0]
print(func(100))
# 调用list2中下标是1对应的函数,并且传参为10
x = 10
list2[1](10)
# 直接将函数作为列表的元素
funcs = [func1]
funcs[0]()
3、将函数作为字典的值
def sub(*nums):
"""
累计求差
:param nums: 求差的数
:return: 差
"""
if not nums:
return 0
# 默认是第一个数
sum1 = nums[0]
for item in nums[1:]:
sum1 -= item
return sum1
operation = {'+': lambda *nums: sum(nums), '-': sub, '*': lambda x, y: x*y}
result = operation['-'](10, 20, 30, -100)
print(result)
4、函数作为函数的参数(回调函数)
def clean_kitchen(time):
print('在%s,打扫厨房' % time)
print('收费200元')
return 200
def clean_floor(time):
print('在%s,做地板清洁服务' % time)
print('收费100元')
return 100
# 在指定的时间,叫指定的服务
def call_service(time: str, service):
service(time)
# 将函数作为参数,传给其他函数
call_service('上午10点', clean_kitchen)
call_service('下午2点', clean_floor)
5、函数作为函数的返回值
def operation(operator: str):
if operator == '+':
def my_sum(*nums):
sum1 = 0
for num in nums:
sum1 += num
print(sum1)
# 将求和的函数返回
return my_sum
elif operator == '*':
def my_sum(*nums):
sum1 = 1
for num in nums:
sum1 *= num
print(sum1)
# 将求和的函数返回
return my_sum
# operation('+')的结果是函数
operation('+')(1, 2, 3)
operation('*')(2, 3, 4)
二、生成器
可以把生成器看成一种容器,类似列表。生成器就是来乘胜迭代器。
1、生成式——产生一个迭代器的表达式
a = (x for x in range(10))
2、生成器和迭代器都是通过next来获取里面的数据
print(next(a)) #next(a)获取a值,取出来就存不进去了,没取一次会记录a的取值位置。
3、将生成转换成列表
list1 = [x for x in range(10)]
print(list1)
4、将生成器转换成字典
注意:容器类型的元素是元祖,并且元祖中有且只有两个元素的,才能转换成字典
dict1 = dict((x, x*2) for x in range(5)
print(dict1)
三、函数转换成生成器
yield关键字:只要函数中有yield关键字,那么这个函数就会变成一个生成器。
- 有yield的函数,在调用函数的时候不再是获取返回值,而是产生一个生成器对象,生成器对象中保留的是函数体。
- 当通过next获取生成器中的数据的时候,才会去执行函数体,执行到yield为止,并且将yield后面的结果作为生成的数据返回。同时记录结束的位置,下次再调用next的时候,从上次结束的位置接着往后执行。
def func2():
print('abc')
for x in range(10):
yield x
print('aaa')
gen = func2()
print(next(gen))
print(next(gen))
print(next(gen))
显示效果为:
abc
0
aaa
1
aaa
2
网友评论