facet_wrap中的多个标题(ggplot2)

时间:2015-08-02 20:05:11

标签: r ggplot2

我正在尝试使用facet_wrap和ggplot2为绘图添加多个标题。说你就是拥有两年的季度数据,并希望将季度数据与两个主要标题进行图形比较; 2014年和2015年,以及各季度的标题。

我到目前为止:

data <- rnorm(10)

A1 <- data.frame("Y"=data, "X"=1:10, "Q"=1, "year"=2014)
A2 <- data.frame("Y"=data, "X"=1:10, "Q"=2, "year"=2014)
A3 <- data.frame("Y"=data, "X"=1:10, "Q"=3, "year"=2014)
A4 <- data.frame("Y"=data, "X"=1:10, "Q"=4, "year"=2014)

N1 <- data.frame("Y"=data, "X"=1:10, "Q"=1, "year"=2015)
N2 <- data.frame("Y"=data, "X"=1:10, "Q"=2, "year"=2015)
N3 <- data.frame("Y"=data, "X"=1:10, "Q"=3, "year"=2015)
N4 <- data.frame("Y"=data, "X"=1:10, "Q"=4, "year"=2015)

A <- rbind(A1, A2, A3, A4)
N <- rbind(N1, N2, N3, N4)
tmp <- data.frame(rbind(A, N))

ggplot(data=tmp, aes(x=X, y=Y)) + geom_line() + facet_wrap(~year + Q, scales="free", ncol=4)

给了我这张图: enter image description here

相反,我希望&#34; 2014&#34;和&#34; 2015&#34;在每个子图上方的两个单独的灰色框中。这可能吗?

谢谢!

2 个答案:

答案 0 :(得分:5)

根据Sandy Muspratt的建议使用代码here,我可以提出:

library(ggplot2)
library(gtable)

data <- rnorm(10)

A1 <- data.frame("Y"=data, "X"=1:10, "Q"=1, "year"=2014)
A2 <- data.frame("Y"=data, "X"=1:10, "Q"=2, "year"=2014)
A3 <- data.frame("Y"=data, "X"=1:10, "Q"=3, "year"=2014)
A4 <- data.frame("Y"=data, "X"=1:10, "Q"=4, "year"=2014)

N1 <- data.frame("Y"=data, "X"=1:10, "Q"=1, "year"=2015)
N2 <- data.frame("Y"=data, "X"=1:10, "Q"=2, "year"=2015)
N3 <- data.frame("Y"=data, "X"=1:10, "Q"=3, "year"=2015)
N4 <- data.frame("Y"=data, "X"=1:10, "Q"=4, "year"=2015)

A <- rbind(A1, A2, A3, A4)
N <- rbind(N1, N2, N3, N4)
tmp <- data.frame(rbind(A, N))

p <- ggplot(data=tmp, aes(x=X, y=Y)) + geom_line() + facet_wrap(~year + Q, scales="free", ncol=4)


z <- ggplotGrob(p)

#  New title strip at the top
z <- gtable_add_rows(z, unit(1, "lines"), pos = 0)  # New row added to top

# Display the layout to select the place for the new strip
gtable_show_layout(z)   # New strip goes into row 2 
# New strip spans columns 4 to 13

z <- gtable_add_grob(z, 
                     list(rectGrob(gp = gpar(col = NA, fill = grey(0.8), size = .5)),
                          textGrob("2014", vjust = .27, 
                                   gp = gpar(cex = .75, fontface = 'bold', col = "black"))), 2, 4, 2, 13, name = c("a", "b"))

# Add small gap between strips - below row 2
z <- gtable_add_rows(z, unit(2/10, "line"), 2)


#  New title strip in the middle
z <- gtable_add_rows(z, unit(1, "lines"), pos = 8)  # New row added to top
# Display the layout to select the place for the new strip
gtable_show_layout(z) 
# New strip goes into row 9 
# New strip spans columns 4 to 13

z <- gtable_add_grob(z, 
                     list(rectGrob(gp = gpar(col = NA, fill = grey(0.8), size = .5)),
                          textGrob("2015", vjust = .27, 
                                   gp = gpar(cex = .75, fontface = 'bold', col = "black"))), 9, 4, 9, 13, name = c("a", "b"))

# Add small gap between strips - below row 2
z <- gtable_add_rows(z, unit(2/10, "line"), 9)


# Draw it
grid.newpage()
grid.draw(z)

这将为您提供此图表:enter image description here

答案 1 :(得分:5)

你打败我RHA,在看到你的帖子之前,我给你写了几乎相同的代码。无论如何,谢谢!

我还想删除&#34; 2014&#34;和&#34; 2015&#34;从灰色框(我没有在我的第一篇文章中指定),所以我不得不做一些更改。

借助hereherehere的灵感,我提出了以下(非常难看)的代码:

data14 <- rnorm(10)
data15 <- rnorm(10, mean = 500)

A1 <- data.frame("Y"=data14, "X"=1:10, "Q"=1, "year"=2014)
A2 <- data.frame("Y"=data14, "X"=1:10, "Q"=2, "year"=2014)
A3 <- data.frame("Y"=data14, "X"=1:10, "Q"=3, "year"=2014)
A4 <- data.frame("Y"=data14, "X"=1:10, "Q"=4, "year"=2014)

N1 <- data.frame("Y"=data15, "X"=1:10, "Q"=1, "year"=2015)
N2 <- data.frame("Y"=data15, "X"=1:10, "Q"=2, "year"=2015)
N3 <- data.frame("Y"=data15, "X"=1:10, "Q"=3, "year"=2015)
N4 <- data.frame("Y"=data15, "X"=1:10, "Q"=4, "year"=2015)

A <- rbind(A1, A2, A3, A4)
N <- rbind(N1, N2, N3, N4)
tmp <- data.frame(rbind(A, N))

然后我做了一个简单的函数正确命名变量

labFunc <- function(data, var1, var2, names) {
  data$id <- NA
  loop <- length(levels(factor(data[[var2]])))

  for (i in 1:loop) {
    data[data[[var1]] == 2014 & data[[var2]] == levels(factor(data[[var2]]))[i], "id"] <- names[i]
    data[data[[var1]] == 2015 & data[[var2]] == levels(factor(data[[var2]]))[i], "id"] <- paste(names[i], "")
  }

  first <- levels(factor(data$id))[seq(from=1, to = length(levels(factor(data$id))), by = 2)]
  second <- levels(factor(data$id))[seq(from=2, to = length(levels(factor(data$id))), by = 2)]

  data$id <- factor(data$id, levels=paste(c(first, second)))
  return(data)
}

names <- c("Q1", "Q2", "Q3", "Q4")
data <- labFunc(tmp, "year", "Q", names)

制作情节:

p <-ggplot(data, aes(y = Y, x = X)) +
    geom_line() +
    facet_wrap( ~ id , ncol = 4, scales = "free")

然后最后添加主要标签

z <- ggplotGrob(p)

#  New strip at the top
z <- gtable_add_rows(z, unit(1, "lines"), pos = 0)  # New row added to top
z <- gtable_add_rows(z, unit(1, "lines"), pos = 6)  # New row added to top

#z <- gtable_add_rows(z, unit(9, "lines"), pos = 0)  # New row added to top


# Check the layout
gtable_show_layout(z)   # New strip goes into row 2 
# New strip spans columns 4 to 8

z <- gtable_add_grob(z, 
                     list(rectGrob(gp = gpar(col = NA, fill = grey(0.8), size = .5)),
                          textGrob("2014", vjust = .27, 
                                   gp = gpar(cex = .75, fontface = 'bold', col = "black"))), 2, 4, 2, 14, name = c("a", "b"))

z <- gtable_add_grob(z, 
                     list(rectGrob(gp = gpar(col = NA, fill = grey(0.8), size = .5)),
                          textGrob("2015", vjust = .27, 
                                   gp = gpar(cex = .75, fontface = 'bold', col = "black"))), 7, 4, 7, 14, name = c("a", "b"))



# Add small gap between strips - below row 2
z <- gtable_add_rows(z, unit(2/10, "lines"), 2)
z <- gtable_add_rows(z, unit(5/10, "lines"), 7)

# Draw it
grid.newpage()
grid.draw(z)

enter image description here

这比我想象的要复杂一些,但是谢谢大家的帮助!