使用查找表更改数据框的多个列中的值

时间:2015-07-26 03:18:25

标签: r merge multiple-columns lookup lookup-tables

我正在尝试使用查找表一次更改多个列的值。它们都使用相同的查找表。我知道如何只针对一列执行此操作 - 我只使用merge,但我遇到了多列问题。

下面是一个示例数据框和一个示例查找表。我的实际数据要大得多(~10K列,8行)。

example <- data.frame(a = seq(1,5), b = seq(5,1), c=c(1,4,3,2,5))

lookup <- data.frame(number = seq(1,5), letter = LETTERS[seq(1,5)])

理想情况下,我最终会得到一个如下所示的数据框:

example_of_ideal_output <- data.frame(a = LETTERS[seq(1,5)], b = LETTERS[seq(5,1)], c=LETTERS[c(1,4,3,2,5)])

当然,在我的实际数据中,数据帧是数字,但查找表要复杂得多,所以我不能只使用像LETTERS这样的函数来解决问题。

提前谢谢!

2 个答案:

答案 0 :(得分:4)

这是一个使用lapply()依次对每列进行处理的解决方案:

as.data.frame(lapply(example,function(col) lookup$letter[match(col,lookup$number)]));
##   a b c
## 1 A E A
## 2 B D D
## 3 C C C
## 4 D B B
## 5 E A E

或者,如果您不介意切换到矩阵,您可以实现“更多矢量化”解决方案,因为矩阵允许您只调用一次match()和索引lookup$letter整个输入:

matrix(lookup$letter[match(as.matrix(example),lookup$number)],nrow(example));
##      [,1] [,2] [,3]
## [1,] "A"  "E"  "A"
## [2,] "B"  "D"  "D"
## [3,] "C"  "C"  "C"
## [4,] "D"  "B"  "B"
## [5,] "E"  "A"  "E"

(当然,您之后可以通过as.data.frame()强制回复data.frame,但如果您需要,还必须恢复列名称,这可以通过setNames(...,names(example))来完成但如果你真的想坚持使用data.frame,我的第一个解决方案可能更可取。)

答案 1 :(得分:4)

使用dplyr

f <- function(x)setNames(lookup$letter, lookup$number)[x] 
library(dplyr)
example %>% 
  mutate_each(funs(f))
#  a b c
#1 A E A
#2 B D D
#3 C C C
#4 D B B
#5 E A E

data.table

library(data.table)
setDT(example)[, lapply(.SD, f), ]
#   a b c
#1: A E A
#2: B D D
#3: C C C
#4: D B B
#5: E A E