如何在Python中分离列表?

Question

fruit,apple
fruit,tomato
vegetable,carrot
fruit,pear
vegetable,celery
vegetable,cabbage
vegetable,cauliflower
fruit,banana

我想像这样把单子和水果蔬菜分开

水果

^{pr2}$

蔬菜

carrot
celery
cabbage
cauliflower

这是我迄今为止所做的代码

infile = open("hw16.txt","r")
lines = infile.readlines()
infile.close()

print lines
x = []
y = []
for i in range(0, len(lines)):
    tokens = lines[i].rstrip('\n').split(",")
    x.append(str(tokens[0]))
    y.append(str(tokens[1]))
print  x
print y

谢谢你

Answer 1

与其他答案相似

mydict = {'fruit' : [], 'vegetable' : []}

for line in file:   
    key, val = line.rstrip().split(',')     
    mydict[key].append(val)

Answer 2

vegetables = []
fruits = []

for line in file:
    type , name = line.strip().split(',') 
    if type == 'fruit':
        fruits.append(name)
    else:
        vegetables.append(name)

输出：

^{pr2}$

Answer 3

from collections import defaultdict
catergories = defaultdict(list)
with open("hw16.txt") as f:
  for line in f:
    cat, _, name = line.rstrip().partition(",")
    catergories[cat].append(name)
for cat, values in catergories.iteritems():
  print cat
  for x in values:
    print x
  print