python - 按 pandas 中的排名求和值

Question

我正在尝试进行多年财务预测。我需要根据当年的排名计算支出，然后重新排名并计算下一年的支出。

我的数据框包含汽车零件以及在预算有限的情况下修复/更换它的成本(400 美元)，一旦预算用完，那一年我就无法再修复任何其他东西，但它将被重新优先考虑以下内容年。

在第 1 年，预算仅涵盖传输，但在第 2 年传输很好，因此排名最低。如何根据排名运行累积总和直至达到预算限制？

   df
   Part           Rank(Year 1)     Cost(Y1)   Spend(Y1)   Rank(Y2)   Spend(Y2)

   Transmission        1               400       400         4          0
   Tires               2               400        0          1         400
   Windshield          3               100        0          2          0
   Wipers              4                20        0          3          0

还想补充:

    for val in df['Spend']:
        if val.cumsum() >= budget_var:
          val = 0

现在就是这样，但我需要将总和按排名顺序排列。

Answer 1

sorted_df = df.sort_values(col_name)

另外，我不认为你的 cumsum() 是正确的；您将其应用于列中的各个值，而不是整个列。

试试这个

df['Spend'] = max(0,df['Cost'].cumsum())