numpy数组扩展函数repeat和tile用法

numpy数组扩展函数有repeat和tile,由于数组不能进行动态扩展,故函数调用之后都重新分配新的空间来存储扩展后的数据。

repeat函数功能:对数组中的元素进行连续重复复制

用法有两种:

1) numpy.repeat(a, repeats, axis=None)

2) a.repeats(repeats, axis=None)

其中a为数组,repeats为重复的次数,axis表示数组维度

>>>import numpy as np

>>> a = np.arange(10)  

>>> a  

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

>>> a.repeat(5)  

array([0, 0, 0, 0, 0, 1, 1, 1, 1, 1, 2, 2, 2, 2, 2, 3, 3, 3, 3, 3, 4, 4, 4,  

        4, 4, 5, 5, 5, 5, 5, 6, 6, 6, 6, 6, 7, 7, 7, 7, 7, 8, 8, 8, 8, 8, 9,  

        9, 9, 9, 9])  

>>> a

np.array([0, 1, 2, 3, 4, 5, 6, 7, 8, 9])  #a数组的内容没改变

 

>>> a=np.array([10,20])

>>>a

array([10,20])

>>> a.repeat([3,2])   

array([10, 10, 10, 20, 20])  

>>> repeat(a,[3,2])   #  对a数组中的对应元素进行重复复制,需要注意的是len(repeats)==a.shape[axis]

array([10, 10, 10, 20, 20]) 

 

>>> a=np.array([[10,20],[30,40]])  

>>> a.repeat([3,2],axis=0)  

array([[10, 20],  

         [10, 20],  

         [10, 20],  

         [30, 40],  

         [30, 40]])  

>>> a.repeat([3,2],axis=1)  

array([[10, 10, 10, 20, 20],  

         [30, 30, 30, 40, 40]]) 

 

tile函数功能:对整个数组进行复制拼接

用法:numpy.tile(a, reps)

其中a为数组,reps为重复的次数

>>> np.tile(a,2)  

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

>>> a

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

 

>>> a=np.array([10,20])

>>>a

array([10,20])

>>>np.tile(a, (3,2)) #构造3*2个copy

array([[10, 20, 10, 20],  

          [10, 20, 10, 20],  

          [10, 20, 10, 20]])