1. 程式人生 > >numpy中實用但不常見的方法(2)np.repeat

numpy中實用但不常見的方法(2)np.repeat

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], [3, 4]]) #第1行元素重複1次,第2行元素重複2