How To Return The Index Of An Array In Python