创建根据 Z 轴着色的 3D 绘图

Question

 library(Sleuth2)

 mlr<-lm(ex1222$Buchanan2000~ex1222$Perot96*ex1222$Gore2000)


for (i in 0:3) {
           assign(paste("betaHat", i, sep=""), 
           summary(mlr)$coeff[i+1,1])
               }

x<-sort(ex1222$Perot96)
y<-sort(ex1222$Gore2000)


z1 <- outer(x, y, function(a,b) betaHat0+betaHat1*a+betaHat2*b+betaHat3*a*b)
nrz <- nrow(z)
ncz <- ncol(z)

# Create a function interpolating colors in the range of specified colors
jet.colors <- colorRampPalette( c("blue", "red") ) 

# Generate the desired number of colors from this palette
nbcol <- 100
color <- jet.colors(nbcol)

# Compute the z-value at the facet centres
zfacet <- z[-1, -1] + z[-1, -ncz] + z[-nrz, -1] + z[-nrz, -ncz]

# Recode facet z-values into color indices
facetcol <- cut(zfacet, nbcol)

persp(x, y, z1, col=color[facetcol],theta=-30, lwd=.3,xlab="Perot 96", ylab="Gore 2000", zlab="Predicted Votes for Buchanan")

你好，

我正在尝试为上面的图着色。我在想我想要更高的“z”值，颜色更深的红色（或任何颜色）。

任何有关如何实现这一目标的帮助将不胜感激。

此外，您也可以随意提出不同的功能来实现这一点。

谢谢！

编辑....在查看 ?persp 上的示例后，我放置了我的新代码。我想更改颜色，但我对新情节的可读性不太满意

Answer 1

我稍微修改了你的代码。

library(Sleuth2)

使用data 参数通常比使用通过$ 从数据框中提取的预测变量更好：

mlr<-lm(Buchanan2000~Perot96*Gore2000,data=ex1222)

我们可以使用expand.grid() 和predict() 以干净的方式获得回归结果：

perot <- seq(1000,40000,by=1000)
gore <-  seq(1000,400000,by=2000)

如果您希望在观察位置评估方面，可以改用perot <- sort(unique(ex1222$Perot96)); gore <- sort(unique(ex1222$Gore2000))。

pframe <- with(ex1222,expand.grid(Perot96=perot,Gore2000=gore))
mlrpred <- predict(mlr,newdata=pframe)

现在将预测转换为矩阵：

nrz <- length(perot)
ncz <- length(gore)
z <- matrix(mlrpred,nrow=nrz)

我选择了从浅红色（#ffcccc，红色加上相当多的蓝色/绿色）到深红色（#cc0000，一点红色，没有别的）。

jet.colors <- colorRampPalette( c("#ffcccc", "#cc0000") ) 

您也可以使用grep("red",colors(),value=TRUE) 来查看 reds R 内置了什么。

# Generate the desired number of colors from this palette
nbcol <- 100
color <- jet.colors(nbcol)

# Compute the z-value at the facet centres
zfacet <- z[-1, -1] + z[-1, -ncz] + z[-nrz, -1] + z[-nrz, -ncz]
# Recode facet z-values into color indices
facetcol <- cut(zfacet, nbcol)

persp(perot, gore, z,
      col=color[facetcol],theta=-30, lwd=.3,
      xlab="Perot 96", ylab="Gore 2000", zlab="Predicted Votes for Buchanan")

你说你“对情节的可读性不太满意”，但这不是很具体......我会花一些时间在?persp页面上看看你的一些选择是什么......

另一个选择是rgl 包：

library(rgl)
## see ?persp3d for discussion of colour handling
vertcol <- cut(z, nbcol)
persp3d(perot, gore, z,
      col=color[vertcol],smooth=FALSE,lit=FALSE,
      xlab="Perot 96", ylab="Gore 2000", zlab="Predicted Votes for Buchanan")

也许还值得看看car 包中的scatter3d（关于 SO 的其他帖子描述了如何调整它的一些图形属性）。

library(car)
scatter3d(Buchanan2000~Perot96*Gore2000,data=ex1222)