根據 if 條件合並和求和兩個 pandas 行

Question

我有一個巨大的 dataframe ，我只想merge only two rows in it based on if condition 。 下面是一個示例數據框，當我嘗試進行groupby sum時，其他行也會受到影響。

我只希望列jb_name與generic得到merged and sum 。

    jb_name   jb_count
0   generic    10
1   generic1    2
2   generic    15
3   other      14

嘗試了以下，但正如我所說，它也會影響其他行

df = df.groupby(['jb_name'])['jb_count'].sum().reset_index()

我想要最終的df如下

    jb_name   jb_count
0   generic    25
1   generic1    2
3   other      14

Answer 1

利用：

mask = df['jb_name'] == 'generic'
df = df[mask].groupby('jb_name', as_index=False).sum().append(df[~mask], ignore_index=True)

Answer 2

或者，我們可以將索引設置為jb_name並在索引為generic的level 0上使用sum ：

df = df.set_index('jb_name')
mask = (df.index == 'generic')
df1 = pd.concat([df[mask].sum(level=0), df[~mask]]).reset_index()

結果：

# print(df1)

    jb_name  jb_count
0   generic        25
1  generic1         2
2     other        14

Answer 3

#Bollean select, droupby as you sum the duplicated and append tthe no duplicates

m=df.jb_name=='generic'
df[m].groupby(by='jb_name', axis=0).sum().reset_index().append(df[~m])



jb_name  jb_count
0   generic        25
1  generic1         2
3     other        14