numpy.repeat

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()函数是一个强大的功能，可以帮助我们重复数组的元素。无论是重复整个数组还是重复特定元素，都可以通过这个函数轻松实现。