转载自:https://blog.csdn.net/NockinOnHeavensDoor/article/details/80273268 (扩展用法)
https://blog.csdn.net/cetrol_chen/article/details/79147878(基本用法)
首先看一下这个函数
numpy.repeat(a, repeats, axis=None) 功能:
将矩阵A按照给定的axis将每个元素重复repeats次数 参数:
a:输入矩阵, repeats:每个元素重复的次数, axis:需要重复的维度 返回值:
输出矩阵
>>> np.repeat(3, 4)
array([3, 3, 3, 3]) #每个元素重复4次
>>> 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]]) #每个元素按照列重复3次
>>> np.repeat(x, [1, 2], axis=0)
array([[1, 2],
[3, 4],