Python Numpy Index Array Java