如何使用 R 交换 data.table 中的列值

Question

我有下面给出的玩具数据。

library(data.table)
(tmp <- data.table(R1 = c('D','D','D','T','C'), y = 10:1, R2 = c('D','A','Z','D','D')))

    R1  y R2
 1:  D 10  D
 2:  D  9  A
 3:  D  8  Z
 4:  T  7  D
 5:  C  6  D
 6:  D  5  D
 7:  D  4  A
 8:  D  3  Z
 9:  T  2  D
10:  C  1  D

我想交换 R1 和 R2 列中的值，以便所有 A 都列在 R1 下，而不常见的值转到 R2。有人可以告诉我怎么做吗？这是所需的输出。

    R1  y R2
 1:  D 10  D
 2:  D  9  A
 3:  D  8  Z
 4:  D  7  T
 5:  D  6  C
 6:  D  5  D
 7:  D  4  A
 8:  D  3  Z
 9:  D  2  T
10:  D  1  C

以下是下面提供的答案的性能结果 -

Unit: milliseconds
    expr      min       lq     mean   median       uq       max neval cld
   akrun 5.524562 5.587740 7.526681 5.605406 5.938955 14.976740     5   b
 r2evans 1.466862 1.489944 1.509321 1.500263 1.536402  1.553134     5  a

Answer 1

基于更新，我们可以在 i 上指定一个逻辑表达式并交换列值以分配

library(data.table)
val <- "D"
tmp[R2 == val, c("R1", "R2") := .(R2, R1)]

-输出

> tmp
    R1  y R2
 1:  D 10  D
 2:  D  9  A
 3:  D  8  Z
 4:  D  7  T
 5:  D  6  C
 6:  D  5  D
 7:  D  4  A
 8:  D  3  Z
 9:  D  2  T
10:  D  1  C

Answer 2

我怀疑其他答案可能是最适用的，但如果您的需求不是基于字典排序（实际上只是存在于一组“有趣”值中），那么

interesting <- c("A")
tmp[, c("R1", "R2") := .(
  fifelse(R2 %in% interesting & !R1 %in% interesting, R2, R1),
  fifelse(R2 %in% interesting & !R1 %in% interesting, R1, R2))]
tmp
#         x     R1     R2
#     <int> <char> <char>
#  1:     1      A      A
#  2:     2      A      F
#  3:     3      A      T
#  4:     4      A      G
#  5:     5      A      I
#  6:     6      A      A
#  7:     7      A      F
#  8:     8      A      T
#  9:     9      A      G
# 10:    10      A      I

我承认这看起来有点笨拙，双重计算条件。这可以很容易地作为 tmp 框架内部或外部的临时变量更有效地工作，例如：

tmp[, swap := R2 %in% interesting & !R1 %in% interesting
  ][, c("R1", "R2") := .(fifelse(swap, R2, R1), fifelse(swap, R1, R2))
  ][, swap := NULL]

如果您确定 R2 %in% interesting 与 !R2 %in% interesting 完全对齐（也就是说，R1 和 R2 永远不会是 interesting, ... 或者如果两者都很有趣你不关心交换，如第 1 行和第 6 行），那么你可以将其简化为

tmp[, c("R1", "R2") := .(
  fifelse(R2 %in% interesting, R2, R1),
  fifelse(R2 %in% interesting, R1, R2))]

如何使用 R 交换 data.table 中的列值

How to swap column values in a data.table using R

r

data.table