嵌套数据框在使用 ifelse 条件发生变异后丢失格式

Question

我正在尝试使用 ifelse 条件更改嵌套数据框中的新变量。但问题是在实现 ifelse 条件后，嵌套的数据框变成了一个列表。 我想用iris 数据集展示这个问题：

这里可以看到原来的嵌套格式：

iris %>% nest(data = -Species)

# A tibble: 3 x 2
  Species    data             
  <fct>      <list>           
1 setosa     <tibble [50 x 4]>
2 versicolor <tibble [50 x 4]>
3 virginica  <tibble [50 x 4]>

现在我想在嵌套数据框中改变一个新变量：

iris %>%
  nest(data = -Species) %>%
  mutate(data = map(data, function(x)
    x %>% mutate(`Sepal.Length^2` = Sepal.Length^2)))

# A tibble: 3 x 2
  Species    data             
  <fct>      <list>           
1 setosa     <tibble [50 x 5]>
2 versicolor <tibble [50 x 5]>
3 virginica  <tibble [50 x 5]>

此代码有效。 data-column 在 tibble 格式中是所需的。

但如果我现在插入 ifelse 条件，则 tibble 格式会丢失：

iris %>%
  nest(data = -Species) %>%
  mutate(data = map(data, function(x)
    ifelse(!is.na(x), x %>% mutate(`Sepal.Length^2` = Sepal.Length^2), NA)))

# A tibble: 3 x 2
  Species    data        
  <fct>      <list>      
1 setosa     <list [200]>
2 versicolor <list [200]>
3 virginica  <list [200]>

即使使用ifelse-条件，我也想保留tibble-格式。

谁能帮帮我？

Answer 1

在map()计算的第一步，即setosa中的数据，你自定义函数的输入x其实是

x <- iris[1:50, 1:4]

然后你把x 放入ifelse()

ifelse(!is.na(x),                                        # part 1
       x %>% mutate(`Sepal.Length^2` = Sepal.Length^2),  # part 2
       NA)                                               # part 3

第一部分是!is.na(x)，它返回50x4=200逻辑值。因此，第二部分和第三部分将被回收到长度 200。但是，第二部分，即

x %>% mutate(`Sepal.Length^2` = Sepal.Length^2)

是一个有5个变量的tibble，也是一个长度为5的list，所以这个tibble中的每个变量都会被回收40次，随后会创建一个长度为200的list。这就是为什么你会得到 3 个长度为 200 的列表。

---

在您的情况下，ifelse() 可能不适用。您可以将其调整为

iris %>%
  nest(data = -Species) %>%
  add_row(Species = "example", data = NA) %>% 
  mutate(data = map(data, function(x) {
    if(is.data.frame(x))
      x %>% mutate(`Sepal.Length^2` = Sepal.Length^2)
    else
      NULL
  }))

# # A tibble: 4 x 2
#   Species    data             
#   <chr>      <list>           
# 1 setosa     <tibble [50 × 5]>
# 2 versicolor <tibble [50 × 5]>
# 3 virginica  <tibble [50 × 5]>
# 4 example    <NULL>   

确保if() 中的条件必须是单个逻辑值。

---

感谢 @27ϕ9 提供map_if() 的简洁版本：

iris %>%
  nest(data = -Species) %>%
  add_row(Species = "example", data = NA) %>% 
  mutate(data = map_if(data, is_tibble,
                       ~ mutate(.x, `Sepal.Length^2` = Sepal.Length^2),
                       .else = NULL))