Python NumPy put方法用法及代碼示例

例子

基本用法

a = np.array([5,6,7,8])
np.put(a, 1, 9)
a



array([5, 9, 7, 8])

二維數組

考慮以下二維數組：

a = np.array([[5,6],[7,8]])
a



array([[5, 6],
       [7, 8]])

設置單個值

a = np.array([[5,6],[7,8]])
a.put(3, 9)
a



array([[5, 6],
       [7, 9]])

設置多個值

a = np.array([[5,6],[7,8]])
np.put(a, [1,0], 9)
a



array([[9, 9],
       [7, 8]])

這裏，請注意[1,0]並不代表第2行第1列。相反，它表示 a 的扁平化版本的第 1 個和第 0 個索引。因此，我們可以等效地使用[0,1] 來代替。

不同模式

考慮以下二維數組：

a = np.array([[5,6],[7,8]])
a



array([[5, 6],
       [7, 8]])

增加

a = np.array([[5,6],[7,8]])
np.put(a, [5], 9, mode="raise")
a



IndexError: index 5 is out of bounds for axis 0 with size 4

引發此錯誤的原因是展平數組中不存在索引 5。

裹

a = np.array([[5,6],[7,8]])
np.put(a, [5], 9, mode="wrap")
a



array([[5, 9],
       [7, 8]])

在這裏，我們圍繞 a 進行了一個循環，其大小為 4，因此索引 5 轉換為索引 5-4=1。

夾子

a = np.array([[5,6],[7,8]])
np.put(a, [5], 9, mode="clip")
a



array([[5, 6],
       [7, 9]])

在這裏，我們指定的索引超出了界限，因此最後一個索引是目標，在本例中是索引 3。