r - 如何有效地交换数据框中的元素？

Question

假设我们有以下数据框:

set.seed(1)
(tmp <- data.frame(x = 1:10, R1 = sample(LETTERS[1:5], 10, replace =
TRUE), R2 = sample(LETTERS[1:5], 10, replace = TRUE)))

    x R1 R2
1   1  B  B
2   2  B  A
3   3  C  D
4   4  E  B
5   5  B  D
6   6  E  C
7   7  E  D
8   8  D  E
9   9  D  B
10 10  A  D

我想做以下事情:如果级别索引之间存在差异因子 R1 和因子 R2 的水平为奇数，则需要在两个因素之间进行切换，可以进行通过以下代码:

for(ii in 1:dim(tmp)[1]) {
   kk <- which(levels(tmp$R2) %in% tmp[ii,'R2'], arr.ind = TRUE) -
which(levels(tmp$R1) %in% tmp[ii,'R1'], arr.ind = TRUE)
   if(kk%%2!=0) { # swap the their levels between the two factors
      qq <- tmp[ii,]$R1
      tmp[ii,]$R1 <- tmp[ii,]$R2
      tmp[ii,]$R2 <- qq
  }
}

更简洁有效的方法来实现这一目标？

P.S.以下情况略有不同。

set.seed(1)
(tmp <- data.frame(x = 1:10, R1 = sample(LETTERS[1:5], 10, replace = 
 TRUE), R2 = sample(LETTERS[2:6], 10, replace = TRUE)))

   x R1 R2
   1  C  B
   2  B  B
   3  C  E
   4  E  C
   5  E  B
   6  D  E
   7  E  E
   8  D  F
   9  C  D
  10  A  E

请注意，R1 和 R2 这两个因素之间的因素水平滑动了一个水平；也就是说，因子 R1 没有级别 F，而因子 R2 没有级别 A。我想根据两个因子的组合级别交换因子级别，如下所示:

tl <- unique(c(levels(tmp$R1), levels(tmp$R2)))
for(ii in 1:dim(tmp)[1]) {
   kk <- which(tl %in% tmp[ii,'R2'], arr.ind = TRUE) - which(tl %in% 
      tmp[ii,'R1'], arr.ind = TRUE)
   if(kk%%2!=0) { # swap the their levels between the two factors
      qq <- tmp[ii,]$R1
      tmp[ii,]$R1 <- tmp[ii,]$R2
      tmp[ii,]$R2 <- qq
  }
}

这个案子该怎么办？谢谢!

Answer 1

#Find out the indices where difference is odd
inds = abs(as.numeric(tmp$R1) - as.numeric(tmp$R2)) %% 2 != 0

#create new columns where values for the appropriate inds are from relevant columns
tmp$R1_new = replace(tmp$R1, inds, tmp$R2[inds])
tmp$R2_new = replace(tmp$R2, inds, tmp$R1[inds])

tmp
#    x R1 R2 R1_new R2_new
#1   1  B  B      B      B
#2   2  B  A      A      B
#3   3  C  D      D      C
#4   4  E  B      B      E
#5   5  B  D      B      D
#6   6  E  C      E      C
#7   7  E  D      D      E
#8   8  D  E      E      D
#9   9  D  B      D      B
#10 10  A  D      D      A

如有必要，删除旧的 R1 和 R2