矩阵中的最大行数

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

也许是这样的？

t(apply(A, 1, function(x) as.numeric(x == max(x))))
#      [,1] [,2] [,3]
# [1,]    0    0    1
# [2,]    0    0    1
# [3,]    0    1    0
# [4,]    1    0    0

请注意，在多个值与行中的最大值匹配的情况下，每行可能有多个“1”。

Answer 2

以下几乎与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