Numpy中的repeat函数使用

numpy.repeat()函数是用于将数组中的元素重复指定的次数，生成一个新的数组。该函数的语法如下：

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

其中，参数a是要复的数组参数repeats是重复的次数，参数axis是指定重复的轴。如果不指定axis，则将展平重复。

下面是两个示例，演示如何使用numpy.repeat()函数。

示例1：重复数组中的元素

import numpy as np

# 创建一个数组
a = np.array([1, 2, 3])

# 重复每个元素3次
b = np.repeat(a, 3)

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

在上面的示例中，我们创建了一个数组a，然后使用numpy.repeat()函数将每个元素重复次，生成一个新的数组b。

示例2：重复数组中的行或列

import numpy as np

# 创建一个二维数组
a = np.array([[1, 2], [3, 4]])

# 沿着行重复每行2次
b = np.repeat(a, 2, axis=0)

# 沿着列重复每列3次
c = np.repeat(a, 3, axis=1)

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

在上面的示例中，我们创建了一个二维数组a，然后使用numpy.repeat()函数沿着行重复每行2次，沿着列重复每列3次，生成两个新的数组b和c。