python for element in list with index