将函数应用于同一组中的所有对

Question

我想将一个函数应用于同一组中的所有项目对，例如

示例输入：

Group  Item   Value  
A      1       89   
A      2       76  
A      3       2  
B      4       21  
B      5       10  

所需的输出是同一组中所有项目的函数输出的向量。

例如如果函数是：

addnums=function(x,y){  
  x+y  
}

那么所需的输出将是：

165, 91, 78, 31

我曾尝试在 dplyr 包中使用 summarise 来执行此操作，但这只能在输出为单个值时使用。

Answer 1

我们可以通过 data.table 使用 group by 选项

library(data.table)
setDT(df1)[, combn(Value, 2, FUN = sum), Group]
#   Group  V1
#1:     A 165
#2:     A  91
#3:     A  78
#4:     B  31

如果我们想使用 OP 帖子中的addnums

setDT(df1)[, combn(Value, 2, FUN = function(x) addnums(x[1], x[2])), Group]
 #  Group  V1
#1:     A 165
#2:     A  91
#3:     A  78
#4:     B  31

---

或使用tidyverse

library(dplyr)
library(tidyr)
df1 %>% 
  group_by(Group) %>%
  summarise(Sum = list(combn(Value, 2, FUN = sum)))  %>% 
  unnest
# A tibble: 4 x 2
#  Group   Sum
#  <chr> <int>
#1 A       165
#2 A        91
#3 A        78
#4 B        31

使用addnums

df1 %>% 
 group_by(Group) %>%
 summarise(Sum = list(combn(Value, 2, FUN = 
         function(x) addnums(x[1], x[2])))) %>% 
 unnest

---

或者使用base R 和aggregate

aggregate(Value ~ Group, df1, FUN = function(x) combn(x, 2, FUN = sum))

数据

df1 <- structure(list(Group = c("A", "A", "A", "B", "B"), Item = 1:5, 
    Value = c(89L, 76L, 2L, 21L, 10L)), class = "data.frame", row.names = c(NA, 
-5L))

Answer 2

我们可以为每个Group 提供split Value，然后使用combn 计算每对的sum。

sapply(split(df$Value, df$Group), combn, 2, sum)

#$A
#[1] 165  91  78

#$B
#[1] 31

如果需要作为一个向量，我们可以使用unlist。

unlist(sapply(split(df$Value, df$Group), combn, 2, sum), use.names = FALSE)
#[1] 165  91  78  31

---

如果您对tidyverse 使用我们可以做的相同逻辑的解决方案感兴趣

library(dplyr)
library(purrr)

df %>%
  group_split(Group) %>%
  map(~combn(.x %>% pull(Value), 2, sum)) %>% flatten_dbl

#[1] 165  91  78  31