详解Numpy repeat()（重复数组元素）函数的作用与使用方法

Numpy库中的repeat()函数是用来对数组元素进行重复操作的方法。该方法会将原始数组的每个元素复制若干遍，生成一个新的重复数组。

使用方法

numpy.repeat(arr, repeat_times, axis=None)

参数说明：

• arr：要进行重复操作的数组；
• repeat_times：指定每个元素需要重复的次数；
• axis：指定操作的轴向。

示例1

import numpy as np
a = np.array([1, 2, 3])
b = np.repeat(a, 3)
print(b)

输出结果为：

array([1, 1, 1, 2, 2, 2, 3, 3, 3])

本例中，我们将数组a中的每个元素重复了三遍，形成了一个新的数组。

示例2

import numpy as np
a = np.array([[1, 2], [3, 4]])
b = np.repeat(a, 2, axis=0)
print(b)

输出结果为：

array([[1, 2],
       [1, 2],
       [3, 4],
       [3, 4]])

本例中，我们对二维数组a进行了重复操作，每个元素重复了两遍，结果生成了一个新的二维数组。在这个例子中，我们指定了axis=0，表示对每行进行重复操作。

总结

通过重复操作，我们可以快速扩充数据集，并且可以用于实现算法中的数据扩维和插值操作。在使用时，需要注意指定操作的轴向。