Python numpy.repeat函数方法的使用-CJavaPy

numpy.repeat

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

重复数组的元素。

参数：

a ：array_like

输入数组。

repeats ：int 或 int类型的array

每个元素的重复次数。广播重复以适应给定axis的shape。

axis ：int, 可选

重复值所沿的轴。默认情况下，使用扁平的输入数组，

并返回扁平的输出数组。

返回值：

repeated_array ：ndarray

输出阵列，其形状与a相同，但沿给定轴除外。

例子

1）一维数组的重复

import numpy as np

a = np.array([1, 2, 3])
print(np.repeat(a, 2))

2）使用不同的重复次数

import numpy as np

a = np.array([1, 2, 3])
print(np.repeat(a, [2, 3, 1]))

3）二维数组的重复

import numpy as np

a = np.array([[1, 2], [3, 4]])
print(np.repeat(a, 2))

4）沿特定轴重复

import numpy as np

a = np.array([[1, 2], [3, 4]])
print(np.repeat(a, 2, axis=0))  # 沿着行（axis=0）重复
print(np.repeat(a, 2, axis=1))  # 沿着列（axis=1）重复