r - условие для списка данных столбца и номера строки

Question

У меня есть фрейм данных со списком-столбцом, как показано ниже:

L1 <- data.frame(a = I(list(
  c(1, 2, 6, 7),
  c(1, 2, 3, 6, 7),
  c(2, 3, 5),
  c(4, 5),
  c(4, 5, 7),
  c(6)
)))

Мне нужно создать два новых столбца на основе значения номера строки и значения в списке-столбце на основе следующих предположений:

Answer 1

Вы можете пройти через это, например, sapply.

df <- cbind(L1,
  t(
    sapply(seq(nrow(L1)), function(i) {
      cbind(col1 = any((L1$a[[i]] - i) == 1) + 0,
            col2 = any((L1$a[[i]] - i) == -1) + 0)
      })
    )
) 

colnames(df) <- c(colnames(L1), "col1", "col2")

df

             a col1 col2
1   1, 2, 6, 7    1    0
2 1, 2, 3,....    1    1
3      2, 3, 5    0    1
4         4, 5    1    0
5      4, 5, 7    0    1
6            6    0    0

Answer 2

Мы перебираем столбец list с map2 и проводим сравнение с соответствующим row_number(), чтобы создать два столбца

library(tidyverse)
 L1 %>% 
   mutate(n = row_number(),
             col1 = map2_int(a, n, ~ as.integer(any((.x - 1) > .y))),
             col2 = map2_int(a, n, ~ as.integer( any((.x - .y) < 0)))) %>%
   select(-n)    
#              a col1 col2
#1   1, 2, 6, 7    1    0
#2 1, 2, 3,....    1    1
#3      2, 3, 5    1    1
#4         4, 5    0    0
#5      4, 5, 7    1    1
#6            6    0    0