Return Index Of Element In Array Python