Numpy Array Get Index Of Max Value In Vector