问题描述
mat <- matrix(1:9,ncol = 3)
[,1] [,2] [,3]
[1,] 1 4 7
[2,] 2 5 8
[3,] 3 6 9
现在我想将矩阵的每一列乘以一个标量
scalar = c(1,2,3)
我希望标量的第一个元素乘以第一列,第二个元素乘以第二列,第三个标量乘以第三列。
获得以下输出
[,] 1 8 21
[2,] 2 10 24
[3,] 3 12 27
有人能帮我解决这个问题吗?
解决方法
使用 %*%
和 diag
的另一个基本 R 选项
> mat %*% diag(scalar)
[,1] [,2] [,3]
[1,] 1 8 21
[2,] 2 10 24
[3,] 3 12 27
,
我们复制“标量”并相乘
mat * scalar[col(mat)]
或者用sweep
sweep(mat,2,scalar,`*`)
[,] 3 12 27