Function 相當於 Excel 的 SUMIFS()

Question

我有一個包含item 、 week和sales列的銷售表。 我想創建一周至今的銷售列 ( wtd sales )，這是每個項目的每周銷售匯總。

我不知道如何在 Python 中創建它。

我被困在groupby() ，這可能不是答案。 有人可以幫忙嗎？

output_df['wtd sales'] = input_df.groupby(['item'])['sales'].transform(wtd)

Answer 1

正如我在評論中所述，您正在尋找cumsum() ：

import pandas as pd

df = pd.DataFrame({
    'items': ['A', 'A', 'A', 'A', 'B', 'B', 'B', 'B'],
    'weeks': [1, 2, 3, 4, 1, 2, 3, 4],
    'sales': [100, 101, 102, 130, 10, 11, 12, 13]
})

df.groupby(['items'])['sales'].cumsum()

結果是：

0    100
1    201
2    303
3    433
4     10
5     21
6     33
7     46
Name: sales, dtype: int64

我正在使用：

pd.__version__
'1.5.1'
 

把它們放在一起：

import pandas as pd

df = pd.DataFrame({
    'items': ['A', 'A', 'A', 'A', 'B', 'B', 'B', 'B'],
    'weeks': [1, 2, 3, 4, 1, 2, 3, 4],
    'sales': [100, 101, 102, 130, 10, 11, 12, 13]
})

df['wtds'] = df.groupby(['items'])['sales'].cumsum()

導致：

  items  weeks  sales  wtds
0     A      1    100   100
1     A      2    101   201
2     A      3    102   303
3     A      4    130   433
4     B      1     10    10
5     B      2     11    21
6     B      3     12    33
7     B      4     13    46