numpy之array获取运算总结

获取

numpy的数组获取运算比较容易混淆,这里会详细解释各种运算,
以4*4的二维数组为例:

import numpy as np
array = array = np.arange(16).reshape(4,4)


array[1][1]
输出:5
解释:在一维数组中,array[1]代表着获取索引为1的元素;在二维数组中,array2[1]代表着获取索引为1的行,而这行是一个一维数组;所以array[1][1]是间接获取元素的方法,先是获取第二行的数组,然后获取该一维数组的第二个元素。


array[1,1]
输出:5
解释:[1,1]是二维数组直接获取元素的运算,意思是获取第2行第2列的元素值。


array[1:-1][:1]
输出:[[4 5 6 7]]
解释:-1是指倒数第一,所以array[1:-1]是获取第2行到倒数第1行(不包括)的数组,结果是一个二维数组[[4 5 6 7][8 9 10 11]];[:1]是指[0:1],所以是获取第1行到第2行(不包括)的数组,结果是一个二维数组[[4 5 6 7]]。


array[1:-1,1]
输出:[5 9]
解释:运算的意思是获取第2行到倒数第1行(不包括)的第2个元素


array[1:-1,:1]
输出:[[4][8]]
解释:运算的意思是获取第2行到倒数第1行(不包括)的第1个元素到第2个元素(不包括)


array[::2,:2]
输出:[[0 1][8 9]]
解释:::2意思是从第1行开始,每隔1行获取,并且获取第1个元素到第3个元素(不包括)


array[1::2][::2]
输出:[[4 6][12 14]]
解释:1::2意思是从索引1开始,每隔1行获取,并且获取从索引0开始,每隔一个元素的元素。

总结:
[n] : 获取索引为n的元素
[n:m]:获取索引n到m-1的元素
[n::m]:获取索引n开始,相隔m-1索引的元素

猜你喜欢

转载自blog.csdn.net/u012587024/article/details/82020527
今日推荐