r - 矩阵中的最大行数

Question

我有一个包含 R 中三列的矩阵。该矩阵可能如下所示:

A=matrix(c(1,2,3,4,0.5,1,7,1.2,3,4,2,1),nrow=4, ncol=3)

我想创建一个基于 A 的矩阵，在 A 的每一行中，该矩阵返回 1 表示该行中的最高值，否则返回 0。因此，在上面的具体情况下，我需要一个如下所示的矩阵:

B=matrix(c(0,0,0,1,0,0,1,0,1,1,0,0),nrow=4,ncol=3)

我尝试在论坛中搜索，但找不到正确的答案。

谢谢。

Answer 1

下面的内容与 Ananda 的答案几乎相同，但如果你的 A 足够大，这些微小的变化可能会在速度方面产生影响

> A<-matrix(rnorm(1000*1000),nrow=1000)
> system.time(t(apply(A, 1, function(x) as.numeric(x == max(x)))))
   user  system elapsed
  0.117   0.024   0.141
> system.time(1*(A==apply(A,1,max)))
   user  system elapsed
  0.056   0.008   0.065