How to get the index of a maximum element in a NumPy array along one axis November 22, 2022 by Tarik >>> a.argmax(axis=0) array([1, 1, 0])