numpy.repeat
numpy.repeat(a, repeats, axis=None) [source]
重复数组的元素。
参数 : | a :array_like 输入数组。 repeats : 每个元素的重复次数。广播重复以适应给定axis的shape。 axis : 重复值所沿的轴。 默认情况下,使用扁平的输入数组, 并返回扁平的输出数组。 |
返回值 : | repeated_array : 输出阵列,其形状与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]])