Find Index In Array In Python