numpy索引切片

import numpy as np
#获取数组中(0,0)，(1,1)和(2,0)位置处的元素
x = np.array([[1,2],  [3,4],  [5,6]]) 
y = x[[0,1,2],  [0,1,0]]
y  >>>   array([1,4,5])

#获取4X3数组中的四个角的元素。(0,0)，(0,2),(4,0)，(4,2)
x = np.array([[0,1,2],[3,4,5],[6,7,8],[9,10,11],[12,13,14]]) 
rows = np.array([[0,0],[4,4]])
cols = np.array([[0,2],[0,2]]) 
y = x[rows,cols] 
y >>>    array([[ 0,  2],
                [12, 14]])

#2维数组，获取(0,0,1),(1,1,1),(1,0,1)
b = np.array([[[1,2],[3,4],[5,6]],[[7,8],[9,10],[11,12]]])
c = b[[0,1,1],[0,1,0],[1,1,1]]
c  >>>    array([ 2, 10,  8])

 

 

 

 

根据索引对ndarray切片，作图便于理解。