numpy repeat

numpy repeat

在NumPy中，numpy.repeat函数可以用来沿着指定的轴重复数组中的元素。

语法

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

参数说明：

• a：要重复的数组。
• repeats：整数，指定每个元素要重复的次数。
• axis：整数，指定重复的轴。

示例

1. 一维数组重复

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]

在这个示例中，一维数组[1, 2, 3, 4, 5]中的每个元素被重复了3次。

2. 多维数组重复

import numpy as np

arr = np.array([[1, 2], [3, 4]])
repeated_arr = np.repeat(arr, 2, axis=0)

print(repeated_arr)

输出：

[[1 2]
 [1 2]
 [3 4]
 [3 4]]

在这个示例中，二维数组[[1, 2], [3, 4]]中的每一行被重复了2次沿着axis=0的方向。

3. 指定轴重复

import numpy as np

arr = np.array([[1, 2], [3, 4]])
repeated_arr = np.repeat(arr, 2, axis=1)

print(repeated_arr)

输出：

[[1 1 2 2]
 [3 3 4 4]]

在这个示例中，二维数组[[1, 2], [3, 4]]中的每一列被重复了2次沿着axis=1的方向。

注意事项

• 重复次数可以为0，即按照指定的轴，每个元素都重复0次。
• 重复的轴的长度会相应扩展。

以上就是关于numpy.repeat函数的详细介绍和示例。