Get Index Of Maximum Value In Numpy Which Function