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