numpy.repeat

原文:https://docs.scipy.org/doc/numpy/reference/generated/numpy.repeat.html

译者:飞龙 UsyiyiCN

校对:(虚位以待)

numpy.repeat(a, repeats, axis=None)[source]

重复数组的元素。

参数:

a:array_like

输入数组。

repeats:int或数组的整数

每个元素的重复次数。重复被广播以适合给定轴的形状。

axis:int,可选

沿其重复值的轴。默认情况下,使用扁平输入数组,并返回一个扁平输出数组。

返回:

repeated_array:ndarray

输出数组与a具有相同的形状,除了沿给定轴。

也可以看看

tile
平铺数组。

例子

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