【Python学习】Numpy函数repeat和tile用法

2022-10-11 14:26:18

numpy数组用扩展函数repeattile,但是数组不能进行动态扩展,所以在调用上述函数进行扩展的时候,系统会重新分配新的空间进行存储扩展后的数据。

repeat函数

功能:可以对数组中的元素进行连续复制
用法:

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

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

1.Test1

Input:

import numpyas np
arr = np.arange(5)
arr_ = arr.repeat(3)
print('arr: ', arr)
print('arr_: ', arr_)

Output:

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

repeat函数不会修改原有的numpy数组,重复运算大致上可以这么理解,遍历原先数组的每一个元素,然后进行逐个元素的重复。

2.Test2

Input:

import numpy as np
arr = np.array([[10,20,30], [40,50,60]])print(arr.shape)
arr1 = arr.repeat([3,2], axis=0)
arr2 = arr.repeat([3,2,1] ,axis=1)print(arr1)print(arr2)

Output:

(2,3)[[10 20 30]
 [10 20 30]
 [10 20 30]
 [40 50 60]
 [40 50 60]][[10 10 10 20 20 30]
 [40 40 40 50 50 60]]

numpy数组的维度从左往右是原数组从外到内的顺序,例子中,最外层是两个元素,所以第一个维度是2。axis=0表示按照shape0位置进行重复,axis=1表示按照shape1位置进行重复。因此repeats的长度应该等于axis对应维度的数长度。

tile函数

功能:对整个数组进行复制拼接
用法:

  • numpy.tile(a, reps)

其中a为numpy数组,reps为数组的重复次数,注意该函数是不能直接使用numpy数组名进行调用。

Input:

import numpyas np
arr = np.array([1,2])
arr_ = np.tile(arr,2)
arr__ = np.tile(arr, (3,2))print(arr)print(arr_)print(arr__)

Output:

[12]
[1212]
[[1212]
 [1212]
 [1212]]

第一次是直接对原数组进行复制两次,第二次是首先对第一维重复三次,第二维重复两次,一共出现了6次原数组的copy。

参考文献:
http://blog.csdn.net/zyl1042635242/article/details/43052403

  • 作者:TwT520Ly
  • 原文链接:https://blog.csdn.net/TwT520Ly/article/details/79428360
    更新时间:2022-10-11 14:26:18