python - 为 pandas 列创建类别计数字典

Question

我有以下数据框:

person    pets
John     [cat, dog]
Amy      [bird, fish, cat]
Dave     [cow, horse, dog]
Mary     [lamb, camel, rino]
Jim      [bird, dog]

我想要聚合 pets 列以查找每种宠物类型的出现情况。此示例的预期答案应该是:

{cat: 2, dog: 3, bird:2, fish:1, cow:1, horse:1, lamb: 1, camel: 1, rino:1}

除了逐行循环整个数据框之外，还有更优雅的方式来获取结果吗？谢谢!

Answer 1

通过仅使用内置函数，您可以执行以下操作:

a = [j for i in df['pets'] for j in i]

{i:a.count(i) for i in set(a)}

{'fish': 1,'bird': 2,'dog': 3,'camel': 1,'cat': 2,'lamb': 1,'horse': 1,'cow': 1,'rhino': 1}