为数据框中的每个客户编号创建字典

Question

我有以下客户互动数据

customer  interacted  
c1           i1  
c1           i2  
c1           i1  
c2           i3  
c2           i1

我想将此数据框转换为另一个数据框，其中一列为客户，第二列为存储交互历史的字典。

第二个df应该像

customer   interhist

    c1     {'i1': 2, 'i2': 1}  
    c2     {'i3':1, 'i1':1}

Answer 1

collections 模块中的 defaultdict 和 Counter 类对于此类问题非常有用。您可以尝试以下方法：

from collections import defaultdict, Counter

data_map = defaultdict(Counter)
for line in your_data_source:
    customer, interacted = line.split()
    data_map[customer][interacted] += 1

这将为您提供一个字典，将客户映射到另一个字典，将他们的交互映射到一个整数。