Python numpy repeat用法及代碼示例

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

用法:

numpy.repeat(a, repeats, axis=None)

重複數組的元素。

參數：

a： array_like

輸入數組。

repeats： int 或整數數組

每個元素的重複次數。廣播重複以適應給定軸的形狀。

axis： 整數，可選

沿其重複值的軸。默認情況下，使用扁平化的輸入數組，並返回一個扁平的輸出數組。

返回：

repeated_array： ndarray

輸出數組的形狀與a，除了沿給定軸。

例子：

>>> np.repeat(3, 4)
array([3, 3, 3, 3])
>>> x = np.array([[1,2],[3,4]])
>>> np.repeat(x, 2)
array([1, 1, 2, 2, 3, 3, 4, 4])
>>> np.repeat(x, 3, axis=1)
array([[1, 1, 1, 2, 2, 2],
       [3, 3, 3, 4, 4, 4]])
>>> np.repeat(x, [1, 2], axis=0)
array([[1, 2],
       [3, 4],
       [3, 4]])