定义函数

Python中,定义函数是用def语句,一次写出函数名、括号、括号中的参数、和冒号:,然后在缩进模块中编写函数体,函数的返回值使用return语句返回。

我们以自定义一个求绝对值的my_abs函数为例:
def my_abs(x):

    if x>=0:
      return x

    else:
      return -x

注意,函数体内的语句在执行时,一旦执行到return,函数就执行完毕。并将结果返回。如果没有return语句,函数返回结果为none。return none可以简写为return。

在Python交互式命令行下,定义函数python会出现...的提示符,函数定义结束后需按两次会车回到>>>的提示符下。

如果已经把my_abs()的函数定义保存到abstest.py文件,那么可以在该文件当前目录下,使用from abstest import my_abs 导入my_abs()函数。注意abstest是文件名,不包含.py扩展名。

空函数

def nop():

    pass

pass语句什么都不做,用作占位符,比如现没想好怎么写函数代码,就可以先放一个pass,让代码能运行起来。

pass还可以用在其他语句里:
if x >0:
    pass

参数检查

函数调用的时候,如果函数的参数个数不对,Python解释器能自动检查出来。并抛出TypeError;

但是函数的参数类型不对,Python解释器无法帮我们查出。

>>> my_abs('A')

Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
  File "<stdin>", line 2, in my_abs
TypeError: unorderable types: str() >= int()
>>> abs('A')
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>TypeError: bad operand type for abs(): 'str'

当传入了不恰当参数时,内置函数会检查出参数错误,而我们定义的my_abs()没有参数检查,会导致if语句出错。对参数类型做检查可以使用isinstance()函数实现。

def my_abs(x):

    if not isinstance(x,(int,float)):
      raise TypeError('bad operand type')

    if x>=0:

      return x

    if x<0:

      return -x

再次调用:

>>> my_abs('A')
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
  File "<stdin>", line 3, in my_abs
TypeError: bad operand type

返回多值

函数可以返回多个值。

比如这个计算坐标的函数:

import mathdef move(x, y, step, angle=0):
    nx = x + step * math.cos(angle)
    ny = y - step * math.sin(angle)    
    return nx,ny

import math语句,导入math包,并允许后续代码引用math中的sin cos函数。

然后我们就可以同时获得返回值:

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

但其实这是一个假象,Python返回的仍是一个单值:

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

返回值是一个tuple。语法上,返回一个tuple可以省略括号,而多个变量可以同时接受一个tuple,按位置赋给对应的值,所以,Python的函数返回的多值就是返回的一个tuple,但写起来更方便。


小结:

定义函数时,需要确定函数名和参数个数;

如果有必要,可以对参数的数据类型做检查;

函数体内可以使用return随时返回函数结果;

函数执行完若没有return,自动return none;

函数可以同时返回多值,但其实就是返回一个tuple。