gpt4 book ai didi

python - 基于其组重新定义 Pandas 数据框

转载 作者:行者123 更新时间:2023-12-04 07:40:54 26 4
gpt4 key购买 nike

我正在使用这个数据框

  source   fruit  2019  2020  2021
0 a apple 3 1 1
1 a banana 4 3 5
2 a orange 2 2 2
3 b apple 3 4 5
4 b banana 4 5 2
5 b orange 1 6 4
我想像这样改进它
 source   fruit  2019  2020  2021
0 a total 9 6 8
1 a seeds 5 3 3
2 a banana 4 3 5
3 b total 8 15 11
4 b seeds 4 10 9
5 b banana 4 5 2
total 是该年每个来源的所有果实的总和。
种子是每个来源每年含有种子的果实总和。
我试过
追加新的空行: Insert a new row after every nth row & Insert row at any position
但是没有得到预期的结果。
获得所需输出的最佳方法是什么?

最佳答案

尝试:

df1 = df.groupby('source', as_index=False).sum().assign(fruit = 'total')
seeds = ['orange','apple']
df2 = df.loc[df['fruit'].isin(seeds)].groupby('source', as_index=False).sum().assign(fruit = 'seeds')
final_df = pd.concat([df.loc[~df['fruit'].isin(seeds)], df1,df2])

关于python - 基于其组重新定义 Pandas 数据框,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/67468917/

26 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com