根据一个值计算R中的下面的行数

Question

我有一个显示的数据库，并希望根据一列中的一个值计算下面的行总和。

例如：

• 如果＆＃34;安全库存= 2个月＆＃34;，对于同一产品，计算的安全库存单位为以下2行的总和
• 如果＆＃34;安全库存= 3个月＆＃34;，对于同一产品，计算的安全库存单位为以下3行的总和

有人知道如何进行R？

Answer 1

假设＆＃39; Safety_Stocks＆＃39;对于每个＆＃39;产品，我们都可以使用data.table。转换＆＃39; data.frame＆＃39;到＆＃39; data.table＆＃39; （setDT(df1)），按产品＆＃39;分组，通过指定＆＃39; n＆＃39;获取lead值shift。作为＆＃39; Safety_Stocks＆＃39;的第一个元素序列，使用Reduce获取相应list元素的总和，并将输出分配（:=）到新列（＆＃39; New_SafetyStocks＆＃39）。

library(data.table)
setDT(df1)[, New_SafetyStocks := Reduce(`+`, shift(Sales_Forecast, 
           type="lead", n=seq(Safety_Stocks[1L]))) , by = Product]
df1
#      Product Sales_Forecast Safety_Stocks New_SafetyStocks
# 1: Product A            100             2              300
# 2: Product A            100             2              300
# 3: Product A            200             2              400
# 4: Product A            100             2              600
# 5: Product A            300             2              400
# 6: Product A            300             2              200
# 7: Product A            100             2              300
# 8: Product A            100             2              500
# 9: Product A            200             2              400
#10: Product A            300             2              200
#11: Product A            100             2               NA
#12: Product A            100             2               NA
#13: Product B            500             3             2000
#14: Product B            500             3             3000
#15: Product B           1000             3             3500
#16: Product B            500             3             3500
#17: Product B           1500             3             2500
#18: Product B           1500             3             2000
#19: Product B            500             3             3000
#20: Product B            500             3             3000
#21: Product B           1000             3             2500
#22: Product B           1500             3               NA
#23: Product B            500             3               NA
#24: Product B            500             3               NA

＆LT;

数据

df1 <- data.frame(Product = rep(c("Product A", "Product B"), each = 12),
       Sales_Forecast = c(100, 100, 200, 100, 300, 300, 100, 100, 200, 300,
    100, 100, 500, 500, 1000, 500, 1500, 1500, 500, 500, 1000, 1500, 500, 500),
   Safety_Stocks = rep(c(2,3), each = 12))