Find Element Index In Array Python Initialize