我是靠谱客的博主 隐形香烟,最近开发中收集的这篇文章主要介绍Python如何定义分段函数,觉得挺不错的,现在分享给大家,希望可以做个参考。

概述

Python如何定义分段函数呢?下面是一个分段函数:

1564725837(1).png

一开始写了个简单版本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 np.log(x * ind + 1) - np.log(- x * (1.0 - ind) + 1)
# numpy.piecewise()
def log_norm2(x):
    return np.piecewise(x, [x >= 0, x < 0], [lambda x: np.log(1 + x), lambda x: - np.log(1 - x)])
登录后复制

最后,观察一个各个函数的运行时间。

tic = time.time()
for i in range(x.size):
    y[i] = log_norm0(x[i])
toc = time.time()
print('log0: ', toc - tic)
tic = time.time()
y = log_norm1(x)
toc = time.time()
print('log1: ', toc - tic)
tic = time.time()
z = log_norm2(x)
toc = time.time()
print('log2: ', toc - tic)
登录后复制

观察结果,还是使用指示函数的方法最快,不过跟piecewise差别不大。

log0:  33.59282732009888
log1:  0.4863457679748535
log2:  0.5942573547363281
登录后复制

以上就是Python如何定义分段函数的详细内容,更多请关注靠谱客其它相关文章!

最后

以上就是隐形香烟为你收集整理的Python如何定义分段函数的全部内容,希望文章能够帮你解决Python如何定义分段函数所遇到的程序开发问题。

如果觉得靠谱客网站的内容还不错,欢迎将靠谱客网站推荐给程序员好友。

本图文内容来源于网友提供,作为学习参考使用,或来自网络收集整理,版权属于原作者所有。
点赞(96)

评论列表共有 0 条评论

立即
投稿
返回
顶部