python numpy index of max code example Example: numpy get max index >>> a.argmax(axis=0) array([1, 1, 0])