Get Index Of Maximum In Numpy There Is A Sub Module