熊猫将一列的值与相同的索引合并到列表中

Question

我有一个如下的Pandas DF，我正在努力以漂亮的格式打印它... 有人可以告诉我如何将同一列值中的这两个值合并吗？

data = {'Animal':['DOG','CAT','CAT','BIRD'],
        'Color':['WHITE','BLACK','ORANGE','YELLOW']}
df = pd.DataFrame(data)
df

我希望print的结果完全如下：

DOG, WHITE
CAT, BLACK, ORANGE
BIRD, YELLOW

Answer 1

使用groupby和to_string

print(df.groupby('Animal')['Color'].agg(', '.join).to_string())


Animal
BIRD           YELLOW
CAT     BLACK, ORANGE
DOG             WHITE