按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)