определить ряд значений, перед которыми следует определенное значение

Question

У меня есть последовательность целых чисел.То, что я хотел бы сделать, это определить все последовательности из 3-х, которые предшествуют И, а затем 5. Например:

c(5,3,3,5,5,4,3,3,5)

Желаемый результат будет:

c(F,T,T,F,F,F,F,F,F)

Объяснение:Первой последовательности 3-х предшествует 5. Далее True.Второй последовательности предшествует 4, следовательно False.

Answer 1

У меня очень длинное и уродливое решение, но оно работает: p Я надеюсь, что кто-то найдет более чистое решение :) Сначала я создаю матрицу, которая содержит 1 столбец, который представляет собой каждое число неповторяющимся образом (не уникальным,но без последовательных), а затем 1 столбец с количеством повторений этого числа.Затем я применяю логическую функцию, чтобы увидеть, окружена ли 3 цифрами 5, и на последнем шаге я распутываю вектор обратно к его исходной длине, используя функцию rep () ...

x <- c(5,3,3,5,5,4,3,3,5)

x_reduced <- x[x!=c(x[-1], FALSE)]
x_mat <- matrix(0, ncol = 3, nrow = length(x_reduced))
x_mat[ , 1] <- x_reduced

ctr = 1
x_ctr = 1
while (ctr < length(x)) {
  x_mat[x_ctr ,1] = x[ctr]
  x_mat[x_ctr, 2] = x_mat[x_ctr, 2] + 1 
  if(x[ctr+1] == x[ctr]){
    ctr = ctr + 1
  } else {
    x_ctr = x_ctr + 1
    ctr = ctr + 1
  }
}
x_mat[nrow(x_mat), 1] <- x[length(x)]
x_mat[nrow(x_mat), 2] <- x_mat[nrow(x_mat), 2] + 1

check_element <- function(pos) {
  if(pos == 1 | pos == nrow(x_mat)) return(FALSE)
  if(x_mat[pos+1, 1] == 5 & x_mat[pos-1, 1] == 5){
    return(TRUE)
  } else {
    return(FALSE)
  }
}

x_mat[,3] <- sapply(1:nrow(x_mat), check_element)
rep(x_mat[,3], x_mat[,2])

Answer 2

Есть место для оптимизации, но, конечно, это возможно с dplyr и rle().

> df_result
# A tibble: 9 x 1
  result
  <lgl> 
1 FALSE 
2 TRUE  
3 TRUE  
4 FALSE 
5 FALSE 
6 FALSE 
7 FALSE 
8 FALSE 
9 FALSE 

код

df_result <- df %>%
    group_by(seq = {seq = rle(value); rep(seq_along(seq$lengths), seq$lengths)}) %>%
    ungroup() %>%
    mutate(last_3 = case_when(lag(seq) != seq ~ as.numeric(lag(value) == 5),
                              TRUE ~ NA_real_),
           next_5 = case_when(lead(seq) != seq ~ as.numeric(lead(value) == 5),
                              TRUE ~ NA_real_)) %>%
    group_by(seq, value) %>%
    mutate(result = case_when(sum(last_3, na.rm = TRUE) + sum(next_5, na.rm = TRUE) == 2 ~ TRUE,
                              TRUE ~ FALSE)) %>%
    ungroup() %>%
    select(result)

Данные

library(dplyr)
df <- data.frame(value = c(5,3,3,5,5,4,3,3,5))

Answer 3

Не удалось придумать более разумное решение, поэтому вот цикл for

x <- c(5,3,3,5,5,4,3,3,5) #Initial vector
current_inds <- numeric() #Variable to hold indices which need to be changed
saw_3 <- FALSE  #If 3 was seen before
output <- rep(FALSE, length(x))  #output vector
num_to_check <- 5   #Value to compare
last_val <- 0 #Last non-3 value

for (i in seq_along(x)) {
    #If the current value is not equal to 3
    if (x[i] != 3 ) {
      #Check if we previously saw 3 and if previous non_3 value was 5
      # and the next value is 5
      if(saw_3 & x[i + 1] == num_to_check & last_val == num_to_check) {
         #Change the group of 3 indices to TRUE
         output[current_inds] <- TRUE
         #Make the saw_3 flag as FALSE
         saw_3 <- FALSE
       }
      #Update the last seen non_3 value to current value
      last_val = x[i]
      }
    else {
     #If it is a 3 then append the indices in current_inds
     current_inds <- c(current_inds, i)
     #Make saw_3 flag TRUE
     saw_3 = TRUE
    }
}

output
#[1] FALSE  TRUE  TRUE FALSE FALSE FALSE FALSE FALSE FALSE