r - 在 R 中必要时矩阵未填充 1

Question

我正在尝试用 (1) 填充矩阵，其中索引对应于相同维度的另一个矩阵的最大值。
下面是一个可重现的例子

set.seed(1)
matric <- matrix(rnorm(3000), nrow = 500, ncol = 6)

best <- matrix(0, nrow = 500, ncol = 6)
n_end <- 500

for (i in n_end) {
  best[which(matric[i,] == max(matric[i,]))] <- 1
}
## the best model has the higher sum
best1 <- colSums(best)

请帮忙!! :)

Answer 1

我们可以使用 seq_len或 1:n_end在循环内部，当我们分配给“最佳”的同一行时，也对该矩阵进行索引

for(i in 1:n_end) {
  best[i,][matric[i,] == max(matric[i,])] <- 1
 }

或者如果每行只有一个最大值，则使用 max.col因为它更快且矢量化

best[cbind(seq_len(nrow(best)), max.col(matric, 'first'))] <- 1

或与 rowMaxs来自 matrixStats

library(matrixStats)
+(matric == rowMaxs(matric))