R中自定义函数的等高线图

时间:2013-09-29 14:26:01

标签: r plot contour

我正在使用一些自定义函数,我需要根据参数的多个值为它们绘制轮廓。

这是一个示例函数:

enter image description here

我需要绘制这样的轮廓图:

enter image description here

有什么想法吗?

感谢。

2 个答案:

答案 0 :(得分:14)

首先构造一个函数fourvar,将这四个参数作为参数。在这种情况下,您可以使用3个变量完成它,其中一个变量是lambda_1上的lambda_2。 Alpha1固定为2,因此alpha_1 / alpha_2将在0-10之间变化。

fourvar <- function(a1,a2,l1,l2){ 
  a1* integrate( function(x) {(1-x)^(a1-1)*(1-x^(l2/l1) )^a2} , 0 , 1)$value }

诀窍是要意识到integrate函数返回一个列表,你只想要该列表的'value'部分,因此它可以是Vectorize() - ed。

其次,您使用该函数构造矩阵:

  mat <- outer( seq(.01, 10, length=100),  
                seq(.01, 10, length=100), 
                Vectorize( function(x,y) fourvar(a1=2, x/2, l1=2, l2=y/2) ) )

然后,只能使用lattice::contourplot轻松完成在这些位置创建带标签的情节的任务。在进行了合理的搜索后,似乎geom_contour标记的解决方案仍然是ggplot2中正在进行的工作。我发现的唯一标签策略是外部包装。但是,在这种情况下,'directlabels'包的函数directlabel似乎没有足够的控制来正确地传播标签。在我看到的其他例子中,它确实在标绘区域周围散布标签。我想我可以查看代码,但由于它依赖于'proto'包,它可能会被奇怪地封装,所以我没看过。

require(reshape2)
mmat <- melt(mat)
str(mmat) # to see the names in the melted matrix
g <- ggplot(mmat, aes(x=Var1, y=Var2, z=value) )
g <- g+stat_contour(aes(col = ..level..), breaks=seq(.1, .9, .1) )
g <- g + scale_colour_continuous(low = "#000000", high = "#000000") # make black
install.packages("directlabels", repos="http://r-forge.r-project.org", type="source")
require(directlabels)
direct.label(g)

请注意,这些是矩阵的索引位置,而不是参数的比例,但这应该很容易修复。

enter image description here

另一方面,这是一个在格子中构建它的容易程度(我认为它看起来更“干净”:

  require(lattice)
  contourplot(mat, at=seq(.1,.9,.1))

enter image description here

答案 1 :(得分:1)

我认为这个问题仍然很重要,metR包中的轮廓图标签已经有了一些发展。添加到前面的示例中,您还可以使用ggplot2

require(metR)
g + geom_text_contour(rotate = TRUE, nudge_x = 3, nudge_y = 5)

Labeled contour plot