How To Find The Index Of An Array In Python