Get Index Of Max Value In Numpy There Is A Sub Module