代码之家 › 专栏 › 技术社区 › user9092346

汇总数据集并创建新变量

pandas python

user9092346 · 技术社区 · 5 年前

我有一个数据集,它按国家、季度、部门、交易类型和值列出各个交易。我想根据前三个变量对其进行总结,但为其他两个变量创建新的列。数据集如下所示:

Country   Quarter   Division    Type    Value
   A         1        Sales      A        50
   A         2        Sales      A       150
   A         3        Sales      B        20
   A         1        Sales      A       250
   A         2        Sales      B        50
   A         3        Sales      B        50
   A         2      Marketing    A        50

现在,我想聚合数据,以获得作为新变量的按类型划分的事务数。按前三个变量分组的事务总数很容易:

df.groupby(['Country', 'Quarter', 'Division'], as_index=False).agg({'Type':'count', 'Value':'sum'})

但是,我希望我的新数据框架如下所示:

Country   Quarter   Division    Type_A   Type_B     Value_A   Value_B
   A         1        Sales        2        0         300         0
   A         2        Sales        1        1         150        50
   A         3        Sales        0        2           0        70
   A         2      Marketing      1        0          50         0

我该怎么做?

1 回复 | 直到 5 年前

jezrael 5 年前

在后面指定列 groupby 有元组 agg 函数,用于具有聚合函数的新列名称,然后按 DataFrame.unstack 最后一次转换 MultiIndex 按列显示 map :

df1 = (df.groupby(['Country', 'Quarter', 'Division', 'Type'])['Value']
         .agg([('Type','count'), ('Value','sum')])
         .unstack(fill_value=0))
df1.columns = df1.columns.map('_'.join)
df1 = df1.reset_index()
print (df1)
  Country  Quarter   Division  Type_A  Type_B  Value_A  Value_B
0       A        1      Sales       2       0      300        0
1       A        2  Marketing       1       0       50        0
2       A        2      Sales       1       1      150       50
3       A        3      Sales       0       2        0       70

推荐文章

Google User · Django管理员在`list_display中未显示`creation_date`字段`

3 月前

user29747013 · 如何创建一个新的数据框架,其中包含原始数据框架中列的聚合列?

3 月前

ÎÎÎ½Î· ÎÎ®Î¹Î½Î¿Ï · Python lxml.html语法错误:使用lxml find时XPATH的谓词无效

3 月前

user29715306 · from_users=和chats=电视节目中的差异

3 月前

Redshoe · 当执行numpy.genfromtxt()时,python是否会读取文件的所有行?

3 月前

RASEL MAHMUD · 为什么以及如何在is_even()函数内的IF条件中递归X变量在满足0后递增?[副本]

3 月前

prayner · 更新嵌套字典包含列表中的项

3 月前

Bringo Jr · 我可以在O(n)中解决这个问题吗?

3 月前

Dave · 如何在for循环中修改列表值

3 月前

Shukurullox Komiljonov · 从记录中获得相互和解。使用SQL

3 月前