Java Get Index Of Max Value In Numpy