在 for 循环中使用 dplyr/ggplot2

Question

我正在尝试根据存储在向量中的列中的 11 个不同级别过滤和绘制 11 个不同的图表。

对于一个快速的、可重现的示例，这基本上是我尝试过的：

library(dplyr)
library(ggplot)
positions = c("forward", "defense")

df <- data.frame(player = c("Sergio Ramos", "Lionel Messi",
                            "Dani Alvez", "Christiano Ronaldo"),
                 position = c("forward", "defense", "defense", "forward"),
                 goals = c(12, 8, 2, 23))

for (i in 1:length(positions)) {
  df %>%
    filter(position == positions[i]) %>%
    print(ggplot(aes(x = player, y = goals)) +
      geom_bar(stat = "identity"))
}

如果我只是将过滤器包装在 print() 中并运行它，我会得到两个子集：

             player position goals
1       Sergio Ramos  forward    12
2 Christiano Ronaldo  forward    23
        player position goals
1 Lionel Messi  defense     8
2   Dani Alvez  defense     2

但是上面的代码给了我

错误：ggplot2 不知道如何处理 uneval 类的数据。

如果我在没有循环的情况下运行代码，它的图形效果很好。

我希望能够遍历变量并将它们全部发布到 Rmd 中或全部保存。有人可以解释为什么上述方法不起作用吗？谢谢！

Answer 1

使用 dplyr 管道会将数据帧放在错误的位置。试试这样吧：

library(dplyr)
library(ggplot2)
positions = c("forward", "defense")

df <- data.frame(player = c("Sergio Ramos", "Lionel Messi",
                            "Dani Alvez", "Christiano Ronaldo"),
                 position = c("forward", "defense", "defense", "forward"),
                 goals = c(12, 8, 2, 23))

for (i in 1:length(positions)) {
  df %>%
    filter(position == positions[i]) %>%
    ggplot(aes(x = player, y = goals)) +
            geom_bar(stat = "identity") -> g
    print(g)
}

你也可以考虑用 lapply 重写循环。

Answer 2

你真的不需要任何类型的循环来获得你需要的东西。 facet_wrap 可以满足您的需求。即

ggplot(df, aes(x = player, y = goals)) + geom_bar(stat = 'identity') +facet_wrap(~position)

Answer 3

如果你想把所有的地块独立地放在一个地方（不使用 facet_wrap），你可以尝试以下方法：

• 将数据框分成由位置定义的组。

  dfx <- split(df,as.factor(df$position))

lapply 解决方案

您还可以在拆分的数据帧上使用 lapply，如下所示：

• 定义一个产生你想要的情节的函数

  my_plot <- function(x){ g <- ggplot(x,aes(x = player, y = goals)) + geom_bar(stat = "identity") }

• 使用 lapply

ggy <- lapply(dfx,my_plot)

您将拥有 ggy 中的所有地块：

ggy$defense
ggy$forward

for循环解决方案

• 创建一个空列表以保留所有绘图。

  ggx <- list()

• 循环并将每个图保存在 ggx 列表中。

  for(i in 1:length(positions)){
      # create a barplot and save in g
      g <- ggplot(dfx[[i]],aes(x = player, y = goals)) +
              geom_bar(stat = "identity")
  
      n <- dfx[[i]]$position %>% unique() %>% as.character()
  
      # add the plot to the list
      ggx[[n]] <- g
  

  }

现在你的地块在：

ggx[[1]] or ggx$defense
ggx[[2]] or ggx$forward

您可以随时随地打印、保存或获取信息。

> ggx$defense$data
        player position goals
2 Lionel Messi  defense     8
3   Dani Alvez  defense     2

Answer 4

下面将在您的工作目录中的 pdf 的每一页中保存一个绘图

pdf("out.pdf", width = 7, height = 7)
for (i in 1:length(positions)) {
 print( df %>%
    filter(position == positions[i]) %>%
  ggplot(aes(x = player, y = goals)) +
            geom_bar(stat = "identity"))
}
dev.off() 

对于以 the markdown cheat sheet here 开头的降价文档。

但是我认为您可能想考虑使用 facet_grid 或 facet_wrap 函数并以与上述相同的方法将其保存为 pdf

Answer 5

你也可以这样做：

for (i in positions) {
  dfplt <- df %>%
    filter(position == positions) 
    plt <- ggplot(dfplt,aes(x = player, y = goals)) +
            geom_bar(stat = "identity")
    ggsave(plt, file=paste0(i,"_plot.png"))
}