基于逻辑表达式计算矩阵

Question

我需要根据逻辑表达式计算矩阵。

我有一个名为A的矩阵和一个名为B的矩阵：

A=matrix(c(1,1,1,1),nrow=2,ncol=2)
B=matrix(c(2,2,2,2),nrow=2,ncol=2)

根据一个名为lambda的变量是否等于1，我想计算一个矩阵C，它等于A.另一方面，如果lambda等于零，矩阵C应该等于矩阵B.它应该看起来像这样，但它似乎不起作用：

C=ifelse(lambda==1,A,B)

感谢。