np.repeat用法

np.repeat是numpy库中用于重复数组元素的函数。其用法如下：

np.repeat(a, repeats, axis=None)

其中，a表示要重复的数组；repeats表示每个元素需要重复的次数；axis表示要沿着哪个方向重复。如果没有指定axis参数，则会将整个数组展平并重复。

例如，我们可以这样使用np.repeat函数：

import numpy as np

a = np.array([1, 2, 3])
b = np.repeat(a, 2)
c = np.repeat(a, 3, axis=0)

print(b) # 输出 [1 1 2 2 3 3]
print(c) # 输出 [[1 1 1] [2 2 2] [3 3 3]]

在这个例子中，我们首先创建了一个数组a，然后使用np.repeat函数将数组a中的每个元素重复2次，得到了一个新的数组b。接着，我们使用np.repeat函数将数组a沿着axis=0的方向重复3次，得到了一个新的数组c。