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]] |