这一章说说Python中的函数,和OC一样,Python也内置了许许多多的函数,大家夸一自行前往查看。
调用函数
首先使用一个最简单的求绝对值的函数:
>>> abs(100)
100
>>> abs(-20)
20
>>> abs(12.34)
12.34
在终端调用的话就是这么简单,其中abs
是函数名,100
是参数。
调用函数的时候,如果传入的参数数量不对,会报TypeError
的错误,并且Python会明确地告诉你:abs()
有且仅有1个参数,但给出了两个:
>>> abs(1, 2)
Traceback (most recent call last):
File "<stdin>", line 1, in <module>
TypeError: abs() takes exactly one argument (2 given)
同样参数类型不对也会报错:
>>> abs('a')
Traceback (most recent call last):
File "<stdin>", line 1, in <module>
TypeError: bad operand type for abs(): 'str'
比如max
函数就可以接受多个参数,返回最大的一个。
自定义函数
自定义函数的时候需要使用def
语句,与OC不同。
def my_abs(x):
if x >= 0:
return x
else:
return -x
print(my_abs(-10))
10
基本上和OC差不多,遇到return
就返回,但是如果没有return
语句的话,也会返回结果,会返回一个none
。
如果写一个函数暂时不想做任何事情,可以这么写:
if a > 18:
pass
可以使用pass
语句,就可以避免报错。
自定义函数的时候也可以做参数检查,比如我们要输入的是int
类型,如果输入的是字符串,那么就给他跑出个错误,用上面的my_abs
函数来做个例子:
def my_abs(x):
if not isinstance(x, (int, float)):
raise TypeError('bad operand type')
if x >= 0:
return x
else:
return -x
第2行对参数x
做了个检查,如果不是int
或者float
的话,就抛出一个异常。
如果我们自定义的函数保存在本地的话,比如保存在了test.py
的文件中,那么我们就需要cd
到此文件的同意目录下,然后在这个目录下打开python
解释器,也就是在这个目录下,在终端输入python3
,然后在交互环境中,通过from abstest import my_abs
导入这个函数,执行了这一句后,在同一目录下会生成一个__pycache__
的文件夹,里面是个.pyc
的文件,这个是什么,目前还不懂,后面再解释。
当然函数也可以返回多个值:
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
,类似于OC的不可变数组。
验证的话就是这样:
>>> r = move(100, 100, 60, math.pi / 6)
>>> print(r)
(151.96152422706632, 70.0)