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]])