How To Return Index Of Array Python