一、 函数高级内容
Python 中的函数是“一等公民”,因此函数本身也是一个对象,函数即可用于赋值,也可用作其函数的参数,还可作为其他函数的返回值。
1、 使用函数变量
Python的函数也是一种值。所有函数都是 function 对象,可以把函数本身赋值给变量,就像把整数、列表、元组等赋值给变量一样。把函数赋值给变量后,就可通过变量来调用函数。示例如下:
1 def pow(base, exponent): 2 """计算乘方"""
3 result = 1
4 for i in range(1, exponent + 1): 5 result *= base 6 return result 7 my_fun = pow # 将 pow 函数赋值给 my_fun 变量
8 print(my_fun(3, 3)) # 计算3的3次方,输出:27
9
10 def area(length, width): 11 """计算面积"""
12 return length * width 13 my_fun = area # 将 area 函数赋值给 my_fun 变量
14 print(my_fun(4, 5)) # 计算面积,输出:20
这里对 my_fun 变量赋值不同的函数,可以让 my_fun 在不同的时间指向不同的函数来让程序更加灵活。使用函数变量的好处是让程序更加灵活。
2、 使用函数作为函数形参
在函数中定义函数形参,这样可在调用该函数时传入不同的函数作为参数,从而动态改变这段代码。示例如下:
1 def map(data, fn): 2 """定义函数类型的形参,其中 fn 是一个函数"""
3 result = [] 4 # 遍历 data 列表中的每个元素,并用 fn 函数对每个元素进行计算
5 # 然后将计算结果作为新数组的元素
6 for e in data: 7 result.append(fn(e)) 8 return result 9 # 定义一个计算平方的函数
10 def square(n): 11 return n * n 12 # 定义一个计算立方的函数
13 def cube(n): 14 return n * n * n 15 # 定义一个计算阶乘的函数
16 def factorial(n): 17 result = 1
18 for i in range(2, n + 1): 19 result *= i 20 return result 21 data = [2, 4, 6, 8, 10] 22 print("列表原始数据:", data) 23 print("计算列表元素的平方:") 24 print(map(data, square)) 25 print("计算元素的立方:") 26 print(map(data, cube)) 27 print("计算元素的阶乘:") 28 print(map(data, factorial)) 29 print(type(map)) 30
31 运行程序输出如下: 32 列表原始数据: [2, 4, 6, 8, 10] 33 计算列表元素的平方: 34 [4, 16, 36, 64, 100] 35 计算元素的立方: 36 [8, 64, 216, 512, 1000] 37 计算元素的阶乘: 38 [2, 24, 720, 40320, 3628800] 39 <class 'function'>
这里定义的 map() 函数的第二个参数是一个函数类型参数,这意味着每次调用函数时可以动态传入一个函数。传入不同的函数,可以动态改变 map() 函数中的部分计算代码。Python3 内置的 map 函数与这里定义的 map 函数功能相似,但是更强大。
3、 使用函数作为返回值
Python支持使用函数作为其他函数的返回值。示例如下:
1 def foo(type): 2 """定义一个函数,该函数包含局部函数"""
3 def square(n): 4 """定义一个计算平方的局部函数"""
5 return n * n 6 def cube(n): 7 """定义一个计算立方的局部函数"""
8 return n * n * n 9 def factorial(n): 10 """定义一个计算阶乘的局部函数"""
11 result = 1
12 for i in range(2, n + 1): 13 result *= i 14 return result 15 # 调用局部函数
16 if type == 'square': 17 # 注意返回的函数不能加括号,只返回函数名称
18 return square 19 elif type == 'cube': 20 return cube 21 else: 22 return factorial 23 # 调用 foo 函数,程序返回一个嵌套函数
24 cube = foo('cube') 25 print(cube(10)) # 计算立方,输出:1000
26 square = foo('square') 27 print(square(5)) # 计算平方:输出:25
28 factorial = foo('factorial') 29 print(factorial(5)) # 计算阶乘:输出:120
这次定义的 foo 函数体定义了3个局部函数,foo 函数由参数来决定返回哪一个函数。
二、 局部函数与 lambda 表达式
lambda 表达式是功能更灵活的代码块,可以在程序中被传递和调用。
从前面的局部函数内容可知,局部函数的作用域仅在其封闭函数体内,离开这个封闭函数体,局部函数的函数名就失去了意义(当然可以用返回函数名的方法进行调用),可考虑用 lambda 表达式简化局部函数的写法。
1、 使用 lambda 表达式代替局部函数
使用 lambda 表达式简化前面的 foo 函数,代码如下:
1 def foo(type): 2 result = 1
3 # 该函数返回的是 lambda 表达式
4 if type == 'square': 5 return lambda n: n * n 6 elif type == 'cube': 7 return lambda n: n * n * n 8 else: 9 return lambda n: (n + 1) * n / 2
10 # 调用 foo 函数,程序返回的是一个嵌套函数
11 square = foo('square') 12 print(square(10)) # 计算平方,输出:100
13 cube = foo('cube') 14 print(cube(10)) # 计算立方,输出:1000
15 other = foo('other') 16 print(other(10)) # 求和,输出:55.0
这次定义的 foo 函数体内,return 后面使用 lambda 关键字定义的是 lambda 表达式。Python 要求 lambda 表达只能是单行表达式,不允许使用复杂的函数形式。
lambda 表达式的语法格式如下:
lambda [参数列表]: 表达式
lambda 表达式的几个要点:
(1)、lambda 表达式必须使用 lambda 关键字定义。
(2)、在 lambda 关键字之后、冒号左边的是参数列表,可以不给参数,也可以给多个参数。多个参数用逗号分隔,冒号右边是 lambda 表达的返回值。
(3)、lambda 表达式也叫做匿名函数,也是单行函数体的函数。
函数比 lambda 表达式的适应性更强,lambda 表达式只能创建简单的函数对象(适合函数体为单行的情形)。lambda表达的两个用途:
(1)、对于单行函数,使用 lambda 表达式更简洁,省去定义函数的过程。
(2)、对于不需要多次复用的函数,使用 lambda 表达式可以在用完之后立即释放,提高性能。
下面示例使用内置的 map() 函数,参数是 lambda 表达式:
1 # 计算平方
2 x = map(lambda x: x * x, range(5)) 3 print([i for i in x]) 4 # 计算偶数的平方
5 y = map(lambda x: x * x if x % 2 ==0 else 0, range(5)) 6 print([i for i in y])
这段代码中 map 函数是内置函数,第一个参数是函数,第二个参数通常是列表、元组等。这里传入的 lambda 表达式,是简化的函数,这样程序更加简洁,性能也更好。
三、 小结
函数和lambda 表达式是 Python 编程的两大核心机制之一。Python 支持面向过程编程,也支持面向对象编程。函数和 lambda 表达式是面向过程编程的语法基础,需要重点掌握。
要掌握函数的定义、函数调用语法,函数位置参数与关键字参数的区别和用法、形参默认值等特性。此外,函数也是一个 function 对象,函数可作为其他函数的参数,也可作为其它函数的返回值。把函数当成参数传入其他函数,可以让编程变得更加灵活。
lambda 表达式是单行函数的简化版本,功能比较简单。
练习:
1、 定义一个函数,该函数可接收一个 list 作为参数,该函数使用直接选择排序对 list 排序。
1 def choose_sort(sequence): 2 length = len(sequence) 3 for i in range(length): 4 for j in range(i + 1, length): 5 if sequence[i] > sequence[j]: 6 sequence[i], sequence[j] = sequence[j], sequence[i] 7
8 seq = [100, 31, -50, 1, 3, 28, -11, 85] 9 choose_sort(seq) 10 print(seq) # 输出:[-50, -11, 1, 3, 28, 31, 85, 100]
2、定义一个函数,该函数接收一个 list 作为参数,该函数使用冒泡排序对 list 排序。
默认按升序排序,即从小到大排序。
冒泡排序思路:首先,列表每两个相邻的数,如果前边的数比后边的数大,那么交换这两个数,第一次遍历完后,最大的这个数就在列表的末尾,接着第二次遍历,指针回到列表起始处,继续比较两个相邻的数,如果前边比后边大的,就交换这两个数,以此类推。
1 def bubble_sort(sequence): 2 length = len(sequence) 3 for i in range(length): 4 is_sorted = True # 如果内层循环已经是从小到大的顺序,可通过判断这个变量来结束循环
5 for j in range(length - i - 1): # 减 1 是避免遍历到最后一个元素
6 if sequence[j] > sequence[j + 1]: 7 sequence[j], sequence[j + 1] = sequence[j + 1], sequence[j] 8 is_sorted = False 9 if is_sorted: return # 如果内层循环已排序完成,就结束外层循环
10
11 seq = [100, 31, -50, 1, 3, 28, -11, 85] 12 bubble_sort(seq) 13 print(seq) # 输出:[-50, -11, 1, 3, 28, 31, 85, 100]
3、定义一个 is_leap(year) 函数,判断 year 是否为闰年。若是闰年则返回 True; 否则返回 False.
闰年分为普通闰年和世纪闰年。普通闰年是指公历年份是4的倍数,例如2016年就是闰年。世纪闰年是指公历年份是整百数,必须是400的倍数才是闰年,例如1900年不是世纪闰年,2000是世纪闰年。
1 def is_year(year): 2 year = int(year) 3 # 过滤整百数年份,判断普通闰年
4 if (year % 4 == 0) and (year % 100 != 0): 5 return True 6 # 判断世纪闰年
7 elif year % 400 == 0: 8 return True 9 else: 10 return False 11
12 while True: 13 year = input("请输入年份:") 14 if year == 'q' or year == 'exit': 15 exit(0) 16 else: 17 print("%s是否是闰年?%s" % (year, is_year(year)))
4、 定义一个 count_str_char(my_str) 函数,该函数返回参数字符串中包含多少个数字、多少个英文字母、多少个空白字符、多少个其它字符。
1 def count_str_char(my_str): 2 char_num, digit_num, space_num, other_num = 0, 0, 0, 0 3 for s in my_str: 4 if s.isdigit(): digit_num += 1
5 elif s.isalpha(): char_num += 1
6 elif s.isspace(): space_num += 1
7 else: other_num += 1
8 return char_num, digit_num, space_num, other_num 9
10 while True: 11 my_str = input("请输入一个字符串:") 12 if my_str == 'q': 13 exit() 14 char_num, digit_num, space_num, other_num = count_str_char(my_str) 15 print("字符个数是:", char_num) 16 print("数字个数是:", digit_num) 17 print("空白个数是:", space_num) 18 print("其它字符个数是:", other_num) 19 print("-" * 30)
5、定义一个函数fn1(n),该函数返回 1到n 的立方和,即求:1+2*2*2+3*3*3+...+n*n*n
1 def fn1(n): 2 if n < 2: 3 exit() 4 result = 1
5 for i in range(2, n + 1): 6 result += i * i * i 7 return result 8 n = int(input("请输入一个数字:")) 9 sum = fn1(n) 10 print("1~%d的立方和是:%s" % (n, sum))
6、定义一个fn2(n) 函数,该函数返回 n 的阶乘。
1 def fn2(n): 2 if n < 2: 3 return 1
4 result = 1
5 for i in range(2, n + 1): 6 result *= i 7 return result 8
9 def fn3(n): 10 """用递归函数来实现阶乘"""
11 if n <= 1: 12 return 1
13 else: 14 return fn3(n - 1) * n 15
16 def fn4(n): 17 """用列表的方式计算阶乘"""
18 if n <= 1: 19 return 1
20 my_list = [1] 21 for _ in range(n): 22 my_list.append(my_list[-1] * len(my_list)) 23 return my_list[-1] 24
25 n = int(input("请输入一个整数:")) 26 print("%d的阶乘是:%d" % (n, fn4(n)))
7、定义一个函数,接收一个列表作为参数,函数用于去除列表中的重复元素。
1 def no_repeat(sequence): 2 my_list = [] 3 for s in sequence: 4 if s not in my_list: 5 my_list.append(s) 6 return my_list 7
8 def no_repeat_2rd(sequence): 9 """利用字典的 fromkeys 方法,将列表的元素转换为字典的键, 10 因字典的键不能重复,通过获取字典的键列表,就是去重后的列表"""
11 my_list = list({}.fromkeys(sequence).keys()) 12 return my_list 13
14 seq = ['a', 'java', 'c', 'python', 'c', 'stark', 'michael', 'c', 'a', 'python'] 15 print(no_repeat_2rd(seq)) 16
17 输出: 18 ['a', 'java', 'c', 'python', 'stark', 'michael']
8、定义一个函数 fn5(n) ,用于返回一个包含 n 个不重复的 0~100 之间整数的元组。
注意:输入的整数大于100时,程序会陷入死循环。
1 import random 2
3 def fn5(n): 4 i, my_list =0, [] 5 while True: 6 temp_num = random.randint(0, 100) 7 if temp_num not in my_list: 8 my_list.append(temp_num) 9 i += 1
10 if i == n: 11 break # 结束循环
12 return tuple(my_list) 13
14 n = int(input("请输入一个整数:")) 15 my_tuple = fn5(n) 16 print("元组元素个数是:%d" % len(my_tuple)) 17 print(my_tuple)
9、定义一个函数 fn6(n),该函数返回一个包含 n 个不重复的大写字母的元组。
1 import random 2 def fn6(n): 3 i, my_list =0, [] 4 while True: 5 num = random.randint(65, 90) 6 s = chr(num) 7 if s not in my_list: 8 my_list.append(s) 9 i += 1
10 if i == n: 11 break
12 return tuple(my_list) 13
14 n = int(input("请输入一个整数:")) 15 print(fn6(n))
10、定义一个函数 fn7(n),参数 n 表示一个 n 行 n 列的矩阵。在输出时,先输出 n 行 n 列的矩阵,再输出该矩阵的转置形式。例如参数是3时,先输出
1 2 3
4 5 6
7 8 9
再输出:
1 4 7
2 5 8
3 6 9
1 def fn7(n): 2 # 输出矩阵
3 for i in range(n): 4 for j in range(n): 5 print(" %2d " % (1 + j + i * n), end="") 6 print() 7 print("-" * (4 * n)) 8
9 # 输出转置矩阵
10 for i in range(n): 11 for j in range(n): 12 print(" %2d " % (1 + i + j * n), end="") 13 print() 14 n = int(input("请输入整数:")) 15 fn7(n)