给定列表的元素，如何在列表中恢复其索引？

Question

我的问题基本上是这样的：

我有一个列表L，其中每个元素都是相同维度的矩阵。我需要将此列表中的每个矩阵乘以外部向量h中的相应元素，然后对所有矩阵求和。

set.seed(101)
L <- replicate(3, matrix(rnorm(4), 2), simplify = FALSE)
h <- 2:4

# I need this
L[[1]] * h[1] + L[[2]] * h[2] + L[[3]] * h[3]

鉴于我需要尝试不同数量的矩阵，并且我有一堆它们，我必须以聪明的方式做到这一点。我的想法是

L1 <- lapply(L, function(x) x * h[x])
L2 <- Reduce('+', L1)

其中“h [x]”将向量h索引列表L内的矩阵x的索引，所以我会得到

 L1 = list(L[[1]] * h[1], L[[2]] * h[2], L[[3]] * h[3])

所以，问题是，如何使用元素本身获取列表中元素的索引？像h [L [[m1]]]得到h [1]。

或者，如果您有其他方法可以解决我的问题，我该怎么做？

Answer 1

我认为您正在寻找mapply() / Map()（Map稍微容易一点，因为它不会尝试简化结果）：

?Map：

  

'Map'将函数应用于给定向量的相应元素...'Map'是'mapply'的简单包装器，它不会尝试简化结果......

?mapply：

  

'mapply'将'FUN'应用于每个...参数的第一个元素，第二个元素，第三个元素，等等

设置示例：

set.seed(101)
L <- replicate(3,matrix(rnorm(4),2),simplify=FALSE)
h <- 2:4

做到：

Reduce("+",Map("*",L,h))

Answer 2

我可能会这样做

Reduce(
  function(z,i) z + L[[i]]*h[i], 
  seq_along(L), 
  init = array(0, dim(L[[1]]))
)

在Ben的示例数据上测试它，我看到相同的结果，所以我猜它可行。

这类似于an answer @eddi posted on my question about linear combos。

Answer 3

使用foreach包，我们可以得到如下结果：

library(foreach)
foreach(i=seq_along(h), .combine="+") %do% {
  L[[i]] * h[i]
}

%do%循环遍历执行逐元素乘法，然后使用.combine参数求和输出。这会输出与Ben的Map - Reduce方法相同的结果。