Numpy Index Of Maximum Value In Vector