使用等高线图,x,y轴的值不正确

时间:2015-01-02 22:53:17

标签: r ggplot2

我想使用ggplot库创建等高线图。我检查了这个主题的documentation,并找到了完成此操作的代码。不幸的是,代码使用存储表面的矩阵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()

1 个答案:

答案 0 :(得分:1)

您需要将x - 和y - 值替换为当前位于外部调用的&#34; melt&#34; -ed结果中的行号和列号:

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) )