📜  Python中的 numpy.repeat()

📅  最后修改于: 2022-05-13 01:54:27.483000             🧑  作者: Mango

Python中的 numpy.repeat()

numpy.repeat()函数重复数组的元素——arr。句法 :

numpy.repeat(arr, repetitions, axis = None)

参数 :

array       : [array_like]Input array. 
repetitions : No. of repetitions of each array elements along the given axis.
axis        : Axis along which we want to repeat values. By default, it returns 
           a flat output array.

返回 :

An array with repetitions of array - arr elements as per repetitions, number of times 
we want to repeat arr  

代码 1:

Python
# Python Program illustrating
# numpy.repeat()
 
import numpy as geek
 
#Working on 1D
arr = geek.arange(5)
print("arr : \n", arr)
 
repetitions = 2
a = geek.repeat(arr, repetitions)
print("\nRepeating arr 2 times : \n", a)
print("Shape : ", a.shape)
 
repetitions = 3
a = geek.repeat(arr, repetitions)
print("\nRepeating arr 3 times : \n", a)
# [0 0 0 ..., 4 4 4] means [0 0 0 1 1 1 2 2 2 3 3 3 4 4 4]
# since it was long output, so it uses [ ... ]
print("Shape : ", a.shape)


Python
# Python Program illustrating
# numpy.repeat()
 
import numpy as geek
 
arr = geek.arange(6).reshape(2, 3)
print("arr : \n", arr)
 
repetitions = 2
print("\nRepeating arr : \n", geek.repeat(arr, repetitions, 1))
print("arr Shape : \n", geek.repeat(arr, repetitions).shape)
 
 
repetitions = 2
print("\nRepeating arr : \n", geek.repeat(arr, repetitions, 0))
print("arr Shape : \n", geek.repeat(arr, repetitions).shape)
    
repetitions = 3
print("\nRepeating arr : \n", geek.repeat(arr, repetitions, 1))
print("arr Shape : \n", geek.repeat(arr, repetitions).shape)


输出 :

arr : 
 [0 1 2 3 4]

Repeating arr 2 times : 
 [0 0 1 1 2 2 3 3 4 4]
Shape :  (10,)

Repeating arr 3 times : 
 [0 0 0 ..., 4 4 4]
Shape :  (15,)

代码 2:

Python

# Python Program illustrating
# numpy.repeat()
 
import numpy as geek
 
arr = geek.arange(6).reshape(2, 3)
print("arr : \n", arr)
 
repetitions = 2
print("\nRepeating arr : \n", geek.repeat(arr, repetitions, 1))
print("arr Shape : \n", geek.repeat(arr, repetitions).shape)
 
 
repetitions = 2
print("\nRepeating arr : \n", geek.repeat(arr, repetitions, 0))
print("arr Shape : \n", geek.repeat(arr, repetitions).shape)
    
repetitions = 3
print("\nRepeating arr : \n", geek.repeat(arr, repetitions, 1))
print("arr Shape : \n", geek.repeat(arr, repetitions).shape)

输出 :

arr : 
 [[0 1 2]
 [3 4 5]]

Repeating arr : 
 [[0 0 1 1 2 2]
 [3 3 4 4 5 5]]
arr Shape : 
 (12,)

Repeating arr : 
 [[0 1 2]
 [0 1 2]
 [3 4 5]
 [3 4 5]]
arr Shape : 
 (12,)

Repeating arr : 
 [[0 0 0 ..., 2 2 2]
 [3 3 3 ..., 5 5 5]]
arr Shape : 
 (18,)