R 中所有列向量的相关性 (dplyr)

Question

我在 R 中有一个数据框（取自 dplyr 网站 此处）

library(dplyr)
gdf <-
  tibble(g = c(1, 1, 2, 3), v1 = 10:13, v2 = 20:23) %>%
  group_by(g)
gdf

： ：

# A tibble: 4 × 3
# Groups:   g [3]
      g    v1    v2
  <dbl> <int> <int>
1     1    10    20
2     1    11    21
3     2    12    22
4     3    13    23

现在我有一个向量：

y <- rnorm(4);y

我想同时测量 y 与 v1 的相关性以及 y 与 v2 的相关性。

across() 函数可能可以完成这项工作

gdf %>% mutate(across(v1:v2, ~ cor(.x,y)))

，但 R 报告我一个错误：

Error: Problem with `mutate()` input `..1`.
ℹ `..1 = across(v1:v2, ~cor(.x, y))`.
x incompatible dimensions
ℹ The error occurred in group 1: g = 1.
Run `rlang::last_error()` to see where the error occurred.

Answer 1

由于 cor() 要求 x 和 y 具有相同的维度，因此您不能将行分组在一起，否则它们将没有 4 个元素可以匹配y 中有 4 个值。

准备数据和库

library(dplyr)

gdf <-
  tibble(g = c(1, 1, 2, 3), v1 = 10:13, v2 = 20:23)

y <- rnorm(4)
[1] 0.59390132 0.91897737 0.78213630 0.07456498

mutate()

如果要在输出中保留 v1 和 v2，请使用 .names 参数来指示新的专栏。 {.col} 指的是 across 正在作用的列名称。

gdf %>% mutate(across(v1:v2, ~ cor(.x,y), .names = "{.col}_cor"))

# A tibble: 4 x 5
      g    v1    v2 v1_cor v2_cor
  <dbl> <int> <int>  <dbl>  <dbl>
1     1    10    20 -0.591 -0.591
2     1    11    21 -0.591 -0.591
3     2    12    22 -0.591 -0.591
4     3    13    23 -0.591 -0.591

summarise()

如果您只想在结果中输出 cor() ，则可以使用 summarise

gdf %>% summarize(across(v1:v2, ~ cor(.x,y)))

# A tibble: 1 x 2
      v1     v2
   <dbl>  <dbl>
1 -0.591 -0.591

Answer 2

使用基本 R：

cor(gdf[,-1], y)

#>         [,1]
#> v1 0.5080586
#> v2 0.5080586

---

另一种可能的解决方案，基于 purrr::map_dfc：

library(tidyverse)

gdf <-
  tibble(g = c(1, 1, 2, 3), v1 = 10:13, v2 = 20:23)

set.seed(123)
y <- rnorm(4)

map_dfc(gdf[,-1], ~ cor(.x, y))

#> # A tibble: 1 × 2
#>      v1    v2
#>   <dbl> <dbl>
#> 1 0.508 0.508