python-3.x - 将每个组乘以 python groupby 中列表中的特定值

Question

我有以下数据框和列表。

d = ({
    'Item' : ['a','a','a','b','b','b','b'],                 
    'Count' : [0,1,2,3,4,5,6],                                      
    })

df = pd.DataFrame(data = d)

list_factor=[10,100] # a list of factors with same length as the unique Items in dataframe.

现在我想对于每个项目，将计数乘以与该项目相对应的 list_factor 元素。例如，a 中的所有计数将乘以 10，b 中的所有计数将乘以 100。因此最终的数据框将如下所示:

'Item' : ['a','a','a','b','b','b','b'],                 
'Count' : [0,10,20,300,400,500,600],                                      
    

我尝试使用 groupby 函数并向其应用乘法函数，但我只能对常量值执行此操作，而不能对列表中的值执行此操作。这是我所做的:

df.groupby('Item')['Count'].apply(lambda x : x*list_factor[?!])

这只是一个示例数据框以及我想要达到的目标。实际上，我有一个包含 1000 个不同项目的数据框，因此 list_factor 的长度为 1000。(数据框的总行数为 40000)

Answer 1

您可以使用factorize来提取Item的顺序，然后使用numpy索引来传播因子:

factors = np.array(list_factor)[df.Item.factorize()[0]]

df['Count'] *= factors

输出:

  Item  Count
0    a      0
1    a     10
2    a     20
3    b    300
4    b    400
5    b    500
6    b    600