极客教程numpy.repeat
numpy.repeat()函数用于重复数组中的元素。您可以使用此函数重复数组的特定元素或整个数组。
使用方法
import numpy as np
arr = np.array([1, 2, 3, 4, 5])
repeated_arr = np.repeat(arr, 3)
print(repeated_arr)
运行结果：
[1 1 1 2 2 2 3 3 3 4 4 4 5 5 5]
在上面的示例代码中，我们首先导入numpy库，然后定义一个包含整数的NumPy数组arr。接下来，我们使用np.repeat()函数将arr中的每个元素重复3次，并将结果存储在repeated_arr中。最后，我们打印repeated_arr的值，以查看重复后的数组。
参数说明
如果要重复整个数组，可以将repeats参数设置为一个大于1的整数。如果要重复特定元素，可以将repeats参数设置为一个包含重复次数的数组。
arr = np.array([[1, 2], [3, 4]])
repeated_arr = np.repeat(arr, [2, 3], axis=0)
print(repeated_arr)
运行结果：
[[1 2]
[1 2]
[3 4]
[3 4]
[3 4]]
在上面的示例代码中，我们定义了一个二维NumPy数组arr，然后使用np.repeat()函数重复arr的每一行，第一行重复2次，第二行重复3次。最后，我们打印了重复后的数组repeated_arr。
总结
numpy.repeat()函数是一个强大的功能，可以帮助我们重复数组的元素。无论是重复整个数组还是重复特定元素，都可以通过这个函数轻松实现。