numpy.repeat(重复维度数据)

如果本地没有numpy的环境，可以使用以下命令进行安装

pip install numpy

conda install numpy

介绍

numpy.repeat 函数是 NumPy 库中用于重复数组元素的函数。

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

其中：

• a：要重复的输入数组。
• repeats：指定每个元素重复的次数的数组。repeats应为整数，或者是与输入数组形状相同的数组。
• axis：指定沿着哪个轴进行重复。默认情况下，数组将被扁平化。

该函数返回一个新的数组，其中的元素是原始数组 a 中的元素按照 repeats 中的规定重复生成的。

示例

import numpy as np

# 示例 1: 一维数组的重复
arr = np.array([1, 2, 3])
repeated_arr = np.repeat(arr, 3)
print(repeated_arr)
# 输出: [1 1 1 2 2 2 3 3 3]

# 示例 2: 多维数组的重复
arr_2d = np.array([[1, 2], [3, 4]])
repeated_arr_2d = np.repeat(arr_2d, 2, axis=1)
print(repeated_arr_2d)
# 输出:
# [[1 1 2 2]
#  [3 3 4 4]]

# 示例 3: 使用不同的重复次数
repeats_arr = np.array([1, 2])
repeated_arr_2d_custom = np.repeat(arr_2d, repeats_arr, axis=0)
print(repeated_arr_2d_custom)
# 输出:
# [[1 2]
#  [3 4]
#  [3 4]]

在这些示例中，可以看到 numpy.repeat 的功能是将数组的元素按照给定的规则重复生成新的数组。具体的重复次数由 repeats 参数指定，可以是一个整数，也可以是一个形状相同的数组。