在pandas DataFrame中基于其他列对一列求和

pgccezyw 于 2023-05-21 发布在其他

关注(0)|答案(2)|浏览(174)

我在pandas DataFrame中有几列（'SEC'，'DeviceName'，' DAY '和' SPRAY'）。我需要在“DAY”列中按“DeviceName”对每天的“SEC”求和，其中“SPRAY”== 1。
例如：

DeviceName       SEC     DAY     SPRAY
Applied_AA-6     1800    4       1
Applied_AA-6     600     4       0
Applied_AA-6     1373    5       1
Applied_AA-15    10      4       0
Applied_AA-15    1800    5       1
Applied_AA-15    13      5       1

结果应为：

Day 4   Day 5
Applied_AA-6     1800    1373
Applied_AA-15    0       1813

pandas

来源：https://stackoverflow.com/questions/76266658/sum-a-column-in-pandas-dataframe-based-on-other-columns

2条答案

按热度按时间

e5nqia271#

使用groupby和unstack

df[df['SPRAY'].eq(1)].groupby(['DeviceName', 'DAY'])['SEC'] \
                     .sum() \
                     .unstack() \
                     .fillna(0)

DAY                 4       5
             
Applied_AA-15       0  1813.0
Applied_AA-6   1800.0  1373.0

赞(0）回复(0）举报 2023-05-21

lvjbypge2#

另一种可能的解决方案，基于pandas.pivot_table：

(df[df['SPRAY'].eq(1)]
 .pivot_table(
     index='DeviceName', columns='DAY', values='SEC', 
     aggfunc='sum', sort=False, fill_value=0)
 .rename_axis(None, axis=1).reset_index())

输出：

DeviceName     4     5
0   Applied_AA-6  1800  1373
1  Applied_AA-15     0  1813

赞(0）回复(0）举报 2023-05-21

我来回答

在pandas DataFrame中基于其他列对一列求和

2条答案

相关问题

热门标签

最新问答