python numpy寻找二维数组最大值index

import numpy as np


a = np.array([[1, 2, 3],
           [2, 3, 15],
           [3, 4, 5]])
print(a)
i = np.where(a==np.max(a[:, 2]))
print(i)
print(i[0])

 

版权声明：本文为qq_41816368原创文章，遵循CC 4.0 BY-SA版权协议，转载请附上原文出处链接和本声明。