Python定義分段函數(shù)的方法

這篇文章主要介紹Python定義分段函數(shù)的方法，文中介紹的非常詳細(xì)，具有一定的參考價(jià)值，感興趣的小伙伴們一定要看完！

Python如何定義分段函數(shù)呢？下面是一個(gè)分段函數(shù)：

一開(kāi)始寫(xiě)了個(gè)簡(jiǎn)單版本log_norm0，只能逐元素一個(gè)一個(gè)得進(jìn)行。不用想，對(duì)元素個(gè)數(shù)多的向量，肯定慢成烏龜。

后來(lái)想到使用一個(gè)指示函數(shù)，來(lái)區(qū)分分段的兩種情況，就得到log_norm1。不過(guò)這種情況比較特殊，不是每次都能成功構(gòu)造的。

最后，查到numpy中有函數(shù)piecewise(x, condlist, funclist, *args, **kw)，它是專門用來(lái)構(gòu)造分段函數(shù)，x是輸入，condlist表示分段的條件，funclist就表示對(duì)應(yīng)分段的處理函數(shù)。這就得到了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)])

最后，觀察一個(gè)各個(gè)函數(shù)的運(yùn)行時(shí)間。

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)

觀察結(jié)果，還是使用指示函數(shù)的方法最快，不過(guò)跟piecewise差別不大。

log0:  33.59282732009888
log1:  0.4863457679748535
log2:  0.5942573547363281

以上是Python定義分段函數(shù)的方法的所有內(nèi)容，感謝各位的閱讀！希望分享的內(nèi)容對(duì)大家有幫助，更多相關(guān)知識(shí)，歡迎關(guān)注億速云行業(yè)資訊頻道！