将.SD与重命名的变量组合在一起会导致.SD列名称混乱

时间:2020-06-09 11:19:08

标签: r data.table

在我的代码中,我想以编程方式选择一些变量,然后以硬编码方式选择其他变量并重命名。我知道我可以使用setnames()分两步实现,但我很好奇如何一步完成。

我认为通过.SDcols与我非常接近。但是,当我尝试将.SD与重命名的列组合时,“ .SDcols列”的前缀为“ .SD”。如何避免使用前缀?

library(data.table)
dt <- as.data.table(mtcars)[1:5]
dt
#>     mpg cyl disp  hp drat    wt  qsec vs am gear carb
#> 1: 21.0   6  160 110 3.90 2.620 16.46  0  1    4    4
#> 2: 21.0   6  160 110 3.90 2.875 17.02  0  1    4    4
#> 3: 22.8   4  108  93 3.85 2.320 18.61  1  1    4    1
#> 4: 21.4   6  258 110 3.08 3.215 19.44  1  0    3    1
#> 5: 18.7   8  360 175 3.15 3.440 17.02  0  0    3    2

my_vars <- c("cyl", "vs")
# with .SDcol
dt[, .(.SD, z = gear), .SDcol = my_vars]
#>    .SD.cyl .SD.vs z    # Note the prefix that had been added to the .SDcols
#> 1:       6      0 4
#> 2:       6      0 4
#> 3:       4      1 4
#> 4:       6      1 3
#> 5:       8      0 3

# with named vector
all_vars <- c(my_vars, z = "gear")
dt[, ..all_vars]
#>    cyl vs gear
#> 1:   6  0    4
#> 2:   6  0    4
#> 3:   4  1    4
#> 4:   6  1    3
#> 5:   8  0    3

1 个答案:

答案 0 :(得分:3)

我认为这是因为您将.SD包裹在list.())中。 list(.SD)生成一个包含list .SD data.table ,而不是.SD 。然后,这使命名混乱了。

检查str中包裹的.SD中的list

# dt[, str(.(.SD)), .SDcol = my_vars]
# List of 1
# $ :Classes ‘data.table’ and 'data.frame': 5 obs. of  2 variables:
#   ..$ cyl: num [1:5] 6 6 4 6 8
#   ..$ vs : num [1:5] 0 0 1 1 0

相应的输出具有.SD.前缀:

dt[ , .(.SD), .SDcol = my_vars]
#    .SD.cyl .SD.vs
# 1:       6      0
# 2:       6      0
# 3:       4      1
# 4:       6      1
# 5:       8      0

仅检查str中的.SD

# dt[, str(.SD), .SDcol = my_vars]
# Classes ‘data.table’ and 'data.frame':    5 obs. of  2 variables:
#   $ cyl: num  6 6 4 6 8
#   $ vs : num  0 0 1 1 0

鉴于j的基本属性-“只要j返回一个列表,列表中的每个元素都将成为结果data.table中的一列” –以及{{1 }}已经是.SD(选中list),我们可以使用dt[ , is.list(.SD)]c与其他列表元素结合起来,例如您重命名的列包含在.SD中:

list