python中numpy.repeat函数

numpy.repeat函数

numpy.repeat函数的主要作用：重复数组的元素。

用法

import numpy as np

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

参数

• a：给定数组；
• repeats：指定重复次数；
• axis：指定轴。

实例

实例1

b = np.arange(9)
b

输出：

array([0, 1, 2, 3, 4, 5, 6, 7, 8])

使用numpy.repeat函数：

np.repeat(b,10)

输出：

array([0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 2, 2,
       2, 2, 2, 2, 2, 2, 2, 2, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 4, 4, 4, 4,
       4, 4, 4, 4, 4, 4, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 6, 6, 6, 6, 6, 6,
       6, 6, 6, 6, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 8, 8, 8, 8, 8, 8, 8, 8,
       8, 8])

实例2

a = np.array([[0, 1, 2],[3, 4, 5]])
a

输出：

array([[0, 1, 2],
       [3, 4, 5]])

使用numpy.repeat函数：

np.repeat(a,2)

输出：

array([0, 0, 1, 1, 2, 2, 3, 3, 4, 4, 5, 5])

由实例2，可以看出，在不指定参数axis的情况下，numpy.repeat函数将二维数组中的每个元素都重复了两遍，并输出由原二维数组中的元素组成的新一维数组。

实例3

使用实例2中的数组a，并指定沿0进行元素重复：

np.repeat(a,2,axis=0)

输出：

array([[0, 1, 2],
       [0, 1, 2],
       [3, 4, 5],
       [3, 4, 5]])

沿1进行元素重复：

np.repeat(a,2,axis=1)

输出：

array([[0, 0, 1, 1, 2, 2],
       [3, 3, 4, 4, 5, 5]])