How To Return Index Of Array In Python