Python Numpy Index Of Max Value In Vector