Python Numpy Index Array Function In Google