how to select an element from a list in python