在R中套用功能族

Question

我有这种形式的双循环：

for (j in 1:col(mat))
{
    for (i in 1:nrow(mat))
    {
       decision[i][j] = ifelse(mat[i][j] > 3, 1, 0)
    }
}

是否可以通过应用功能之一以显着提高的速度来实现此功能？

Answer 1

您不需要任何循环。 如果您创建一个看起来像这样的矩阵

mat<-matrix(sample(1:10, 5*7, replace=T), ncol=7)
#     [,1] [,2] [,3] [,4] [,5] [,6] [,7]
#[1,]    9    6   10    7    6   10    6
#[2,]    7    6    3    3    6    8    3
#[3,]    7    9    7    5    6    7    6
#[4,]    2    3    8    6   10    1    5
#[5,]    4    1    5    6    1   10    6

那你就可以做

decision <- (mat>3)+0
decision;
#     [,1] [,2] [,3] [,4] [,5] [,6] [,7]
#[1,]    1    1    1    1    1    1    1
#[2,]    1    1    0    0    1    1    0
#[3,]    1    1    1    1    1    1    1
#[4,]    0    0    1    1    1    0    1
#[5,]    1    0    1    1    0    1    1

Answer 2

R是向量化语言，因此对于这种对矩阵的简单操作，不需要apply类型函数，只需执行以下操作：

decision <- ifelse(mat > 3, 1, 0)

（我假设您要遍历矩阵的元素，这意味着您将在循环中说ncol(mat) ； col给出了一些完全不同的东西）。