使用R查找基于条件返回值的函数

Question

我有一个包含值的表格

KId sales_month quantity_sold
100        1    0
100        2    0
100        3    0
496        2    6
511        2    10
846        1    4
846        2    6
846        3    1
338        1    6
338        2    0    

现在我需要输出为

KId sales_month quantity_sold result
100           1     0         1
100           2     0         1
100           3     0         1
496           2     6         1
511           2     10        1
846           1     4         1
846           2     6         1
846           3     1         0
338           1     6         1
338           2     0         1

在这里，如果 3 月 (3) 月份的销售量小于 1 月 (1) 和 2 月 (2) 两个月的 60%，则计算必须如此结果应该是1，否则应该显示0。需要解决方案来执行此操作。

提前致谢。

Answer 1

如果我理解得很好，您的要求是将 t 月的销售量与 t-1 月和 t-2 月的销售量总和进行比较。如果是这样，我可以建议使用 dplyr 包，它提供了在数据框中对行进行分组和改变列的好功能。

resultData <- group_by(data, KId) %>% 
    arrange(sales_month) %>% 
    mutate(monthMinus1Qty = lag(quantity_sold,1), monthMinus2Qty = lag(quantity_sold, 2)) %>% 
    group_by(KId, sales_month) %>%
    mutate(previous2MonthsQty = sum(monthMinus1Qty, monthMinus2Qty, na.rm = TRUE)) %>%  
    mutate(result = ifelse(quantity_sold/previous2MonthsQty >= 0.6,0,1)) %>%
    select(KId,sales_month, quantity_sold, result)

结果如下：

添加

select(KId,sales_month, quantity_sold, result)

最后让我们只显示我们关心的列（而不是所有这些中间步骤）。

我相信这应该可以满足您的要求。 NA 是结果列是由于前几个月的 0/0 除法或根本没有数据。 如果您需要将计算扩展到超过一个日历年，您可以添加 year 列并适当调整 group_by() 参数。 有关 dplyr 包的更多信息，请关注this link