Question

我正在尝试从表中提取一些信息，并且我试图避免任何 for 循环或应用类型函数。

假设向量m

 m=c(1:20)  
 m
 [1]  1  2  3  4  5  6  7  8  9 10 11 12 13 14 15 16 17 18 19 20

和矩阵g

x1=c(0,1,0,1,2,0,1,2,3,0,1,2,3,4,0,1,2,3,4,5)   
x2=c(1,0,2,1,0,3,2,1,0,4,3,2,1,0,5,4,3,2,1,0)  
u=.4*x1^.5+.6*x2^.5  
g=cbind(x1,x2,u)  
 g
      x1 x2         u  
 [1,]  0  1 0.6000000  
 [2,]  1  0 0.4000000  
 [3,]  0  2 0.8485281  
 [4,]  1  1 1.0000000    
 [5,]  2  0 0.5656854  
 [6,]  0  3 1.0392305  
 [7,]  1  2 1.2485281  
 [8,]  2  1 1.1656854  
 [9,]  3  0 0.6928203  
[10,]  0  4 1.2000000  
[11,]  1  3 1.4392305  
[12,]  2  2 1.4142136  
[13,]  3  1 1.2928203  
[14,]  4  0 0.8000000  
[15,]  0  5 1.3416408  
[16,]  1  4 1.6000000  
[17,]  2  3 1.6049159  
[18,]  3  2 1.5413485  
[19,]  4  1 1.4000000  
[20,]  5  0 0.8944272

我想要m的每个元素，检查总和g [，1] + g [，2]是否等于该元素。对于条件为TRUE的所有情况，我希望我的代码返回具有最高g [，3]值的那个的位置。例如，当m = 5时，条件x [，1] + x [，2] == 5在15,16,17,18,19和20处为TRUE。在这6中，条目17具有最高值，所以我希望我的代码返回值17.

所以最后，我期望一个长度=长度（m）的向量，它将指示m的每个元素，其中g [，3]的最大值满足上述条件。矢量应该是这样的：

1,4,7,11,17,0,0,0,0,0,0...,0

当m = 1时，g [，1] + g [，2] == 1的行是行1 当m = 2时，g [，1] + g [，2] == 2的行是第4行，依此类推。

目前我在g的每一行使用 apply ，但此过程重复了数千次，而且我的代码非常慢。为了加快速度，我采用了 ifelse 来努力以矢量化的方式做事。

我要做的是以下内容：

ifelse(m>0,which(g[,3]==max(g[which(g[,1]+g[,2]==m),3])),0)

当我跑步时，我正在

 [1] 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1

如果我用5代替m，则返回17s的向量。它似乎只使用m的第一个元素而不是整个向量。任何关于如何使这项工作或可以做同样工作的替代方案的建议都非常受欢迎。

Answer 1

我们可以将outer与max.col

一起使用

m1 <- t(outer( g[,1] + g[,2], m,  `==`)* g[,3])
max.col(m1) * (rowSums(m1!= 0) !=0)
#[1]  1  4  7 11 17  0  0  0  0  0  0  0  0  0  0  0  0  0  0  0

使用ifelse从矩阵中提取信息

1 个答案: