R循环矩阵比较第一列是否与另一列相同

时间:2014-08-10 16:47:41

标签: r if-statement

我想检查矩阵的每一列中的值是否与第一列中的值相同:

 x1  <- c("x", "y", "x")
 x2  <- c("x", "y", "y")
 x3  <- c("y", "y","x")

 dat <- data.frame(x1,x2,x3)

   x1 x2 x3
1  x  x  y
2  y  y  y
3  x  y  x

因此,如果x2!= 2,则应该为“1”,否则为NA。我怎样才能做到这一点?在这种情况下的结果将是:

x2     x3 
NA     1
NA     NA
1      NA

我的(不工作)解决方案是:

fun <- function (x) {
for(j in 2: ncol(x)){
  ifelse(x[,1]== j, NA,1)
}
}

乐趣(DAT)

我需要一个函数来执行lapply。我怎样才能做到这一点?非常感谢!

3 个答案:

答案 0 :(得分:3)

使用列操作通常很容易因为参数回收这一事实而采用不同的方法:

 dat <- data.frame(x1,x2,x3, stringsAsFactors=FALSE) # keeps as character 

 ne.dat1 <-  (dat != dat[,1])[ , -1]  # True/False rather than 1,NA
 is.na(ne.dat1) <- !ne.dat1
 ne.dat1
       x2   x3
[1,]   NA TRUE
[2,]   NA   NA
[3,] TRUE   NA

答案 1 :(得分:2)

你可以这样做:

 m1 <- (!(as.character(dat[,1])==dat[,-1])) +1
 m1[] <- c(NA,1)[m1]
 m1
 #    x2 x3
 #[1,] NA  1
 #[2,] NA NA
 #[3,]  1 NA

答案 2 :(得分:0)

使用包装器sapply会更容易,因为它应用了名称到结果:

sapply(dat[-1], 
       function(x) 
          ifelse(as.character(x)==as.character(dat[[1]]), NA, 1)
)
##      x2 x3
## [1,] NA  1
## [2,] NA NA
## [3,]  1 NA