Python numpy insert用法及代碼示例

本文簡要介紹 python 語言中 numpy.insert 的用法。

用法:

numpy.insert(arr, obj, values, axis=None)

在給定索引之前沿給定軸插入值。

參數：

arr： array_like

輸入數組。

obj： 整數、切片或整數序列

定義插入值的索引或索引的對象。

當 obj 是單個標量或具有一個元素的序列時，支持多次插入(類似於多次調用 insert)。

values： array_like

要插入的值arr.如果類型值不同於arr,值轉換為的類型arr.值應該是這樣的arr[...,obj,...] = values是合法的。

axis： 整數，可選

沿其插入值的軸。如果axis為None，則首先將arr展平。

返回：

out： ndarray

一份arr和值插入。注意insert不會就地發生：返回一個新數組。如果軸是無，out是一個扁平數組。

注意：

請注意，對於更高尺寸的刀片對象=0表現得非常不同對象=[0]就像arr[:,0,:] = 值不同於arr[:,[0],:] = 值.

例子：

>>> a = np.array([[1, 1], [2, 2], [3, 3]])
>>> a
array([[1, 1],
       [2, 2],
       [3, 3]])
>>> np.insert(a, 1, 5)
array([1, 5, 1, ..., 2, 3, 3])
>>> np.insert(a, 1, 5, axis=1)
array([[1, 5, 1],
       [2, 5, 2],
       [3, 5, 3]])

序列和標量之間的區別：

>>> np.insert(a, [1], [[1],[2],[3]], axis=1)
array([[1, 1, 1],
       [2, 2, 2],
       [3, 3, 3]])
>>> np.array_equal(np.insert(a, 1, [1, 2, 3], axis=1),
...                np.insert(a, [1], [[1],[2],[3]], axis=1))
True

>>> b = a.flatten()
>>> b
array([1, 1, 2, 2, 3, 3])
>>> np.insert(b, [2, 2], [5, 6])
array([1, 1, 5, ..., 2, 3, 3])

>>> np.insert(b, slice(2, 4), [5, 6])
array([1, 1, 5, ..., 2, 3, 3])

>>> np.insert(b, [2, 2], [7.13, False]) # type casting
array([1, 1, 7, ..., 2, 3, 3])

>>> x = np.arange(8).reshape(2, 4)
>>> idx = (1, 3)
>>> np.insert(x, idx, 999, axis=1)
array([[  0, 999,   1,   2, 999,   3],
       [  4, 999,   5,   6, 999,   7]])