for x in list python get index