R 使用ggplot对密度图的不同部分进行固定填充

给定从一个

rnorm

和一个

c

绘制的图形，我希望我的绘图使用以下颜色：

红色表示

-c

蓝色表示

-c

和

c

对于

c

例如，如果我的数据是：

set.seed(9782)
mydata <- rnorm(1000, 0, 2)
c <- 1

---

唉，我不知道如何在保留百分比作为颜色标签的同时将颜色分配给不同的部分。

我们使用

密度

函数来创建实际绘制的数据帧。然后，我们使用

cut

函数使用数据值的范围创建组。最后，我们计算每个组的概率质量，并将其用作实际的图例标签

我们还创建一个带标签的颜色向量，以确保在给定的x值范围内始终使用相同的颜色，而不管数据是否包含给定x值范围内的任何值

下面的代码将所有这些打包成一个函数

library(tidyverse)
library(gridExtra)

fill_density = function(x, cc=1, adj=1, drop_levs=FALSE) {

  # Calculate density values for input data
  dens = data.frame(density(x, n=2^10, adjust=adj)[c("x","y")]) %>% 
    mutate(section = cut(x, breaks=c(-Inf, -1, cc, Inf))) %>% 
    group_by(section) %>% 
    mutate(prob = paste0(round(sum(y)*mean(diff(x))*100),"%"))

  # Get probability mass for each level of section
  # We'll use these as the label values in scale_fill_manual
  sp = dens %>% 
    group_by(section, prob) %>% 
    summarise %>% 
    ungroup

  if(!drop_levs) {
   sp = sp %>% complete(section, fill=list(prob="0%"))
  }

  # Assign colors to each level of section
  col = setNames(c("red","blue","green"), levels(dens$section))

  ggplot(dens, aes(x, y, fill=section)) +
    geom_area() +
    scale_fill_manual(labels=sp$prob, values=col, drop=drop_levs) +
    labs(fill="")
}

现在，让我们在几个不同的数据分布上运行该函数：

set.seed(3)
dat2 = rnorm(1000)
grid.arrange(fill_density(mydata), fill_density(mydata[mydata>0]),
             fill_density(mydata[mydata>2], drop_levs=TRUE), 
              fill_density(mydata[mydata>2], drop_levs=FALSE),
             fill_density(mydata[mydata < -5 | mydata > 5], adj=0.3), fill_density(dat2),
             ncol=2)

set.seed（3）
dat2=rnorm（1000）
网格排列（填充密度（mydata），填充密度（mydata[mydata>0]），
填充密度（mydata[mydata>2]，drop_levs=TRUE），
填充密度（mydata[mydata>2]，drop_levs=FALSE），
填充密度（mydata[mydata<-5 | mydata>5]，调整=0.3），填充密度（dat2），
ncol=2）

---

接近，但不完全正确。我想展示图例，让它等于每个部分对应的概率。几乎正确，但不完全正确。第二个图应该是绿色的，因为所有的东西都在

c

的右边。同样，如果我把C设置为100，所有的东西都应该是蓝色的，因为所有的东西都在100到100之间。第二个情节MyDAT[MyDATA＞2 ]应该是绿色的，因为所有的数据都是C＝＝1的。我没有考虑两个部分具有相同的概率质量的情况（在你标记的情况下都是零）。这将导致重复的因子标签，并且仅使用具有给定因子标签get的节的第一级。我已经在更新的代码中修复了这个问题。我只是简化了代码。我忘记了我可以使用

scale\u fill\u manual

中的

labels

参数将标签设置为概率质量值。我还添加了一个参数，允许您保留或删除缺少的

部分

级别。

set.seed(3)
dat2 = rnorm(1000)
grid.arrange(fill_density(mydata), fill_density(mydata[mydata>0]),
             fill_density(mydata[mydata>2], drop_levs=TRUE), 
              fill_density(mydata[mydata>2], drop_levs=FALSE),
             fill_density(mydata[mydata < -5 | mydata > 5], adj=0.3), fill_density(dat2),
             ncol=2)