numpy中clip()函數的用法

numpy中的clip()函數用於將數組中的元素控制在一個給定的範圍內，給定需要控制的範圍的上下邊界，clip函數將所有小於下邊界的數值全部改爲下邊界， 將大於上邊界的數值全部改爲上邊界。

numpy中clip()函數的原型爲：clip(self, min=None, max=None, out=None)

其中，self爲需要控制的數組，min爲下邊界，max爲上邊界，out爲存儲輸出結果的數組。

clip函數共有兩種用法，下面用具體的例子來說明：

import numpy as np


# 先隨機生成一個數組
a = np.arange(12).reshape((3, 4))

# 生成一個和a一樣維度的數組用於保存結果
result = np.arange(12).reshape((3, 4))

# 結果如下：
# [[ 0  1  2  3]
#  [ 4  5  6  7]
#  [ 8  9 10 11]]


# 方式一：

np.clip(a, 3, 8, result)
print(result)

# 結果如下：

# [[3 3 3 3]
#  [4 5 6 7]
#  [8 8 8 8]]


# 方式二：
result_ = a.clip(3, 8)
print(result_)

# 結果如下：

# [[3 3 3 3]
#  [4 5 6 7]
#  [8 8 8 8]]

可以看到兩種方法都將數組的值的範圍限制在了給定的最小值3和最大值8之間。