为geom_polygon添加图例

4

我正在尝试使用 geom_point 绘制散点图,其中点被平滑的多边形包围,使用 geom_polygon 实现。

这是我的点数据:

set.seed(1)
df <- data.frame(x=c(rnorm(30,-0.1,0.1),rnorm(30,0,0.1),rnorm(30,0.1,0.1)),y=c(rnorm(30,-1,0.1),rnorm(30,0,0.1),rnorm(30,1,0.1)),val=rnorm(90),cluster=c(rep(1,30),rep(2,30),rep(3,30)),stringsAsFactors=F)

我会根据df$val所在的区间为每个点着色。以下是区间数据:

intervals.df <- data.frame(interval=c("(-3,-2]","(-2,-0.999]","(-0.999,0]","(0,1.96]","(1.96,3.91]","(3.91,5.87]","not expressed"),
                           start=c(-3,-2,-0.999,0,1.96,3.91,NA),end=c(-2,-0.999,0,1.96,3.91,5.87,NA),
                           col=c("#2f3b61","#436CE8","#E0E0FF","#7d4343","#C74747","#EBCCD6","#D3D3D3"),stringsAsFactors=F)

给点分配颜色和间隔:

df <- cbind(df,do.call(rbind,lapply(df$val,function(x){
  if(is.na(x)){
    return(data.frame(col=intervals.df$col[nrow(intervals.df)],interval=intervals.df$interval[nrow(intervals.df)],stringsAsFactors=F))
  } else{
    idx <- which(intervals.df$start <= x & intervals.df$end >= x)
    return(data.frame(col=intervals.df$col[idx],interval=intervals.df$interval[idx],stringsAsFactors=F))
  }
})))

准备颜色以显示每个区间的图例:
df$interval <- factor(df$interval,levels=intervals.df$interval)
colors <- intervals.df$col
names(colors) <- intervals.df$interval

这里是我构建平滑多边形的地方(使用了这个链接提供的函数):

clusters <- sort(unique(df$cluster))
cluster.cols <- c("#ff00ff","#088163","#ccbfa5")


splinePolygon <- function(xy,vertices,k=3, ...)
{
  # Assert: xy is an n by 2 matrix with n >= k.
  # Wrap k vertices around each end.
  n <- dim(xy)[1]
  if (k >= 1) {
    data <- rbind(xy[(n-k+1):n,], xy, xy[1:k, ])
  } else {
    data <- xy
  }
  # Spline the x and y coordinates.
  data.spline <- spline(1:(n+2*k), data[,1], n=vertices, ...)
  x <- data.spline$x
  x1 <- data.spline$y
  x2 <- spline(1:(n+2*k), data[,2], n=vertices, ...)$y
  # Retain only the middle part.
  cbind(x1, x2)[k < x & x <= n+k, ]
}

library(data.table)
hulls.df <- do.call(rbind,lapply(1:length(clusters),function(l){
  dt <- data.table(df[which(df$cluster==clusters[l]),])
  hull <- dt[, .SD[chull(x,y)]]
  spline.hull <- splinePolygon(cbind(hull$x,hull$y),100)
  return(data.frame(x=spline.hull[,1],y=spline.hull[,2],val=NA,cluster=clusters[l],col=cluster.cols[l],interval=NA,stringsAsFactors=F))
}))
hulls.df$cluster <- factor(hulls.df$cluster,levels=clusters)

以下是我的ggplot命令:

library(ggplot2)

p <- ggplot(df,aes(x=x,y=y,colour=interval))+geom_point(cex=2,shape=1,stroke=1)+labs(x="X", y="Y")+theme_bw()+theme(legend.key=element_blank(),panel.border=element_blank(),strip.background=element_blank())+scale_color_manual(drop=FALSE,values=colors,name="DE")
p <- p+geom_polygon(data=hulls.df,aes(x=x,y=y,group=cluster),color=hulls.df$col,fill=NA)

这个代码会产生以下效果:

enter image description here

我的问题是如何在点的图例下添加多边形的图例?我想要一个包含3行的图例,每行根据簇的颜色着色并在相应的线旁显示对应的簇编号。

3个回答

3
稍微修改一下你的代码最后一行,就可以得到略有不同的输出结果,可能会满足你的需求:
p+geom_polygon(data=hulls.df,aes(x=x,y=y,group=cluster, fill=cluster),alpha=0.1)

enter image description here


如果您将多边形放在点之前,则可以使用更高的alpha值。 - Axeman

3

假设您想添加一个the_factor的图例。我的基本思路是:

(1) 通过使用未使用的aes参数,将the_factor放入映射中;aes(xx = the_factor)
(2) 如果(1)影响了某些东西,则使用scale_xx_manual()删除影响。
(3) 使用guides(xx = guide_legend(override.aes = list()))修改legend

在您的情况下,aes(fill)aes(alpha)未使用。前者最好不要使用,因为没有效果。所以我使用了aes(fill=as.factor(cluster))

p <- ggplot(df,aes(x=x,y=y,colour=interval, fill=as.factor(cluster))) +   # add aes(fill=...)
  geom_point(cex=2, shape=1, stroke=1) + 
  labs(x="X", y="Y",fill="cluster") +          # add fill="cluster"
  theme_bw() + theme(legend.key=element_blank(),panel.border=element_blank(),strip.background=element_blank()) + scale_color_manual(drop=FALSE,values=colors,name="DE") +
  guides(fill = guide_legend(override.aes = list(colour = cluster.cols, pch=0))) # add

p <- p+geom_polygon(data=hulls.df,aes(x=x,y=y,group=cluster), color=hulls.df$col,fill=NA)

当然,您可以使用aes(alpha = the_factor))制作相同的图形。由于它具有影响力,您需要使用scale_alpha_manual()进行控制。
g <- ggplot(df, aes(x=x,y=y,colour=interval)) +
  geom_point(cex=2, shape=1, stroke=1, aes(alpha=as.factor(cluster))) +  # add aes(alpha)
  labs(x="X", y="Y",alpha="cluster") +          # add alpha="cluster"
  theme_bw() + theme(legend.key=element_blank(),panel.border=element_blank(),strip.background=element_blank()) + scale_color_manual(drop=FALSE,values=colors,name="DE") +
  scale_alpha_manual(values=c(1,1,1)) +         # add
  guides(alpha = guide_legend(override.aes = list(colour = cluster.cols, pch=0))) # add

g <- p+geom_polygon(data=hulls.df,aes(x=x,y=y,group=cluster), color=hulls.df$col,fill=NA)

enter image description here


2
你所要求的是两个颜色刻度。我的理解是这不可能实现。但是你可以通过一些"欺骗"手段使用填充符号(形状21到25)来给人留下两个颜色刻度的印象。具体方法请参考此链接
p <- ggplot(df, aes(x = x, y = y, fill = interval)) +
  geom_point(cex = 2, shape = 21, stroke = 1, colour = NA)+
  labs(x = "X", y = "Y") +
  theme_bw() +
  theme(legend.key = element_blank(), panel.border = element_blank(), strip.background = element_blank()) +
  scale_fill_manual(drop=FALSE, values=colors, name="DE") + 
  geom_polygon(data = hulls.df, aes(x = x, y = y, colour = cluster), fill = NA) + 
  scale_colour_manual(values = cluster.cols)
p

或者使用填充的多边形,并将alpha值设低

p <- ggplot(df,aes(x=x,y=y,colour=interval))+
  geom_point(cex=2,shape=1,stroke=1)+
  labs(x="X", y="Y")+
  theme_bw() +
 theme(legend.key = element_blank(),panel.border=element_blank(), strip.background=element_blank()) +
  scale_color_manual(drop=FALSE,values=colors,name="DE", guide = guide_legend(override.aes = list(fill = NA))) +
  geom_polygon(data=hulls.df,aes(x=x,y=y,group=cluster, fill = cluster),    alpha = 0.2, show.legend = TRUE) + 
      scale_fill_manual(values = cluster.cols) 
    p

但这可能会使点的颜色难以看到。

网页内容由stack overflow 提供, 点击上面的
可以查看英文原文,
原文链接