Python Numpy Index Array Function In Java