使用轮廓图的x，y轴值不正确

Question

我想使用ggplot库创建轮廓图。 我检查了有关此主题的文档 ，并找到了完成此任务的代码。 不幸的是，代码使用存储表面的矩阵z的索引x和y。 如何将其更改为x和y的实际值？ 在我的代码下面，生成轮廓图。

objective_function <- function(vec) {
  basin_function <- function(vec){
    if(all(vec == 0)) {
      return(0)
    } else{
      return(sum(exp(-2.0/vec^2)+sin(vec*pi*2)))
    }
  }
  return(basin_function(vec))
}

objective_function_wrapper <- function(x_vec, y_vec) {
  vec <-rbind(x_vec,y_vec)
  return(apply(vec,2, objective_function))
}

plotSurf <-function(){
    x <- y <-seq(from=-5, to =5,  by=0.1)
    z <- outer(x,y, objective_function_wrapper)
    surf3d <- melt(z)
    names(surf3d) <- c("x", "y", "z")
    p1 <- ggplot(data=surf3d, aes(x=x, y=y, z=z))
    p1 <- p1 + geom_tile(aes(fill=z))+stat_contour()
    print(p1)
}

plotSurf()

Answer 1

您需要将x和y替换为当前位于外部调用的“融化”结果中的行和列编号：

x <- y <-seq(from=-5, to =5,  by=0.1)
z <- outer(x,y, objective_function_wrapper)
surf3d <- melt(z)
names(surf3d) <- c("x", "y", "z")
surf3d$x <- rep(x, ncol(z) ); surf3d$y <- rep(y, each=nrow(z) )