Python NumPy insert方法用法及代码示例

例子

插入一维数组

要在索引 1 处插入值 8：

a = np.array([4,5,6])
np.insert(a, 1, 8)



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

插入到展平的二维数组

考虑以下二维数组：

a = np.array([[3,4],[5,6]])
a



array([[3, 4],
       [5, 6]])

插入单个值

要插入到数组的扁平版本中：

np.insert(a, 1, 9)



array([3, 9, 4, 5, 6])

在这里，我们在 a 的扁平化版本的索引 1 处插入值 9。

插入多个值

要将多个值插入到数组的扁平版本中：

np.insert(a, [0,3], [8,9])



array([8, 3, 4, 5, 9, 6])

在这里，我们将值 8 和 9 插入到索引 0 和 3 中。

将行插入二维数组

插入单行

考虑以下：

a = np.array([[3,4],[5,6]])
a



array([[3, 4],
       [5, 6]])

要在索引 1 处插入单行：

np.insert(a, 1, [8,9], axis=0)     # axis=0 represents row-insertion



array([[ 3,  4],
       [ 8, 9],
       [ 5,  6]])

插入多行

考虑以下：

a = np.array([[3,4],[5,6]])
a



array([[3, 4],
       [5, 6]])

要在索引 0 和 1 处插入两个不同的行：

np.insert(a, [0,1], [[10,11],[12,13]], axis=0)



array([[10, 11],
       [ 3,  4],
       [12, 13],
       [ 5,  6]])

请注意行 [12,13] 如何添加到原始数组的第一个索引。

将列插入二维数组

插入单列

考虑以下：

a = np.array([[3,4],[5,6]])
a



array([[3, 4],
       [5, 6]])

要在索引 1 处插入单列：

np.insert(a, 1, [8,9], axis=1)      # axis=1 represents column-insertion



array([[ 3, 8,  4],
       [ 5, 9,  6]])

插入多列

考虑以下：

a = np.array([[3,4],[5,6]])
a



array([[3, 4],
       [5, 6]])

要在索引 0 和 2 处插入两个不同的列：

np.insert(a, [0,2], [[10,11],[12,13]], axis=1)      # axis=1 represents column-insertion



array([[10,  3,  4, 11],
       [12,  5,  6, 13]])