python - Pandas:聚合列的值

Question

我有以下数据框:

name      pet
----------------
John      ['cat']
Mary      ['cat','dog','bird']
Ann       ['bird','rat']
Dave      ['cow','dog']

对于每个人，pet 列是动物列表。我需要获得所有宠物的最终列表(无重复):

final_list = ['cat', 'dog', 'bird', 'rat', 'cow']

有没有更优雅的方法来实现这一点(比例如天真地逐行循环数据帧)？

Answer 1

您可以使用tolist函数获取所有值的列表，使用itertools.chain将它们展平，然后转换为集合 获取唯一值:

import itertools

dfList = df['pet'].tolist()
final_list = list(set(itertools.chain.from_iterable(dfList)))

print(final_list)
>>> ['cat', 'dog', 'bird', 'rat', 'cow']