python comprehension with enumerate