Python Numpy Array Get Max Subscription