Агрегированная корреляция (R :: dplyr)

Question

Я пытаюсь вычислить матрицу корреляции при различных поднаборах кадра данных. Я нашел этот фрагмент кода для вычисления корреляции между 2-х переменными в кадре данных:

library(dplyr) 
mtcars %>% group_by(cyl) %>% summarise(V1=cor(hp,wt)) 

Но я хотел бы, чтобы вычислить матрицу корреляции между несколькими переменными, в кадре данных. Я бы хотел, чтобы это возвращалось (желательно) в виде списка корреляционных матриц. Что-то вроде:

mtcars %>% group_by(cyl) %>% cor(data.frame(hp,wt,qsec) 

Могу ли я сделать это с помощью dplyr?

Answer 1

На мой взгляд хороший старый by или dlply лучше, но если вы действительно хотите использовать dplyr, я думаю, что вы можете использовать do:

o <- mtcars %>% group_by(cyl) %>% do(cor=cor(cbind(.$hp, .$wt, .$qsec))) 
# Source: local data frame [3 x 2] 
# Groups: <by row> 

# cyl  cor 
# 1 4 <dbl[3,3]> 
# 2 6 <dbl[3,3]> 
# 3 8 <dbl[3,3]> 

где . относится к отфильтрованной dataframe. Тогда вы могли бы сделать o$cor[1] и т. Д. Я не уверен, как просто получить вывод списка из dplyr, а не вывод dataframe.

---

Использование plyr:

library(plyr) 
dlply(mtcars, .(cyl), function (x) cor(x[, c('hp', 'wt', 'qsec')])) 

Использование базового R и by:

o <- by(mtcars[, c('hp', 'wt', 'qsec')], mtcars$cyl, cor, simplify=F) 

o имеет класс by, но ?by говорит, что это в основном списке.

length(o) # 3 
names(o) # "4" "6" "8" (i.e. the cyl values) 
o[[1]] # =cor(hp, wt, qsec) where cyl==4