Python numpy concatenate用法及代碼示例

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

用法:

numpy.concatenate((a1, a2, ...), axis=0, out=None, dtype=None, casting="same_kind")

沿現有軸連接一係列數組。

參數：

a1, a2, …： 數組 序列

數組必須具有相同的形狀，除了對應於軸的維度(默認情況下為第一個)。

axis： 整數，可選

陣列將沿其連接的軸。如果axis為None，則數組在使用前會被展平。默認值為 0。

out： ndarray，可選

如果提供，則放置結果的目的地。形狀必須正確，與未指定 out 參數時連接將返回的形狀相匹配。

dtype： str 或 dtype

如果提供，目標數組將具有此 dtype。不能與 out 一起提供。

casting： {‘no’, ‘equiv’, ‘safe’, ‘same_kind’, ‘unsafe’}，可選

控製可能發生的數據類型轉換。默認為‘same_kind’。

返回：

res： ndarray

串聯數組。

注意：

當要連接的一個或多個數組是 MaskedArray 時，此函數將返回 MaskedArray 對象而不是 ndarray，但不會保留輸入掩碼。如果需要 MaskedArray 作為輸入，請改用屏蔽數組模塊中的 ma.concatenate 函數。

例子：

>>> a = np.array([[1, 2], [3, 4]])
>>> b = np.array([[5, 6]])
>>> np.concatenate((a, b), axis=0)
array([[1, 2],
       [3, 4],
       [5, 6]])
>>> np.concatenate((a, b.T), axis=1)
array([[1, 2, 5],
       [3, 4, 6]])
>>> np.concatenate((a, b), axis=None)
array([1, 2, 3, 4, 5, 6])

此函數不會保留 MaskedArray 輸入的屏蔽。

>>> a = np.ma.arange(3)
>>> a[1] = np.ma.masked
>>> b = np.arange(2, 5)
>>> a
masked_array(data=[0, --, 2],
             mask=[False,  True, False],
       fill_value=999999)
>>> b
array([2, 3, 4])
>>> np.concatenate([a, b])
masked_array(data=[0, 1, 2, 2, 3, 4],
             mask=False,
       fill_value=999999)
>>> np.ma.concatenate([a, b])
masked_array(data=[0, --, 2, 2, 3, 4],
             mask=[False,  True, False, False, False, False],
       fill_value=999999)