如何使用growby调整panda中的小计列

Question

我正在使用数据帧联接将数据帧导出到Excel。

但是在Join数据帧之后，使用groupby计算小计时，执行下图。

有一个"；小计"；索引列中的单词。

在此处输入图像描述

有没有办法将它移到代码列中并对索引进行排序？

在此处输入图像描述

此处代码：

def subtotal(df__, str):
container = []

for key, group in df__.groupby(['key']):
group.loc['subtotal'] = group[['quantity', 'quantity2', 'quantity3']].sum()
container.append(group)

df_subtotal = pd.concat(container)
df_subtotal.loc['GrandTotal'] = df__[['quantity', 'quantity2', 'quantity3']].sum()
print(df_subtotal)
return (df_subtotal.to_excel(writer, sheet_name=str))

Answer 1

使用np.where()用df.index中的值填充code列中的NaN。然后为df.index分配一个新的索引数组。

import numpy as np
df['code']  = np.where(df['code'].isna(), df.index, df['code'])
df.index = np.arange(1, len(df) + 1)

print(df)
code  key    product  quntity1  quntity2   quntity3
1      cs01767    a    apple-a        10          0      10.0
2     Subtotal  NaN        NaN        10          0      10.0
3       cs0000    b  bannana-a        50         10      40.0
4       cs0000    b  bannana-b         0          0       0.0
5       cs0000    b  bannana-c         0          0       0.0
6       cs0000    b  bannana-d        80         20      60.0
7       cs0000    b  bannana-e         0          0       0.0
8      cs01048    b  bannana-f         0          0       NaN
9      cs01048    b  bannana-g         0          0       0.0
10    Subtotal  NaN        NaN       130         30     100.0
11     cs99999    c    melon-a        50         10      40.0
12     cs99999    c    melon-b        20         20       0.0
13     cs01188    c    melon-c        10          0      10.0
14    Subtotal  NaN        NaN        80         30      50.0
15  GrandTotal  NaN        NaN       220         60     160.0