R 从嵌套 df 中的条件改变主 df 中的变量

发布于07月30日

如何使用基于列表列中嵌套数据的条件语句在主数据框中创建新变量？

如果我们使用汽车数据集:

library(dplyr)

cars_nest <- mtcars %>%
  group_by(cyl) %>%
  nest()

并希望创建一个二进制变量，其中carb&gt；2的任何值等于1，否则为0.我try 了以下操作，但出现错误.

cars_nest  <- cars_nest |> 
  mutate(test = ifelse(any(cars_nest$carb) > 2, 1, 0))

推荐答案

代码中有几个问题:

您的any应该包括条件(即any(... > 2))
您没有在mutate中使用cars_nest变量，这是DataFrame对象的名称，而不是列(它应该是data列)
做这种手术需要rowwise英镑.

其中一个建议是:

由于逻辑比较(... > 2)的输出已经是逻辑的，您可以利用as.integer将其关联为整数，而不使用ifelse

因此，代码应该是:

library(tidyverse)

cars_nest <- mtcars %>%
  group_by(cyl) %>%
  nest() 
  
cars_nest %>% 
  rowwise() %>% 
  mutate(test = as.integer(any(data$carb > 2)))
  # or mutate(test = ifelse(any(data$carb > 2), 1, 0))

# A tibble: 3 × 3
# Rowwise:  cyl
    cyl data                test
  <dbl> <list>             <dbl>
1     6 <tibble [7 × 10]>      1
2     4 <tibble [11 × 10]>     0
3     8 <tibble [14 × 10]>     1