numpy.repeat#

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

重复数组中的每个元素

参数:

a类数组: 输入数组。
repeats整数或整数数组: 每个元素的重复次数。repeats 会被广播以适应给定轴的形状。
axisint, 可选: 沿其重复值的轴。默认情况下，使用扁平化的输入数组，并返回一个扁平化的输出数组。

返回:

repeated_arrayndarray: 输出数组，其形状与a相同，但在给定轴上除外。

另请参阅

tile: 平铺数组。
unique: 查找数组中的唯一元素。

示例

>>> import numpy as np
>>> 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]])