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

我需要根据以下假设基于行号值和list-column中的值生成两个新列：

Answer 1

我们使用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

Answer 2

你可以通过例如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