使用data.table计算新列

时间:2015-10-05 08:20:30

标签: r data.table

我有以下数据

set.seed(5)
dt <- data.table(ID=letters, x = rnorm(26), y = rnorm(26), z = c(rep(15, 13), rep(20,13)))

返回,

    ID           x           y  z
 1:  a -0.84085548  1.41858907 15
 2:  b  1.38435934  1.49877383 15
 3:  c -1.25549186 -0.65708209 15
 4:  d  0.07014277 -0.85279544 15
 5:  e  1.71144087  0.31591504 15
 6:  f -0.60290798  1.10969417 15
 7:  g -0.47216639  2.21546057 15
 8:  h -0.63537131  1.21710364 15
 9:  i -0.28577363  1.47922179 15
10:  j  0.13810822  0.95157383 15
11:  k  1.22763034 -1.00953265 15
12:  l -0.80177945 -2.00047274 15
13:  m -1.08039260 -1.76218587 15
14:  n -0.15753436 -0.14260813 20
15:  o -1.07176004  1.55006037 20
16:  p -0.13898614 -0.80242318 20
17:  q -0.59731309 -0.07457892 20
18:  r -2.18396676  1.89566795 20
19:  s  0.24081726 -0.45656894 20
20:  t -0.25935541  0.56222336 20
21:  u  0.90051195 -0.88700851 20
22:  v  0.94186939 -0.46024458 20
23:  w  1.46796190 -0.72432849 20
24:  x  0.70676109 -0.06921116 20
25:  y  0.81900893  1.46324856 20
26:  z -0.29348185  0.18772610 20

我正在尝试更新列xy,同时将z分开,同时保留列ID。也就是说,最终输出应包含IDx/zy/z

我尝试了以下代码,但它返回错误

dt[,c('x', 'y'):=lapply(.SD, function(x) x/z), .SDcols = names(dt)]

仅供参考,实际数据中有超过100列必须按列z进行划分。

你能给我一些建议吗?

3 个答案:

答案 0 :(得分:5)

更新:问题#495现已解决this recent commit,我们现在可以做到这一点:

require(data.table) # v1.9.7+
nam <- setdiff(names(dt), c("ID", "z"))    
dt[, (nam) := lapply(.SD, `/`, z), .SDcols = nam]
nam <- setdiff(names(dt), c("ID", "z"))    
dt[, (nam) := lapply(.SD, `/`, dt[,z]), .SDcols = nam]

请注意,由于此dt[, z]错误#495,我在lapply内使用data.table
如果您使用.SDcols,则无法在函数调用中使用其他列。

作为解决方法,在#495完成之前,您可以使用mget(),如下所示:

dt[, (nam) := lapply(mget(nam), `/`, z)]

答案 1 :(得分:3)

怎么样

dt[, `:=`(x=x/z, y=y/z, z=NULL)]

编辑:在原始问题中添加了数据表中有两列以上的内容后,我将使用Floo0的答案

答案 2 :(得分:1)

这不起作用吗?

dt$x <- dt$x / dt$z
dt$y <- dt$y / dt$z

dt <- dt[ , seq(1, 3)]

编辑:如果你有很多列,你必须用z除,你可以试试这个:

dt[, seq(2, 101)] <- sapply(dt[, seq(2, 101)], '/', dt$z)
dt <- dt[, seq(1, 101)] #replace with boundaries of your choosing