上QQ阅读APP看本书，新人免费读10天
设备和账号都新为新人
2.3 函数
函数是实现某个特定的功能，可以重复使用的代码块。Python提供了许多内建函数，比如print()。用户既可以根据自己的需求自行创建函数，即用户自定义函数，又可以调用第三方库的函数/模块。下面重点介绍自定义函数和第三方库的函数/模块。
2.3.1 自定义函数
下面代码演示了先定义函数，再调用定义好的函数实现其功能的过程。
'''
定义函数
函数功能：两个数相加
输入：x,y是两个要输入的数字
输出：z是两个数相加的和，用return键来导出
'''
# 1.定义函数
def add (x,y):
z=x+y
return z
# 2.调用函数
a=1
b=2
c=add(a,b)
print ('a和b相加为',c)
a和b相加为 3
2.3.2 第三方库的函数
这里运用2.2.4节用过的例子来说明如何调用第三方库的函数。首先用import talib调用金融库TA-Lib，再结合TA-Lib自带的函数talib.SMA()来定义新的函数SMA。
import talib
def sma(self, n, array=False):
"""简单均线"""
result=talib.SMA(self.close, n)
if array=True:
return result
return result[-1]