How To Use Index In Numpy Which Function