Python Find Index In Np Array