python计算分段函数_Python如何定义分段函数

Python如何定义分段函数呢？下面是一个分段函数：

一开始写了个简单版本log_norm0，只能逐元素一个一个得进行。不用想，对元素个数多的向量，肯定慢成乌龟。

后来想到使用一个指示函数，来区分分段的两种情况，就得到log_norm1。不过这种情况比较特殊，不是每次都能成功构造的。

相关推荐：《Python视频教程》

最后，查到numpy中有函数piecewise(x, condlist, funclist, *args, **kw)，它是专门用来构造分段函数，x是输入，condlist表示分段的条件，funclist就表示对应分段的处理函数。这就得到了log_norm2。# elementwise

def log_norm0(x):

if x >= 0:

return np.log(x + 1)

else:

return - np.log(- x + 1)

# indicator

def log_norm1(x):

# ind = np.where(x > 0, 1, 0)

ind = (x > 0)

return