hidewood

博客园 首页 新随笔 联系 订阅 管理

函数

函数用于封装一段可重复使用的逻辑。使用函数的目的通常有 3 个:

  • 减少重复代码
  • 让程序结构更清晰
  • 让代码更容易测试和维护

调用函数

函数名本身其实就是指向函数对象的引用,所以完全可以把函数名赋值给另一个变量,相当于给这个函数起了一个“别名”:

a = abs
print(a(-1))  # 1

练习:请利用 Python 内置的 hex() 函数,把一个整数转换成十六进制表示的字符串。

>>> hex(255)
'0xff'
>>> hex(1000)
'0x3e8'

调用 Python 的函数时,需要根据函数定义传入正确的参数。函数调用出错时,要重点看报错信息中的:

  • 错误类型
  • 出错位置
  • 具体提示内容

定义函数

定义函数的基本形式如下:

def 函数名(参数1, 参数2, ...):
    函数体
    return 返回值

如果没有 return 语句,函数执行完毕后也会返回结果,只是结果为 None

def show():
    print('hello')

result = show()
print(result)  # None

return None 通常可以简写为 return,甚至直接省略 return

空函数

如果想定义一个什么事也不做的空函数,可以用 pass 语句:

def nop():
    pass

pass 也可以出现在其他语句中,用来占位:

if age >= 18:
    pass

参数检查

我们自己定义的函数如果接收到不合适的参数,解释器不一定会自动给出符合业务语义的错误,这时可以主动检查并抛出异常:

def my_abs(x):
    if not isinstance(x, (int, float)):
        raise TypeError('bad operand type')
    if x >= 0:
        return x
    return -x

上面的代码会先检查 x 是否是 intfloat,如果不是,就主动报错。

返回多个值

函数看起来可以“返回多个值”:

import math

def move(x, y, step, angle=0):
    nx = x + step * math.cos(angle)
    ny = y - step * math.sin(angle)
    return nx, ny
>>> x, y = move(100, 100, 60, math.pi / 6)
>>> print(x, y)
151.96152422706632 70.0

但这本质上不是同时返回多个独立对象,而是返回了一个 tuple

>>> r = move(100, 100, 60, math.pi / 6)
>>> print(r)
(151.96152422706632, 70.0)

也就是说:

return nx, ny

等价于:

return (nx, ny)

之所以默认表现为 tuple,核心原因不是“必须更安全”,而是:

  • Python 会把逗号分隔的返回值自动打包为元组
  • tuple 很适合表达“固定数量、固定位置含义”的返回结果
  • 后续解包也很方便,例如 x, y = move(...)

如果你确实想返回一个列表,也完全可以显式写成:

return [nx, ny]

练习

请定义一个函数 quadratic(a, b, c),接收 3 个参数,返回一元二次方程 ax^2 + bx + c = 0 的两个解。

>>> def quadratic(a, b, c):
...     return (
...         (-b + (b ** 2 - 4 * a * c) ** 0.5) / (2 * a),
...         (-b - (b ** 2 - 4 * a * c) ** 0.5) / (2 * a)
...     )
...
>>> quadratic(2, 3, 1)
(-0.5, -1.0)
>>> quadratic(1, 3, -4)
(1.0, -4.0)

说明:

  • 这个写法默认判别式 b ** 2 - 4 * a * c >= 0
  • 实际开发中,还应考虑 a == 0 和无实数解的情况

小结

  • 定义函数时,需要确定函数名和参数个数
  • 如果有必要,可以先对参数类型或取值范围做检查
  • 函数体内部可以用 return 随时返回结果
  • 函数执行完毕但没有 return 语句时,会自动返回 None
  • 函数“返回多个值”时,本质上返回的是一个 tuple

函数的参数

位置参数

例如:

def power(x, n):
    return x ** n

这里的 xn 就是位置参数。调用函数时,实参会按照位置顺序依次赋值给对应参数。

power(2, 3)  # 8

默认参数

可以给参数提供默认值:

def power(x, n=2):
    return x ** n

此时:

power(5)

等价于:

power(5, 2)

默认参数有几点需要注意:

  • 必选参数在前,默认参数在后
  • 有多个参数时,通常把变化大的参数放前面,变化小的参数放后面

默认参数的好处是:降低调用函数的难度。

def enroll(name, gender, age=6, city='Beijing'):
    print('name:', name)
    print('gender:', gender)
    print('age:', age)
    print('city:', city)

调用时:

enroll('Bob', 'M', 7)
enroll('Adam', 'M', city='Tianjin')

注意:当不按顺序传入部分默认参数时,需要显式写参数名。

默认参数的坑

先定义一个函数,传入一个列表,添加一个 'END' 再返回:

def add_end(L=[]):
    L.append('END')
    return L

当显式传入列表时,结果看起来没什么问题:

>>> add_end([1, 2, 3])
[1, 2, 3, 'END']
>>> add_end(['x', 'y', 'z'])
['x', 'y', 'z', 'END']

但当使用默认参数调用时,结果就开始出现问题:

>>> add_end()
['END']
>>> add_end()
['END', 'END']
>>> add_end()
['END', 'END', 'END']

原因是:默认参数的值会在函数定义时计算一次,而不是在每次调用时重新创建。

也就是说,上面的 [] 只创建了一次。之后每次调用 add_end(),如果没有显式传入 L,用的都是同一个列表对象。

因此,更准确的说法不是“默认参数必须指向不变对象”,而是:

  • 语法上,不变对象和可变对象都可以作为默认参数
  • 实践上,默认参数应尽量使用不变对象,避免共享可变状态带来的逻辑错误

推荐写法如下:

def add_end(L=None):
    if L is None:
        L = []
    L.append('END')
    return L

这里用 None 作为默认值,是因为 None 是不可变对象,而且常用来表示“没有传值”。

可变参数

在 Python 函数中,还可以定义可变参数,也就是参数个数可变:

def calc(*numbers):
    total = 0
    for n in numbers:
        total = total + n * n
    return total

调用示例:

calc(1, 2, 3)  # 14
calc()         # 0

这里的 numbers 接收到的是一个 tuple

如果已经有一个 listtuple,调用可变参数函数时可以用 * 展开:

>>> nums = [1, 2, 3]
>>> calc(*nums)
14

关键字参数

关键字参数允许传入 0 个或任意个带参数名的参数,这些参数在函数内部会自动组装为一个 dict

def person(name, age, **kw):
    print('name:', name, 'age:', age, 'other:', kw)

调用示例:

>>> person('Michael', 30)
name: Michael age: 30 other: {}

>>> person('Bob', 35, city='Beijing')
name: Bob age: 35 other: {'city': 'Beijing'}

>>> person('Adam', 45, gender='M', job='Engineer')
name: Adam age: 45 other: {'gender': 'M', 'job': 'Engineer'}

关键字参数的作用是扩展函数能力。比如做用户注册时,nameage 是必填项,其它信息可以放进关键字参数里。

也可以先组装出一个字典,再用 ** 展开:

>>> extra = {'city': 'Beijing', 'job': 'Engineer'}
>>> person('Jack', 24, **extra)
name: Jack age: 24 other: {'city': 'Beijing', 'job': 'Engineer'}

注意:**extra 表示把 extra 中的键值对作为关键字参数传入。函数内部的 kw 会得到一个新的字典对象,对 kw 的修改不会影响外部的 extra

命名关键字参数

对于普通关键字参数 **kw,调用者可以传入任意名字的关键字参数。如果想限制关键字参数的名字,可以使用命名关键字参数。

例如,只允许传入 cityjob

def person(name, age, *, city, job):
    print(name, age, city, job)

这里的 * 是特殊分隔符,* 后面的参数都必须用“参数名=值”的形式传入。

>>> person('Jack', 24, city='Beijing', job='Engineer')
Jack 24 Beijing Engineer

如果函数定义中已经有了可变参数,那么后面的命名关键字参数就不再需要额外写 *

def person(name, age, *args, city, job):
    print(name, age, args, city, job)

命名关键字参数必须传入参数名,否则会报错:

>>> person('Jack', 24, 'Beijing', 'Engineer')
Traceback (most recent call last):
  ...
TypeError: person() missing 2 required keyword-only arguments: 'city' and 'job'

命名关键字参数也可以有默认值:

def person(name, age, *, city='Beijing', job):
    print(name, age, city, job)

此时可以省略 city

>>> person('Jack', 24, job='Engineer')
Jack 24 Beijing Engineer

如果没有可变参数,就必须保留这个特殊分隔符 *

def person(name, age, city, job):
    # 这里没有 *,city 和 job 是位置参数,不是命名关键字参数
    pass

参数组合

Python 中常见的 5 类参数分别是:

  • 必选参数
  • 默认参数
  • 可变参数
  • 命名关键字参数
  • 关键字参数

它们组合使用时,参数定义顺序通常应为:

必选参数 -> 默认参数 -> 可变参数 -> 命名关键字参数 -> 关键字参数

例如:

def f1(a, b, c=0, *args, **kw):
    print('a =', a, 'b =', b, 'c =', c, 'args =', args, 'kw =', kw)

def f2(a, b, c=0, *, d, **kw):
    print('a =', a, 'b =', b, 'c =', c, 'd =', d, 'kw =', kw)

调用示例:

>>> f1(1, 2)
a = 1 b = 2 c = 0 args = () kw = {}
>>> f1(1, 2, c=3)
a = 1 b = 2 c = 3 args = () kw = {}
>>> f1(1, 2, 3, 'a', 'b')
a = 1 b = 2 c = 3 args = ('a', 'b') kw = {}
>>> f1(1, 2, 3, 'a', 'b', x=99)
a = 1 b = 2 c = 3 args = ('a', 'b') kw = {'x': 99}

>>> f2(1, 2, d=99, ext=None)
a = 1 b = 2 c = 0 d = 99 kw = {'ext': None}

也可以通过一个 tuple 和一个 dict 来调用:

>>> args = (1, 2, 3, 4)
>>> kw = {'d': 99, 'x': '#'}
>>> f1(*args, **kw)
a = 1 b = 2 c = 3 args = (4,) kw = {'d': 99, 'x': '#'}

>>> args = (1, 2, 3)
>>> kw = {'d': 88, 'x': '#'}
>>> f2(*args, **kw)
a = 1 b = 2 c = 3 d = 88 kw = {'x': '#'}

所以,对于任意函数,只要参数匹配,都可以通过 func(*args, **kw) 的形式调用。

练习

以下函数允许计算两个数的乘积,请稍加改造,变成可接收一个或多个数并计算乘积:

def mul(x, y):
    return x * y

# 测试
print('mul(5) =', mul(5))
print('mul(5, 6) =', mul(5, 6))
print('mul(5, 6, 7) =', mul(5, 6, 7))
print('mul(5, 6, 7, 9) =', mul(5, 6, 7, 9))
if mul(5) != 5:
    print('mul(5)测试失败!')
elif mul(5, 6) != 30:
    print('mul(5, 6)测试失败!')
elif mul(5, 6, 7) != 210:
    print('mul(5, 6, 7)测试失败!')
elif mul(5, 6, 7, 9) != 1890:
    print('mul(5, 6, 7, 9)测试失败!')
else:
    try:
        mul()
        print('mul()测试失败!')
    except TypeError:
        print('测试成功!')

参考解答:

def mul(x, *numbers):
    product = x
    for n in numbers:
        product = product * n
    return product

说明:这里故意把第一个参数写成必选参数 x,这样 mul() 不传任何参数时就会触发 TypeError,与题目测试保持一致。

小结

  • Python 的函数参数形式非常灵活
  • 默认参数建议使用不可变对象,或者用 None 作为占位默认值
  • *args 接收的是一个 tuple
  • **kw 接收的是一个 dict
  • 调用函数时,可以用 * 展开序列,用 ** 展开字典
  • 命名关键字参数用于限制允许传入的关键字名称

递归函数

在函数内部,可以调用其他函数。如果一个函数在内部调用自身,这个函数就是递归函数。

举个例子,我们来计算阶乘 n! = 1 x 2 x 3 x ... x n,用函数 fact(n) 表示:

fact(n) = n! = 1 × 2 × 3 × ... × (n - 1) × n = fact(n - 1) × n

因此,fact(n) 可以表示为 n * fact(n - 1),只有 n == 1 时需要特殊处理。

def fact(n):
    if n == 1:
        return 1
    return n * fact(n - 1)
>>> fact(1)
1
>>> fact(5)
120

计算 fact(5) 的过程如下:

fact(5)
= 5 * fact(4)
= 5 * (4 * fact(3))
= 5 * (4 * (3 * fact(2)))
= 5 * (4 * (3 * (2 * fact(1))))
= 5 * (4 * (3 * (2 * 1)))
= 120

递归函数的优点是定义简单、逻辑清晰。理论上,大多数递归都能改写成循环。

递归的风险:栈溢出

函数调用是通过栈(stack)实现的。每进入一层函数调用,就会新增一层栈帧;每返回一次,就会减少一层栈帧。栈空间不是无限的,所以递归层数过深时,会导致栈溢出。

>>> fact(1000)
Traceback (most recent call last):
  ...
RecursionError: maximum recursion depth exceeded

尾递归

尾递归是指:函数返回时,直接返回递归调用本身,并且 return 语句外层不再包含额外表达式。

上面的 fact(n)

return n * fact(n - 1)

不是尾递归,因为返回前还要做乘法。

可以改写为:

def fact(n):
    return fact_iter(n, 1)

def fact_iter(num, product):
    if num == 1:
        return product
    return fact_iter(num - 1, num * product)

调用过程如下:

fact_iter(5, 1)
= fact_iter(4, 5)
= fact_iter(3, 20)
= fact_iter(2, 60)
= fact_iter(1, 120)
= 120

理论上,如果语言或解释器支持尾递归优化,那么尾递归不会随着调用次数增加而不断增长栈空间。

但要注意:Python 标准解释器并不会做尾递归优化。所以即使写成尾递归,递归层数过深时仍然会栈溢出。

练习

汉诺塔可以用递归函数非常简洁地实现。

请编写 move(n, a, b, c) 函数,它接收参数 n,表示第一个柱子 a 上有 n 个盘子,目标是借助柱子 b 把所有盘子移动到柱子 c

示例调用:

move(3, 'A', 'B', 'C')

期待输出:

A --> C
A --> B
C --> B
A --> C
B --> A
B --> C
A --> C

参考解答:

def move(n, a, b, c):
    if n == 1:
        print(a, '-->', c)
    else:
        move(n - 1, a, c, b)  # 先把 n-1 个盘子从 a 借助 c 移到 b
        print(a, '-->', c)    # 再把底部盘子从 a 移到 c
        move(n - 1, b, a, c)  # 最后把 n-1 个盘子从 b 借助 a 移到 c

小结

  • 递归适合描述“问题可以拆成更小的同类问题”的场景
  • 写递归时必须明确终止条件
  • Python 不做尾递归优化,递归层数过深时会触发 RecursionError
  • 能用循环清楚表达时,循环通常更直接

补充:文档字符串与类型标注

前面的内容足够入门函数的定义与调用。如果希望函数更容易被别人理解,还可以补充以下两点。

文档字符串

文档字符串用于说明函数的用途、参数和返回值:

def add(x, y):
    """返回两个数的和。"""
    return x + y

查看方式:

print(add.__doc__)

类型标注

类型标注不会强制 Python 运行时做类型检查,但能提升可读性,并帮助编辑器提示:

def add(x: int, y: int) -> int:
    return x + y

对于初学阶段,不要求到处都写,但建议开始认识这种写法。

常见误区

  • “函数返回多个值”并不是真的返回多个对象,本质上通常是返回一个 tuple
  • 默认参数不是语法上必须用不可变对象,而是为了避免共享可变状态,实践上应尽量这样做
  • *args 是把多余的位置参数收集为元组,**kw 是把多余的关键字参数收集为字典
  • 递归不一定比循环高级;只是某些问题用递归表达更自然

练习题

下面这组练习按难度从低到高排列,建议独立完成。

基础题

  1. 编写函数 temperature_convert(celsius),把摄氏度转换成华氏度并返回结果。
  2. 编写函数 is_even(n),判断一个整数是否为偶数,返回 TrueFalse
  3. 编写函数 max_of_three(a, b, c),返回 3 个数中的最大值。
  4. 编写函数 triangle_area(base, height),返回三角形面积。
  5. 编写函数 login(username, password='123456'),体验默认参数的作用。

参数练习

  1. 编写函数 average(*numbers),接收任意个数字并返回平均值;如果没有传入数字,返回 0
  2. 编写函数 student_info(name, age, **extra),打印学生基本信息和扩展信息。
  3. 编写函数 book_info(title, *, author, price),要求 authorprice 必须按关键字形式传入。
  4. 编写函数 append_item(item, target=None),把 item 追加到列表中,并正确处理默认参数。
  5. 编写函数 calc_stats(*numbers),返回最小值、最大值和平均值。

综合题

  1. 编写函数 quadratic(a, b, c),要求额外处理:
    • a == 0 时给出合理提示
    • 判别式小于 0 时给出合理提示
  2. 编写函数 register(name, age, gender='未填写', **extra),模拟一个用户注册函数。
  3. 编写函数 factorial(n),分别用循环和递归实现,并比较两种写法。
  4. 编写函数 fib(n),返回斐波那契数列第 n 项。
  5. 编写函数 move_count(n),不打印汉诺塔过程,只返回移动总步数。

小项目练手

项目:个人记账小助手

目标:用函数把一个命令行小程序拆分清楚,练习“参数设计、返回值设计、数据校验和功能拆分”。

建议功能:

  • 显示菜单
  • 添加一条收入记录
  • 添加一条支出记录
  • 查看所有记录
  • 按类型统计总金额
  • 统计总收入、总支出、当前结余
  • 按关键词查询记录
  • 删除指定记录

建议拆分的函数:

def show_menu():
    pass

def add_record(records, kind, amount, note):
    pass

def show_records(records):
    pass

def delete_record(records, index):
    pass

def search_records(records, keyword):
    pass

def calc_summary(records):
    pass

def calc_by_kind(records, kind):
    pass

def is_valid_amount(amount):
    pass

def main():
    pass

可以约定每条记录都长这样:

{'kind': '支出', 'amount': 35.5, 'note': '午饭'}

建议要求:

  • 金额必须大于 0
  • 收支类型只能是“收入”或“支出”
  • 删除记录时要检查下标是否合法
  • 所有菜单操作尽量通过函数完成,不要把逻辑全堆在 main()

进阶版本

如果基础版完成了,可以继续扩展:

  • 支持按日期查看记录
  • 支持将记录保存到文件、从文件读取
  • 支持按“餐饮、交通、学习、娱乐”等分类统计
  • 支持设置每月预算,并提示是否超支

项目收获

做完这个项目后,你应该重点回顾这些问题:

  • 我是否把重复逻辑提取成了函数?
  • 每个函数职责是否单一?
  • 哪些参数应该用位置参数,哪些适合默认参数?
  • 有没有错误地把可变对象写成默认参数?
  • 返回值设计是否清晰,调用方是否容易使用?
posted on 2026-05-05 23:30  hidewood  阅读(0)  评论(0)    收藏  举报