numpy的linspace和piecewise

linspace:制定区域内返回均匀间隔数字，方法定义

def linspace(start, stop, num=50, endpoint=True, retstep=False, dtype=None):

start：开始的数字

stop：结束的数字

num：共生成几个数字

endpoint：生成的数字中是否包含结束的数字

retstep：是否显示步长（即相邻两数的距离）

例如，如下代码

a=np.linspace(-2.5, 2.5, 6,endpoint=True,retstep=True)
print a

显示结果

(array([-2.5, -1.5, -0.5,  0.5,  1.5,  2.5]), 1.0)

 

piecewise:评估分段定义的函数

函数定义

def piecewise(x, condlist, funclist, *args, **kw):

x：输入数字可以是标量或向量

condlist：判断条件

funclist：根据条件对输入数字的函数运算

例如如下代码

x=np.linspace(-2.5, 2.5, 6,endpoint=True,retstep=False)
print x

a=np.piecewise(x, [x < 0, x >= 0], [lambda x: -x, lambda x: x])
print a

运行结果

[-2.5 -1.5 -0.5  0.5  1.5  2.5]
[ 2.5  1.5  0.5  0.5  1.5  2.5]

其中lambda 定义了一个匿名函数，lambda x:-x相当于定义了一个函数f(x)=-x