Python Numpy Index Array Functions In Java