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。