在分面后将相关系数放在ggplot散点图上

时间:2017-09-15 15:55:58

标签: r ggplot2

我有问题在facet_wrap之后用另一个变量将相关系数放在我的散点图上。 下面是我使用mtcars数据集进行说明的示例。 当我绘制出来时,两个图都具有相同的相关数。似乎没有为每个方面计算相关系数。我无法找到实现这一目标的方法。真的很感激,如果有人可以帮助那个......

library(ggplot2)
library(dplyr)
corr_eqn <- function(x,y, method='pearson', digits = 2) {
    corr_coef <- round(cor.test(x, y, method=method)$estimate, digits = digits)
    corr_pval <- tryCatch(format(cor.test(x,y, method=method)$p.value, 
                                 scientific=TRUE),
                          error=function(e) NA)
    paste(method, 'r = ', corr_coef, ',', 'pval =', corr_pval)
}

sca.plot <- function (cor.coef=TRUE) {
    df<- mtcars %>% filter(vs==1)
    p<- df %>% 
        ggplot(aes(x=hp, y=mpg))+
        geom_point()+
        geom_smooth()+
        facet_wrap(~cyl, ncol=3)

    if (cor.coef) {
        p<- p+geom_text(x=0.9*max(df$hp, na.rm=TRUE),
                        y=0.9*max(df$mpg, na.rm=TRUE),
                        label = corr_eqn(df[['hp']],df[['mpg']],
                                         method='pearson'))
    }
    return (p)    
}

sca.plot(cor.coef=TRUE)

1 个答案:

答案 0 :(得分:3)

通过变量inputFacet调用facet,遍历此变量以计算corr_enq并使用变量名称get绘制构面。

在闪亮的情况下,您可能会将用户输入设为input$facet,此处称为inputFacet。我们在facet_wrap(~ get(inputFacet), ncol = 3)中绘制了获取此变量的主要情节。接下来,我们使用for(i in seq_along(resCor$facets))遍历所有构面选项,并将结果存储在rescore中。

这应解决“不计算每个方面的相关系数”问题。

library(dplyr)
library(ggplot2)

inputFacet <- "cyl"
cor.coef = TRUE
df <- mtcars

p <- df %>% 
    ggplot(aes(hp, mpg))+
    geom_point()+
    geom_smooth()+
    facet_wrap(~ get(inputFacet), ncol = 3)

if (cor.coef) {

    resCor <- data.frame(facets = unique(mtcars[, inputFacet]))
    for(i in seq_along(resCor$facets)) {
        foo <- mtcars[mtcars[, inputFacet] == resCor$facets[i], ]
        resCor$text[i] <- corr_eqn(foo$hp, foo$mpg)
    }
    colnames(resCor)[1] <- inputFacet

    p <- p + geom_text(data = resCor, 
                       aes(0.9 * max(df$hp, na.rm = TRUE),
                           0.9 * max(df$mpg, na.rm = TRUE),
                           label = text))

}

p

enter image description here