将 mutate 与多个函数一起使用

Question

我试图在 mutate(across(where(is.factor))) 中提供两个函数来排序因子水平并删除未使用的水平。该代码似乎没有按预期工作。可能哪里出了问题？

#---- Libraries ----

library(tidyverse)

#---- Data ----

set.seed(2021)

df <- tibble(
  a1 = factor(ifelse(sign(rnorm(30))==-1, 0, 1), labels = c("No", "Yes")),
  a2 = factor(ifelse(sign(rnorm(30))==-1, 0, 1), labels = c("No", "Yes")),
  gender = gl(2, 15, labels = c("Males", "Females")),
  b2 = gl(3, 10, labels = c("Primary", "Secondary", "Tertiary", "Unknown")),
  c1 = gl(3, 10, labels = c("15-19", "20-24", "25-30", "30-35")),
  outcome = factor(ifelse(sign(rnorm(30))==-1, 0, 1), labels = c("No", "Yes")),
  weight = runif(30, 1, 12)
)

#---- Problem ----

df <- df %>%
  mutate(across(where(is.factor), list(fct_infreq, fct_drop)))

levels(df$b2)

# The unused levels not dropped

Answer 1

问题是您实际上在这里改变了两个新列，因此您会在生成的数据框中看到有两列 b2_1 和 b2_2，每列对应于应用这两个函数.

如果你运行 levels(df$b2_2) 你会看到你想要的输出。

如果您的目标是先放下然后重新排序，那么您需要运行连续变异：

df <- df %>%
  mutate(across(where(is.factor), fct_drop)) %>% 
  mutate(across(where(is.factor), fct_infreq))

或运行 mutate

中的嵌套函数

df <- df %>%
  mutate(across(where(is.factor), ~fct_infreq(fct_drop(.x))))

将 mutate 与多个函数一起使用

Using mutate with multiple functions

r

dplyr

tidyverse

across