python for in list with index