web-dev-qa-db-ja.com

ggplot2：データセットの変数に基づいたfacet_wrapストリップの色

データフレームで提供される変数に基づいてfacet_wrapで作成されたファセットのストリップを埋める方法はありますか？

サンプルデータ：

MYdata <- data.frame(fruit = rep(c("Apple", "orange", "Plum", "banana", "pear", "grape")), farm = rep(c(0,1,3,6,9,12), each=6), weight = rnorm(36, 10000, 2500), size=rep(c("small", "large")))

プロットの例：

p1 = ggplot(data = MYdata, aes(x = farm, y = weight)) + geom_jitter(position = position_jitter(width = 0.3), aes(color = factor(farm)), size = 2.5, alpha = 1) + facet_wrap(~fruit)

私はストリップの背景色を変更する方法を知っています（例えば、オレンジに）：

p1 + theme(strip.background = element_rect(fill="orange"))

facet_wrap and orange strip color

sizeの変数MYdataの値をelement_rectのパラメーターfillに渡す方法はありますか？

基本的に、すべてのストリップの1色の代わりに、小さな果物（リンゴ、プラム、ナシ）の背景色を緑に、大きな果物（オレンジ、バナナ、ブドウ）の背景色を赤にしたいです。

rggplot2facet-wrap
50
Dalmuti71

少しの作業で、プロットを適切なグロブを持つダミーgtableと組み合わせることができます。

enter image description here

d <- data.frame(fruit = rep(c("Apple", "orange", "Plum", "banana", "pear", "grape")), 
                farm = rep(c(0,1,3,6,9,12), each=6), 
                weight = rnorm(36, 10000, 2500), 
                size=rep(c("small", "large")))

p1 = ggplot(data = d, aes(x = farm, y = weight)) + 
  geom_jitter(position = position_jitter(width = 0.3), 
              aes(color = factor(farm)), size = 2.5, alpha = 1) + 
  facet_wrap(~fruit)

dummy <- ggplot(data = d, aes(x = farm, y = weight))+ facet_wrap(~fruit) + 
  geom_rect(aes(fill=size), xmin=-Inf, xmax=Inf, ymin=-Inf, ymax=Inf) +
  theme_minimal()

library(gtable)

g1 <- ggplotGrob(p1)
g2 <- ggplotGrob(dummy)

gtable_select <- function (x, ...) 
{
  matches <- c(...)
  x$layout <- x$layout[matches, , drop = FALSE]
  x$grobs <- x$grobs[matches]
  x
}

panels <- grepl(pattern="panel", g2$layout$name)
strips <- grepl(pattern="strip_t", g2$layout$name)
g2$layout$t[panels] <- g2$layout$t[panels] - 1
g2$layout$b[panels] <- g2$layout$b[panels] - 1

new_strips <- gtable_select(g2, panels | strips)
grid.newpage()
grid.draw(new_strips)

gtable_stack <- function(g1, g2){
  g1$grobs <- c(g1$grobs, g2$grobs)
  g1$layout <- transform(g1$layout, z= z-max(z), name="g2")
  g1$layout <- rbind(g1$layout, g2$layout)
  g1
}
## ideally you'd remove the old strips, for now they're just covered
new_plot <- gtable_stack(g1, new_strips)
grid.newpage()
grid.draw(new_plot)
59
baptiste

その方法を知りたいのですが、それは素晴らしいアイデアです。 1つのアイデアは、各グラフを異なる色で個別に生成し、マルチプロットやビューポートなどを使用して並べて表示することです。もう少し手間がかかります。

このアプローチに必要な凡例を抽出したい場合は、ここでしばらく前に見つけたHadleyのコードをいくつか示します。

g_legend<-function(a.gplot){
  tmp <- ggplot_gtable(ggplot_build(a.gplot))
  leg <- which(sapply(tmp$grobs, function(x) x$name) == "guide-box")
  legend <- tmp$grobs[[leg]]
  return(legend)}

グラフpからどのように抽出されるかを確認し、プロットの凡例から取り出しました<-g_legend（p）lwidth <-sum（legend $ width）＃このpに基づいてビューポートを定義する場合<-p + theme（legend.position = "none"）

その後、最終的にそれを描きます

grid.newpage()
vp <- viewport(width = 1, height = 1)
#print(p, vp = vp)

submain <- viewport(width = 0.9, height = 0.9, x = 0.5, y = 1,just=c("center","top"))
print(p, vp = submain)
sublegend <- viewport(width = 0.5, height = 0.2, x = 0.5, y = 0.0,just=c("center","bottom"))
print(arrangeGrob(legend), vp = sublegend)

がんばろう

0
user1617979