本篇内容介绍了“np.repeat()函数如何使用”的有关知识,在实际案例的操作过程中,不少人都会遇到这样的困境,接下来就让小编带领大家学习一下如何处理这些情况吧!希望大家仔细阅读,能够学有所成!
使用np.repeat()展平二维数组
代码如下:
import numpy as npclass Debug: def __init__(self): self.array1 = np.array([[1, 2], [3, 4]]) def mainProgram(self): print("The value of array1 is: ") print(self.array1) print("The repeated array is: ") array2 = np.repeat(self.array1, repeats=1) print(array2)if __name__ == '__main__': main = Debug() main.mainProgram()"""The value of array1 is: [[1 2] [3 4]]The repeated array is: [1 2 3 4]"""
我们可以看到我们输入的是一个二维数组,当我们设定 repeats 值为 1 时,输出结果变成了一个一维数组,因此这时的 np.repeats 函数类似numpy.ndarray.flatten()函数的功能。
np.repeat()函数的坐标轴问题
接下来我们研究一下关于 axis 坐标轴的问题。
二维
对于数组是二维的情况,代码如下:
import numpy as npclass Debug: def __init__(self): self.array1 = np.array([[1, 2], [3, 4]]) def mainProgram(self): print("The value of array1 is: ") print(self.array1) print("The array2 is: ") array2 = np.repeat(self.array1, repeats=2, axis=0) print(array2) print("The array3 is: ") array3 = np.repeat(self.array1, repeats=2, axis=1) print(array3)if __name__ == '__main__': main = Debug() main.mainProgram()"""The value of array1 is: [[1 2] [3 4]]The array2 is: [[1 2] [1 2] [3 4] [3 4]]The array3 is: [[1 1 2 2] [3 3 4 4]]"""
我们可以看到,axis=0 时表示沿着y方向重复,axis=1 时表示沿着x方向重复。我们可以对比numpy数组的坐标轴表示,二维时,坐标轴为 (y, x),从左向右第一个参数 0 便代表 y 轴,1 代表 x轴。
三维
接下来我们研究一下数组是三维的情况,代码如下:
import numpy as npclass Debug: def __init__(self): self.array1 = np.array([[[1, 2], [3, 4]], [[5, 6], [7, 8]]]) def mainProgram(self): print("The value of array1 is: ") print(self.array1) print("The array2 is: ") array2 = np.repeat(self.array1, repeats=2, axis=0) print(array2) print("The array3 is: ") array3 = np.repeat(self.array1, repeats=2, axis=1) print(array3) print("The array4 is: ") array4 = np.repeat(self.array1, repeats=2, axis=2) print(array4)if __name__ == '__main__': main = Debug() main.mainProgram()"""The value of array1 is: [[[1 2] [3 4]] [[5 6] [7 8]]]The array2 is: [[[1 2] [3 4]] [[1 2] [3 4]] [[5 6] [7 8]] [[5 6] [7 8]]]The array3 is: [[[1 2] [1 2] [3 4] [3 4]] [[5 6] [5 6] [7 8] [7 8]]]The array4 is: [[[1 1 2 2] [3 3 4 4]] [[5 5 6 6] [7 7 8 8]]]"""
我们可以看到,axis=0 对应与沿着z轴重复,axis=1 对应沿着y轴重复,axis=2 对应沿着x轴重复。对比numpy坐标轴的表示,我们知道三维坐标轴为 (z, y, x),所以从左向右,0 对应z轴,1 对应 y轴,2 对应 x 轴。
“np.repeat()函数如何使用”的内容就介绍到这里了,感谢大家的阅读。如果想了解更多行业相关的知识可以关注编程网网站,小编将为大家输出更多高质量的实用文章!