Python 使用特定条件在dataframe中创建汇总行

Python 使用特定条件在dataframe中创建汇总行,python,pandas,dataframe,aggregation,Python,Pandas,Dataframe,Aggregation,假设我有以下数据框,并且我正在尝试对结果进行后处理以生成我的(现在为空)摘要行: code entry_type value1 value2 value3 value4 1 A Holding 1.1 1.2 1.3 1.4 2 A Holding 2.1 2.2 2.3 2.4 3 B Holding 3.1 3.2 3.3 3.

假设我有以下数据框,并且我正在尝试对结果进行后处理以生成我的(现在为空)摘要行:

    code    entry_type  value1  value2  value3  value4
1   A       Holding     1.1     1.2     1.3     1.4
2   A       Holding     2.1     2.2     2.3     2.4
3   B       Holding     3.1     3.2     3.3     3.4
4   C       Holding     4.1     4.2     4.3     4.4
5   C       Holding     5.1     5.2     5.3     5.4
6   A       Summary     nan     nan     nan     nan
7   C       Summary     nan     nan     nan     nan
8   B       Summary     nan     nan     nan     nan
基本上,我希望汇总行中的value1-value4是每个代码中持有的股份的总和:

    code    entry_type  value1  value2  value3  value4
1   A       Holding     1.1     1.2     1.3     1.4
2   A       Holding     2.1     2.2     2.3     2.4
3   B       Holding     3.1     3.2     3.3     3.4
4   C       Holding     4.1     4.2     4.3     4.4
5   C       Holding     5.1     5.2     5.3     5.4
6   A       Summary     3.2     3.4     3.6     3.8
7   C       Summary     9.2     9.4     9.6     9.8
8   B       Summary     3.1     3.2     3.3     3.4
我尝试了几行代码,并得出以下结论:

set = df[df['entry_type']=="Holding"].groupby('code')[['value1', 'value2', 'value3', 'value4']].sum()
这将产生:

        value1  value2  value3  value4
code
    A   3.2     3.4     3.6     3.8
    B   3.1     3.2     3.3     3.4
    C   9.2     9.4     9.6     9.8
但是,我不确定如何将其应用回原始数据帧,特别是因为代码顺序不一定与原始数据帧相同。你对如何应用这个有什么想法吗?还是更好的方法?(注意-其他列中的摘要行中已经存在大量附加数据,因此我不能直接内联生成新行)。

这似乎有助于:

df1  = df[df['entry_type']=="Holding"]
         .groupby('code')[['value1', 'value2', 'value3', 'value4']].sum()
#print (df1)

#if need filter `df` for only rows with Holding use boolean indexing
print (pd.concat([df[df['entry_type']=="Holding"].set_index('code'), df1])
         .fillna({'entry_type':'Summary'})
         .reset_index())

  code entry_type  value1  value2  value3  value4
0    A    Holding     1.1     1.2     1.3     1.4
1    A    Holding     2.1     2.2     2.3     2.4
2    B    Holding     3.1     3.2     3.3     3.4
3    C    Holding     4.1     4.2     4.3     4.4
4    C    Holding     5.1     5.2     5.3     5.4
5    A    Summary     3.2     3.4     3.6     3.8
6    B    Summary     3.1     3.2     3.3     3.4
7    C    Summary     9.2     9.4     9.6     9.8
另一种可能的解决方案是将
NaN
替换为
df1
,并将
df
的值对齐为
index

print (df.set_index('code')
         .combine_first(df1)
         .sort_values(['entry_type'])
         .reset_index())

  code entry_type  value1  value2  value3  value4
0    A    Holding     1.1     1.2     1.3     1.4
1    A    Holding     2.1     2.2     2.3     2.4
2    B    Holding     3.1     3.2     3.3     3.4
3    C    Holding     4.1     4.2     4.3     4.4
4    C    Holding     5.1     5.2     5.3     5.4
5    A    Summary     3.2     3.4     3.6     3.8
6    B    Summary     3.1     3.2     3.3     3.4
7    C    Summary     9.2     9.4     9.6     9.8