Python numpy expand_dims用法及代碼示例

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

用法:

numpy.expand_dims(a, axis)

展開數組的形狀。

插入一個新軸，該軸將出現在展開的數組形狀中的軸位置。

參數：

a： array_like

輸入數組。

axis： int 或整數元組

放置新軸(或多個軸)的擴展軸中的位置。

返回：

result： ndarray

視圖隨著維度數量的增加而增加。

例子：

>>> x = np.array([1, 2])
>>> x.shape
(2,)

以下等效於 x[np.newaxis, :] 或 x[np.newaxis] ：

>>> y = np.expand_dims(x, axis=0)
>>> y
array([[1, 2]])
>>> y.shape
(1, 2)

以下等價於 x[:, np.newaxis] ：

>>> y = np.expand_dims(x, axis=1)
>>> y
array([[1],
       [2]])
>>> y.shape
(2, 1)

axis 也可能是一個元組：

>>> y = np.expand_dims(x, axis=(0, 1))
>>> y
array([[[1, 2]]])

>>> y = np.expand_dims(x, axis=(2, 0))
>>> y
array([[[1],
        [2]]])

請注意，某些示例可能使用 None 而不是 np.newaxis 。這些是相同的對象：

>>> np.newaxis is None
True