r语言 - dplyr突变在列子集上(所有这些列的函数合并)

Question

我有一个带有一些信息和一些测量的数据框架。为了进行测量，我想计算Mahalanobis距离，但我没有进行干净的dplyr诉求。我想拥有类似的东西：

library(anomalyDetection)
test<-data.frame(id=LETTERS[1:10],
                 A = rnorm(10,0,2),
                 B = rnorm(10,5,3))
test<-test%>%
  mutate(MD = mahalanobis_distance(.%>%dplyr::select(one_of(c("A","B")))))

我知道以下作用：

test<-test%>%
  mutate(MD = mahalanobis_distance(test%>%dplyr::select(one_of(c("A","B")))))

但是，如果在突变呼叫之前还有其他步骤，那会分解：

test<-test%>%
  mutate(group = id %in% c(LETTERS[1:5]))%>%
  group_by(group)%>%
  mutate(MD = mahalanobis_distance(test%>%dplyr::select(one_of(c("A","B")))))

Answer 1

我们可以基于逻辑向量进行split，然后使用map_df通过在拆分数据集

上应用mahalanobis_distance来创建'MD'列

library(purrr)
library(dplyr)
library(anomalyDetection)
test %>%
    split(.$id %in% LETTERS[1:5]) %>%  
    map_df(~mutate(., MD = mahalanobis_distance(.[-1])))
#  id          A          B        MD
#1   F -0.7829759 4.22808758 2.9007659
#2   G  2.4246532 5.96043439 1.3520245
#3   H -4.8649537 4.95510794 3.0842137
#4   I  1.2221836 5.36154775 0.2921482
#5   J  0.6995204 5.63616864 0.3708477
#6   A  1.2374543 5.17288708 1.4382259
#7   B -2.7815555 0.06437452 2.1244313
#8   C -2.2160242 2.74747556 0.5088291
#9   D  0.8561507 2.70631852 1.5174367
#10  E -1.6427978 6.23758354 2.4110771

注意：在OP的帖子中创建数据集时没有种子