Python numpy.repeat函数方法的使用

手机查看 2020-10-30

NumPy（Numerical Python的缩写）是一个开源的Python科学计算库。使用NumPy，就可以很自然地使用数组和矩阵。NumPy包含很多实用的数学函数，涵盖线性代数运算、傅里叶变换和随机数生成等功能。本文主要介绍一下NumPy中repeat方法的使用。

numpy.repeat

numpy.repeat(a, repeats, axis=None) [source]

重复数组的元素。

参数：

a ：array_like

输入数组。

repeats ：int 或 int类型的array

每个元素的重复次数。广播重复以适应给定axis的shape。

axis ：int, 可选

重复值所沿的轴。默认情况下，使用扁平的输入数组，

并返回扁平的输出数组。

返回值：

repeated_array ：ndarray

输出阵列，其形状与a相同，但沿给定轴除外。

例子

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

文档：numpy.repeat.html