get index if element in list python