带有na.rm = TRUE参数的data.table和pmin

王艾伦

我正在尝试使用pmin函数和data.table(类似于post -by-row的操作和data.table中的更新)来计算跨行的最小值,但是要使用类似with=FALSE语法的列字符列表,并使用该na.rm=TRUE说法。

DT <- data.table(x = c(1,1,2,3,4,1,9), 
                 y = c(2,4,1,2,5,6,6),
                 z = c(3,5,1,7,4,5,3),
                 a = c(1,3,NA,3,5,NA,2))

> DT
   x y z  a
1: 1 2 3  1
2: 1 4 5  3
3: 2 1 1 NA
4: 3 2 7  3
5: 4 5 4  5
6: 1 6 5 NA
7: 9 6 3  2

我可以直接使用列计算跨行的最小值:

DT[,min_val := pmin(x,y,z,a,na.rm=TRUE)]

给予

> DT
   x y z  a min_val
1: 1 2 3  1       1
2: 1 4 5  3       1
3: 2 1 1 NA       1
4: 3 2 7  3       2
5: 4 5 4  5       4
6: 1 6 5 NA       1
7: 9 6 3  2       2

但是,我正在尝试对自动生成的大型列进行此操作,并且希望能够对存储在col_names变量中的任意列列表进行此操作, col_names <- c("a","y","z')

我可以做这个:

DT[, col_min := do.call(pmin,DT[,col_names,with=FALSE])]

但这给了我NA值。我不知道如何将na.rm=TRUE参数传递给do.call。我尝试过将功能定义为

DT[, col_min := do.call(function(x) pmin(x,na.rm=TRUE),DT[,col_names,with=FALSE])]

但这给我一个错误。我还尝试将参数作为列表中的附加元素传递,但我认为pmin(或do.call)在列名的DT非标准评估和参数之间感到困惑。

有任何想法吗?

阿克伦

如果我们需要得到整个数据集的每一行的最低值,使用pmin.SD串连的na.rm=TRUE作为list.SDdo.call(pmin

DT[, col_min:= do.call(pmin, c(.SD, list(na.rm=TRUE)))]
DT
#   x y z  a col_min
#1: 1 2 3  1       1
#2: 1 4 5  3       1
#3: 2 1 1 NA       1
#4: 3 2 7  3       2
#5: 4 5 4  5       4
#6: 1 6 5 NA       1
#7: 9 6 3  2       2

如果我们只想对存储在“ col_names”中的一部分列名执行此操作,请使用.SDcols

DT[, col_min:= do.call(pmin, c(.SD, list(na.rm=TRUE))), 
                .SDcols= col_names]

本文收集自互联网,转载请注明来源。

如有侵权,请联系 [email protected] 删除。

编辑于
0

我来说两句

0 条评论
登录 后参与评论

相关文章