r - R-通过图例的z值对颜色散点图进行点分配

我有一个散点图，希望通过为每个点分配的z值为点着色。然后，我想在图的右侧获得图例，以使用良好的平滑色谱显示哪些颜色对应于z值。

这是一些您可以使用的x,y,z值，因此这是一个可重现的示例。

x = runif(50)
y = runif(50)
z = runif(50) #determines color of the (x,y) point

我想最好的答案是针对任何颜色函数的通用答案，但我确实希望使用rainbow()

最佳答案

如果您不想使用ggplot2，则我修改了其他人提供的解决方案，我不记得是谁。

scatter_fill <- function (x, y, z,xlim=c(min(x),max(x)),ylim=c(min(y),max(y)),zlim=c(min(z),max(z)),
                          nlevels = 20, plot.title, plot.axes,
                          key.title, key.axes, asp = NA, xaxs = "i",
                          yaxs = "i", las = 1,
                          axes = TRUE, frame.plot = axes, ...)
{
  mar.orig <- (par.orig <- par(c("mar", "las", "mfrow")))$mar
  on.exit(par(par.orig))
  w <- (3 + mar.orig[2L]) * par("csi") * 2.54
  layout(matrix(c(2, 1), ncol = 2L), widths = c(1, lcm(w)))
  par(las = las)
  mar <- mar.orig
  mar[4L] <- mar[2L]
  mar[2L] <- 1
  par(mar = mar)

# choose colors to interpolate
levels <- seq(zlim[1],zlim[2],length.out = nlevels)
col <- colorRampPalette(c("red","yellow","dark green"))(nlevels)
colz <- col[cut(z,nlevels)]
#
plot.new()
plot.window(xlim = c(0, 1), ylim = range(levels), xaxs = "i", yaxs = "i")

rect(0, levels[-length(levels)], 1, levels[-1L],col=col,border=col)
if (missing(key.axes)) {if (axes){axis(4)}}
       else key.axes
   box()
   if (!missing(key.title))
     key.title
   mar <- mar.orig
   mar[4L] <- 1
   par(mar = mar)

   # points
   plot(x,y,type = "n",xaxt='n',yaxt='n',xlab="",ylab="",xlim=xlim,ylim=ylim,bty="n")
   points(x,y,col = colz,xaxt='n',yaxt='n',xlab="",ylab="",bty="n",...)

   ## options to make mapping more customizable

        if (missing(plot.axes)) {
          if (axes) {
            title(main = "", xlab = "", ylab = "")
            Axis(x, side = 1)
            Axis(y, side = 2)
          }
        }
        else plot.axes
        if (frame.plot)
          box()
        if (missing(plot.title))
          title(...)
        else plot.title
        invisible()
 }

只需先运行该函数即可使用。很方便。

# random vectors
vx <- rnorm(40,0,1)
vy <- rnorm(40,0,1)
vz <- rnorm(40,10,10)

scatter_fill(vx,vy,vz,nlevels=15,xlim=c(-1,1),ylim=c(-1,5),zlim=c(-10,10),main="TEST",pch=".",cex=8)

如您所见，它继承了通常的绘图功能。