当前位置: 首页>>编程示例 >>用法及示例精选 >>正文


Python numpy.repeat()用法及代码示例

关于:
numpy.repeat(arr,repetitions, axis = None):重复数组的元素-arr。

参数:

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 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)

输出:

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 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,)

参考文献:
https://docs.scipy.org/doc/numpy/reference/generated/numpy.repeat.html



相关用法


注:本文由纯净天空筛选整理自 numpy.repeat() in Python。非经特殊声明,原始代码版权归原作者所有,本译文未经允许或授权,请勿转载或复制。