python for each in list with index