快速获取所有矩阵列元素乘积对的方法 [英] Fast way to get all pairs of matrix column element-wise products
本文介绍了快速获取所有矩阵列元素乘积对的方法的处理方法,对大家解决问题具有一定的参考价值,需要的朋友们下面随着小编来一起学习吧!
问题描述
假设我有一个数字矩阵
:
set.seed(1)
mat <- matrix(rnorm(1000), ncol = 100)
我想生成所有向量,这些向量是 mat
中所有唯一向量对的元素乘积的结果.
I want to generate all vectors that are the result of the element-wise product of all unique pairs of vectors in mat
.
我们如何改进以下代码:
How can we improve below code:
all.pairs <- t(combn(1:ncol(mat), 2))
res <-
do.call(cbind,
lapply(1:nrow(all.pairs),
function(p) mat[, all.pairs[p, 1]] * mat[, all.pairs[p, 2]]))
推荐答案
我们可以:
n <- ncol(mat)
lst <- lapply(1:n, function (i) mat[,i] * mat[,i:n])
do.call(cbind, lst)
或者,这是一种更快的方法:
Or, here is an even faster way:
n <- ncol(mat)
j1 <- rep.int(1:n, n:1)
j2 <- sequence(n:1) - 1L + j1
mat[, j1] * mat[, j2]
<小时>
请注意,以上将包括一列与其自身的乘法.如果你想禁止,使用
Note, the above will include the multiplication of a column to itself. If you want to forbid that, use
n <- ncol(mat)
lst <- lapply(1:(n-1), function (i) mat[,i] * mat[,(i+1):n])
do.call(cbind, lst)
和
n <- ncol(mat)
j1 <- rep.int(1:(n-1), (n-1):1)
j2 <- sequence((n-1):1) + j1
mat[, j1] * mat[, j2]
<小时>
实际上,上面创建的j1
和j2
只是combn(1:ncol(mat),2)
的第一行和第二行.因此,如果您仍想继续使用 combn
,请使用
Actually, j1
and j2
created above are just the 1st and 2nd row of combn(1:ncol(mat),2)
. So, if you still want to stay with combn
, use
all.pairs <- combn(1:ncol(mat),2)
mat[, all.pairs[1,]] * mat[, all.pairs[2,]]
这篇关于快速获取所有矩阵列元素乘积对的方法的文章就介绍到这了,希望我们推荐的答案对大家有所帮助,也希望大家多多支持IT屋!
查看全文