Таблица данных - выберите значение столбца по имени из другого столбца

Question

У меня есть таблица данных с числом столбцов, содержащих значения. У меня есть другой столбец, который определяет, какой из этих столбцов имеет значение, которое мне нужно выбрать. У меня возникли проблемы с поиском способа сделать это.

Вот простой пример.

> d <- data.table(
    value.1 = c("one", "uno", "1"), 
    value.2 = c("two", "dos", "2"), 
    name.of.col = c("value.1","value.2","value.1")) 

> d 
    value.1 value.2 name.of.col 
1:  one  two  value.1 
2:  uno  dos  value.2 
3:  1  2  value.1 

Я хотел бы добавить столбец «value.of.col», который содержит значение столбца, указанного «name.of.col».

> d 
    value.1 value.2 name.of.col value.of.col 
1:  one  two  value.1 one 
2:  uno  dos  value.2 dos 
3:  1  2  value.1 1 

Answer 1

Другой вариант:

d[,value.of.col:=diag(as.matrix(d[,d[,name.of.col],with=F]))] 
> d 
    value.1 value.2 name.of.col value.of.col 
1:  one  two  value.1   one 
2:  uno  dos  value.2   dos 
3:  1  2  value.1   1 

EDIT добавить более быстрое решение:

d[,value.of.col:= 
     melt(d,id.vars='name.of.col')[name.of.col==variable,value]] 

Answer 2

Вы можете использовать матрицу индексации тянуть значения из первого и второго столбцов:

mx.idx <- d[, cbind(1:nrow(d), match(name.of.col, names(d)))] 
d[, 
    value.of.col:= 
    as.matrix(d[, 1:2, with=F])[mx.idx] 
] 
d 
# value.1 value.2 name.of.col value.of.col 
# 1:  one  two  value.1   one 
# 2:  uno  dos  value.2   dos 
# 3:  1  2  value.1   1 

Answer 3

Следующее должно быть эффективным с точки зрения памяти и немного легче читать/следовать.

for (i in unique(d[["name.of.col"]])) 
    d[ name.of.col==i, value.of.col:=get(i) ] 

d 
    value.1 value.2 name.of.col value.of.col 
1:  one  two  value.1   one 
2:  uno  dos  value.2   dos 
3:  1  2  value.1   1