你如何将ggplot2 grobs与数据联系起来?

时间:2012-01-23 13:48:00

标签: r ggplot2 r-grid

给定一个例如点数的ggplot,你如何找出给定点对应的数据行?

样本图:

library(ggplot2)
(p <- ggplot(mtcars, aes(mpg, wt)) +
    geom_point() +
    facet_wrap(~ gear)
)

我们可以使用grid.ls + grid.get来获取包含点的凹凸。

grob_names <- grid.ls(print = FALSE)$name
point_grob_names <- grob_names[grepl("point", grob_names)]
point_grobs <- lapply(point_grob_names, grid.get)

这最后一个变量包含xy坐标和pointsize等的详细信息(尝试unclass(point_grobs[[1]])),但是如何在mtcars中得到每个点对应的数据行并不明显到。


要回答kohske关于我为什么这样做的问题,我正在使用gridSVG来创建交互式散点图。当您将鼠标滑过一个点时,我想显示上下文信息。在mtcars示例中,我可以显示一个工具提示,其中包含汽车名称或数据框该行的其他值。

到目前为止我的hacky想法是将id列作为不可见的文本标签:

mtcars$id <- seq_len(nrow(mtcars))
p + geom_text(aes(label = id), colour = NA)

然后遍历grob树从点grob到文本grob,并显示由标签索引的数据集行。

这是繁琐的,并不是很普遍。如果有一种方法可以将id值存储在点grob中,那么它会更清晰。

1 个答案:

答案 0 :(得分:7)

此脚本生成一个SVG文件,您可以在其中以交互方式注释这些点。

library(ggplot2)
library(gridSVG)

geom_point2 <- function (...) GeomPoint2$new(...)
GeomPoint2 <- proto(GeomPoint, {
  objname <- "point2"
  draw <- function(., data, scales, coordinates, na.rm = FALSE, ...) {
   data <- remove_missing(data, na.rm, c("x", "y", "size", "shape"), 
        name = "geom_point")
    if (empty(data)) 
        return(zeroGrob())
    name <- paste(.$my_name(), data$PANEL[1], sep = ".")
    with(coordinates$transform(data, scales), ggname(name,
        pointsGrob(x, y, size = unit(size, "mm"), pch = shape, 
            gp = gpar(col = alpha(colour, alpha), fill = fill, label = label,
                fontsize = size * .pt))))
  }}
  )

p <- ggplot(mtcars, aes(mpg, wt, label = rownames(mtcars))) + geom_point2() + facet_wrap(~ gear)
print(p)

grob_names <- grid.ls(print = FALSE)$name
point_grob_names <- sort(grob_names[grepl("point", grob_names)])
point_grobs_labels <- lapply(point_grob_names, function(x) grid.get(x)$gp$label)

library(rjson)
jlabel <- toJSON(point_grobs_labels)

grid.text("value", 0.05, 0.05, just = c(0, 0), name = "text_place", gp = gpar(col = "red"))

script <- '
var txt = null;
function f() {
    var id = this.id.match(/geom_point2.([0-9]+)\\.points.*\\.([0-9]+)$/);
    txt.textContent = label[id[1]-1][id[2]-1];
}

window.addEventListener("load",function(){
    var es = document.getElementsByTagName("circle");
    for (i=0; i<es.length; ++i) es[i].addEventListener("mouseover", f, false);

    txt = (document.getElementById("text_place").getElementsByTagName("tspan"))[0];

},false);
'

grid.script(script = script)
grid.script(script = paste("var label = ", jlabel))

gridToSVG()

您知道我可以上传SVG文件的地方吗?