python 高维数组切片

import numpy as np o = np.array([[[1, 2], [5,4], [7, 8]], [[10, 11], [13, 14], [17,16]]]) print("o") print(o) print("##############") print("o[:, :]") print(o[:, :]) argo=o.argmax(axis=2) print("##############") print("argo") print(argo) print("##############") print("argo[:, :]") print(argo[:, :]) print(argo[:, :]==1)

 

控制台输出结果：

o [[[ 1  2]   [ 5  4]   [ 7  8]]  [[10 11]   [13 14]   [17 16]]] ############## o[:, :] [[[ 1  2]   [ 5  4]   [ 7  8]]  [[10 11]   [13 14]   [17 16]]] ############## argo [[1 0 1]  [1 1 0]] ############## argo[:, :] [[1 0 1]  [1 1 0]] [[ True False  True]  [ True  True False]]