R语言:将向量与数据框每行对应列的SUMPRODUCT结果赋值到prod列
解决R中复刻Excel SUMPRODUCT函数的问题
嘿,我懂你想实现Excel里SUMPRODUCT的功能——计算数据框每行指定列和向量的加权和对吧?其实在R里有好几种简单靠谱的方法,我给你详细说说:
方法1:用基础R的apply函数(直观易理解)
这是最直观的逐行处理方式,适合刚接触R的朋友:
# 先定义你的数据框和向量 df <- data.frame( prod = c(NA, NA, NA), sd = c(NA, NA, NA), c1 = c(0.5, 0.5, 0.5), c2 = c(0.25, 0.5, 0), c3 = c(0.25, 0, 0.5) ) v <- c(0.1, 0.2, 0.3) # 计算每行的SUMPRODUCT并赋值给prod列 df$prod <- apply(df[, c("c1", "c2", "c3")], 1, function(row) sum(row * v))
这里apply的第二个参数1表示按行操作,匿名函数里把当前行的数值和向量v对应相乘后求和,完全对应Excel里SUMPRODUCT的逻辑。
方法2:矩阵乘法(高效适合大数据)
如果你的数据量很大,矩阵乘法的效率会比apply高很多——因为矩阵运算在R里是向量化执行的:
df$prod <- as.matrix(df[, c("c1", "c2", "c3")]) %*% v
把指定列转成矩阵后,和向量v做矩阵乘法,结果就是每行的点积(也就是SUMPRODUCT的结果),R会自动把矩阵结果转换成向量赋值给prod列。
方法3:用dplyr的行操作(适合tidyverse用户)
如果你平时习惯用tidyverse系列工具,用dplyr的行-wise操作也能轻松实现:
library(dplyr) df <- df %>% rowwise() %>% # 开启逐行处理模式 mutate(prod = sum(c_across(c1:c3) * v)) %>% # 计算每行的加权和 ungroup() # 取消行分组,避免影响后续操作
rowwise()让后续的mutate按行执行,c_across(c1:c3)选中需要计算的列,再和v相乘求和即可。
验证结果
三种方法得到的结果都是一致的:
- 第一行prod:
0.175 - 第二行prod:
0.15 - 第三行prod:
0.2
内容的提问来源于stack exchange,提问作者user12135864
相关产品推荐
相关产品推荐

