如何将ggplot2 grobs与数据相关联?

8

如果给定一个ggplot,例如点的图表,您如何找出给定点对应的数据行?

一个示例图:

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

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

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

这个最后的变量包含了x-y坐标、点大小等细节(尝试使用 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树,并显示由标签索引的数据集的行。

这很麻烦,也不太通用。如果有一种方法可以在点grob中存储id值,那么它将更加简洁。


没有简单的方法。你的最终目标是什么?可能有其他选择。 - kohske
@kohske:评论太长了,所以我更新了问题来解释我在做什么。 - Richie Cotton
非常有趣。我了解你的目的。不幸的是,我们无法通过grob访问数据(即使在下一个版本中也很可能如此)。但也许了解数据的信息会很有用(特别是当 stat="identity" 时)。我会进行讨论。 - kohske
一些技术问题是ggplot2支持在一个面板上显示多个数据。因此,“id”可能不足以识别数据行。 - kohske
1个回答

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文件的地方吗?

太棒了!不过我必须删除 filename = "s.js" 参数才能让这个对我起作用。 - joran
我知道这条信息比较老,但是我还是遇到了错误Error in eval(expr, envir, enclos) : could not find function "proto"。这是版本问题吗? - Hugh
也许你需要使用 library(proto),不过我没有测试过。 - kohske
1
我遇到了相同的错误,我添加了library(proto)library(grid)library(scales),并将GeomPoint替换为ggplot2:::GeomPointpointsGrob替换为grid:::pointsGrobzeroGrob替换为ggplot2:::zeroGrob,这可能有所帮助,但仍然出现了Error: attempt to apply non-function... - PatrickT
不行!proto(GeomPoint, { : object 'GeomPoint' not found出错了。执行已停止。 - vjain419

网页内容由stack overflow 提供, 点击上面的
可以查看英文原文,
原文链接