python - 如何对同一列中的字段进行减法 - Python

Question

我有以下初始数据框:

Country      Gender     Q1
Australia    M          Increase
Australia    F          Increase
Australia    M          Increase
Australia    M          Decrease
India        M          Decrease
India        F          Increase
India        M          Decrease
South Africa F          Decrease
UK           F          Increase

这里我们计算每个过滤器的 Q1:

Country       Gender      Q1         Count(Q1)
Australia       M         Increase    2
Australia       F         Increase    1
Australia       M         Decrease    1
India           M         Decrease    2
India           F         Increase    1
South Africa    F         Decrease    1
UK              F         Increase    1

现在对于同一个过滤器，我需要减去计数(增加)和计数(减少)ans 存储在 Q1“Net”的新选项中。因此 Net = count(increase) - count(decrease)。最终的数据框看起来像这样:

这在 Python 中可行吗？

问候，

Answer 1

将 default_count 列添加到初始数据框，然后使用 pivot_table

   df=pd.DataFrame({'country':['Australia','Australia','Australia','Australia','India','India','India','South Africa','UK'],'gender':['M','F','M','M','M','F','M','F','F'],'Q1':['Increase','Increase','Increase','Decrease','Decrease','Increase','Decrease','Decrease','Increase'],'default_count':['1','1','1','1','1','1','1','1','1']})

   df_pivot=df.pivot_table(index=['country','gender'],columns='Q1',aggfunc='count',values='default_count').reset_index().fillna(0)

   df_pivot['Net']=df_pivot['Increase']-df_pivot['Decrease']