Find Index Of Maximum In Numpy Array Python Code