直接获取max:
ccc=np.max(box, axis=0)
print(ccc)
ccc=np.max(box, axis=1)
print(ccc)
ccc = box.max(axis=1)
print(ccc)
二维,每行最大值索引:
box = np.array([[3, 2, 1],[ 0, 4, 5]])
cc= np.argmax(box,axis=1)
for i in range(box.shape[0]):
print(box[i][cc[i]])
print(box[0][cc[0]])
print(box[1][cc[1]])