numpy.repeat()函数的作用和用法详解

numpy.repeat()函数用于将数组中的某个元素重复指定次数，可以实现将一维数组中的元素重复一定次数，或者将多维数组中的某个轴元素重复一定次数，生成新的数组。

使用方法

numpy.repeat(a, repeats, axis=None)，参数a为数组，repeats为重复次数，axis为指定轴，默认值为None，表示将数组中的所有元素重复指定次数。

# 一维数组
import numpy as np
a = np.array([1, 2, 3])
b = np.repeat(a, 3)
print(b)  # [1 1 1 2 2 2 3 3 3]

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

numpy.repeat()函数还支持将列表作为参数，可以将列表中元素重复指定次数，生成新的列表。

# 将列表中元素重复指定次数
import numpy as np
a = [1, 2, 3]
b = np.repeat(a, 3)
print(b)  # [1 1 1 2 2 2 3 3 3]