R data.table为一组行更新类型列的列

时间:2017-05-25 21:36:05

标签: r data.table

假设我们有以下data.table:

> a <- data.table(id=(1:3), val=vector("list",length(3)))
> a$val <- list(c("a","b"),c("c","d","e"),c("f","e"))
> a
   id   val
1:  1   a,b
2:  2 c,d,e
3:  3   f,e

我们现在想要在'id'为2和3的行中从'val'列中删除元素“e”。 我能够通过以下方式实现这一目标:

> for (x in c(2,3)) { 
  a[id==x, val := list(list(val[[1]][ val[[1]] != 'e']))]
  }
> a
   id val
1:  1 a,b
2:  2 c,d
3:  3   f

如何用

之类的东西来完成
a[id %in% c(2,3), ..] ?

1 个答案:

答案 0 :(得分:0)

function_remove<-function(x,y){
  if (y %in% c(2,3)){
    x<-x[[1]]
    list(x[x!="e"])
  } else{
    x
  }
}

a[,list(function_remove(val,id)),by=1:nrow(a)]
   nrow  V1
1:    1 a,b
2:    2 c,d
3:    3   f

或者你可以这样做:

function_remove_1<-function(x){
  x[x!="e"]
}
a[id %in% c(2,3),]$val<-lapply(a[id %in% c(2,3),val],function_remove_1)

最后(如下所述)我们也可以通过参考更新:

a[id %in% c(1,2), val := lapply(val, function(x) x[x!="e"])]