按R data.table中相应事件的数量标记行

Question

我是R中的新成员，我一直在用条件挣扎，我想应用在data.table中。

我的data.table由Order_id和Date排序，看起来像这样。

我需要的是与这些条件检举人变量创建新列：

• 如果有3个以上的连续0 hours_delta柱然后用flag_1之前标记这些线和线

• 如果有小于3，并且在hours_delta 大于1的连续的0，然后前标记这些线和线与flag_2

• 如果仅存在一个0 2是超过0象行索引之间[8]然后标记这些线与flag_3

• 标记所有其余与flag_4

这是我想该表看起来像在新列后面。

任何帮助，将不胜感激。

谢谢！

Answer 1

我觉得像这样的东西可能会为你正在努力完成的工作。

library(dplyr) 

# Create test dataframe 
index <- c(0:19) 
Order_id <- c(rep(001,8),rep(002,3),rep(003,4),rep(004,3),rep(005,2)) 
hours_delta <- c(720,552,rep(0,5),432,0,72,96,121,0,0,0,33,0,0,77,0) 

df <- data.frame(index,Order_id,hours_delta) 


# Start dplyr modifications 
df <- df %>% 
     # Group data by Order_id 
     group_by(Order_id) %>% 
     # Get the number of repitions of 0 for in the hours_delta field for that Order_id 
     mutate(rle = ifelse(hours_delta == 0,rle(hours_delta)[[1]][rle(hours_delta)[[2]] == 0],NA), 
      # Set the row above a zero sequence to the number of repetitions 
      rle = ifelse(is.na(rle),lead(rle),rle)) %>% 
     # ungroup the data 
     ungroup() %>% 
     # Set the flags based on number of repetitions 
     mutate(flagger = case_when(is.na(.$rle) 
           ~ "flag_4", 
           .$rle == 1 
           ~ "flag_3", 
           (.$rle <= 3 & .$rle > 1) 
           ~ "flag_2", 
           .$rle > 3 
           ~ "flag_1" 
           ) 
      ) %>% 
    # Remove the temporary rle column 
    select(-rle)