繁体   English   中英

将行作为参数传递给R dplyr mutate中的函数

[英]Passing row as an argument to a function in R dplyr mutate

我正在编写一个程序,该程序计算数据集的元素与其余元素之间的差异。 我正在使用dplyr mutate,我需要将整行作为参数传递给计算差异的函数。 以鸢尾花为例:

#Difference function
diff_func <- function (e1, e2) {
  return(sum(e1-e2))
}

chosenElement <- iris[1,1:4] # Chosen element
elements <- iris[10:50,1:4] # Elements to compare to

elements %>% 
  rowwise() %>% 
  mutate(difference=diff_func(chosenElement, c(Petal.Width, Petal.Length, Sepal.Width, Sepal.Length)))

这可行,但是当我使用整个行时,我想使用“ this”或“ row”之类的东西,而不是指定该行的所有列:

elements %>% 
  rowwise() %>% 
  mutate(difference=diff_func(chosenElement, row))

有谁知道这可以做到吗?

通过复制selectedElement以使尺寸相同,我们可以在base R非常轻松地完成此操作

elementsNew <- elements - chosenElement[col(elements)]

请注意, mutate用于更改/转换单列/多列->单列的值。 当然,我们可以将其他类型的对象放在list 假设“差异”应与“ chosenElement”的对应元素对应,并与“ chosenElement”的对应元素对应,则mutate不会使用diff_func

更新

根据说明,似乎我们需要获取具有对应的selectedElement的列之间的差(此处已复制),然后执行rowSums

elements %>%
        mutate(difference = rowSums(. - chosenElement[col(.)]))

purrr base组合:

do.call(cbind,purrr::map2(elements,chosenElement,function(x,y) x-y))

由于(a - d) + (b - e) + (c - f) == (a + b + c) - (d + e + f) ,因此这只是elements行总和与chosenElements之和之间的chosenElements ,您可以在基数R中执行以下操作:

elements$dfrnce <- rowSums(elements) - sum(chosenElement)

或者,在dplyr

elements %>%
  mutate(dfrnce = rowSums(.) - sum(chosenElement))

暂无
暂无

声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.

 
粤ICP备18138465号  © 2020-2024 STACKOOM.COM