使用 dplyr/ggplot 函数在 for 循环中调用许多变量

Question

有时在执行探索性分析或生成报告时，我们希望绘制许多变量的单变量分布。经过一些整洁的技巧后，我可以在情节上做这件事，但是有有序的因素，我想让它们在情节上保持有序。

因此，为了以更有效的方式完成它，我构建了一个简单的基于dplyr/ggplot 的函数。我使用vcd包的关节炎数据集在下面做了这个例子。

library(dplyr)
library(ggplot2)

data(Arthritis, package = "vcd")

head(Arthritis)

plotUniCat <- function(df, x) {
  x <- enquo(x)
  df %>%
    filter(!is.na(!!x)) %>%
    count(!!x) %>%
    mutate(prop = prop.table(n)) %>%
    ggplot(aes(y=prop, x=!!x)) +
    geom_bar(stat = "identity")
}

plotUniCat(Arthritis, Improved)

我可以用很短的方式绘制格式化图表，这很酷，但只有一个变量。

我尝试使用 for 循环调用多个变量，但它不起作用。代码运行，但没有任何反应。

variables <- c("Improved", "Sex", "Treatment")

for (i in variables) {
  plotUniCat(Arthritis, noquote(i))
}

我对此进行了搜索，但我仍然不清楚。有人知道我做错了什么或如何使它起作用吗？

提前致谢。

Answer 1

我猜 OP 想将 plotUniCat 用于带引号和不带引号的变量名。如果我们改变函数，它对plotUniCat(Arthritis, Improved) 不起作用。

因此，除了更改函数，我们还可以将函数plotUniCat的调用方式更改为：

for (i in variables) {
    plotUniCat(Arthritis, !!rlang::sym(i))
}

但是，绘图是由for 生成但未返回的。我们可以使用print或者lapply来强制显示或者收集生成的图：

lapply(variables, function(i) plotUniCat(Arthritis, !!rlang::sym(i)))

Answer 2

将函数中的enquo改为sym，将变量字符串转换为符号。也就是说，

plotUniCat <- function(df, x) {
  x <- sym(x)
  df %>%
    filter(!is.na(!!x)) %>%
    count(!!x) %>%
    mutate(prop = prop.table(n)) %>%
    ggplot(aes(y=prop, x=!!x)) +
    geom_bar(stat = "identity")
}

或者，更简洁地说，

plotUniCat <- function(df, x) {
  x <- sym(x)
  df %>%
    filter(!is.na(!!x)) %>%
    ggplot(aes(x = as.factor(!!x))) +
    geom_histogram(stat = "count")
}

然后

out <- lapply(variables, function(i) plotUniCat(Arthritis,i))

最后，使用grid.arrange 显示绘图。例如

library(gridExtra)
do.call(grid.arrange, c(out, ncol = 2))

Answer 3

您需要使用rlang::sym 将字符串转换为符号而不是enquo。我将for 循环替换为purrr::map 以循环遍历variables

library(tidyverse)

data(Arthritis, package = "vcd")

head(Arthritis)
#>   ID Treatment  Sex Age Improved
#> 1 57   Treated Male  27     Some
#> 2 46   Treated Male  29     None
#> 3 77   Treated Male  30     None
#> 4 17   Treated Male  32   Marked
#> 5 36   Treated Male  46   Marked
#> 6 23   Treated Male  58   Marked

plotUniCat2 <- function(df, x) {
  x <- rlang::sym(x)
  df %>%
    filter(!is.na(!!x)) %>%
    count(!!x) %>%
    mutate(prop = prop.table(n)) %>%
    ggplot(aes(y=prop, x=!!x)) +
    geom_bar(stat = "identity")
}

variables <- c("Improved", "Sex", "Treatment")

variables %>% purrr::map(., ~ plotUniCat2(Arthritis, .x))
#> [[1]]

#> 
#> [[2]]

#> 
#> [[3]]

由reprex package (v0.2.0) 于 2018 年 6 月 13 日创建。