获得相似对的总和并将结果相乘以获得相同数据框中的新列

时间:2017-08-23 14:46:50

标签: r dplyr tidyr

df1 <- data.frame(first_pair1_1=c(1,2,3,1,2,3,4),second_pair1_1=c(1,3,3,1,4,3,4),third_pair1_1=c(1,3,3,1,4,3,4),first_pair1_2=c(7,4,3,1,4,3,4),second_pair1_2=c(7,4,3,1,4,3,4),third_pair1_2=c(1,3,3,1,4,3,4),first_pair2_1=c(1,2,3,1,2,3,4),second_pair2_1=c(1,3,3,1,4,3,4),third_pair2_1=c(1,3,3,1,4,3,4),first_pair2_2=c(7,4,3,1,4,3,4),second_pair2_2=c(7,4,3,1,4,3,4),third_pair2_2=c(1,3,3,1,4,3,4))

我有以上数据框,我正在尝试进行以下计算:

(first_pair1_1 * second_pair1_1 * third_pair1_1) + 
(first_pair1_2 * second_pair1_2 * third_pair1_2) + 
(first_pair2_1 * second_pair2_1 * third_pair2_1) + 
(first_pair2_2 * second_pair2_2 * third_pair2_2)

我想在同一数据框中获得一个新列的结果。可能会有更多对,但模式将是相同的。

2 个答案:

答案 0 :(得分:2)

这是一个基于R的想法。我们遍历1_1,1_2,...的唯一后缀,并将包含每个后缀的所有列(即1_1 * 1_1 * 1_1等)相乘。然后我们使用rowSums添加它们,即

ind <- unique(sub('.*pair', '', names(df1)))
rowSums(sapply(ind, function(i) Reduce(`*`, df1[grepl(i, names(df1))])))
#[1] 100 132 108   4 192 108 256

答案 1 :(得分:1)

Reduce("+", lapply(split.default(df1, sub(pattern = "^[^_]*_", "", names(df1))),
                   function(a) Reduce("*", a)))
#[1] 100 132 108   4 192 108 256
相关问题