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