问题描述
我想在基数 R 中制作可变数量的矩阵/向量的成对乘积。
我只有这个丑陋的解决方案(丑陋的是 <<-
),但直觉上认为存在更好的 - 也许是递归的 - 方式,或者甚至是一个函数。我需要一个成对版本的 prod
。
f1 <- function(...) {
input <- list(...)
output <- input[[1]]
sapply(2:length(input),function(m) output <<- output*input[[m]])
return(output)
}
m1 <- matrix(1:6,ncol = 2)
m2 <- matrix(6:1,ncol = 2)
m3 <- 1/matrix(6:1,ncol = 2)
all(f1(m1,m2,m3) == m1*m2*m3) #[1] TRUE
解决方法
使用Reduce
:
f1 <- function(...) {
Reduce(`*`,list(...))
}
all(f1(m1,m2,m3) == m1*m2*m3)
#[1] TRUE