整数列的总和是两倍

时间:2017-08-30 12:49:09

标签: r data.table rowsum

我不明白为什么以及如何防止两个整数列的总和为numeric类,键入double。有什么想法吗?

这是一个小工作示例

library(data.table)

set.seed(123)

A <- rnorm(20, 100, 5)
B <- rnorm(20, 50, 20)

NA.A <- which(A %in% sample(A, 5))
NA.B <- which(B %in% sample(B, 10))

zero.A<- which(A %in% sample(A, 3))
zero.B <- which(B %in% sample(B, 8))

A[NA.A] <- NA
A[zero.A] <- 0
B[NA.B] <- NA
B[zero.B] <- 0

mydt <- data.table(A = as.integer(A), B = as.integer(B))
sapply(mydt, class)
#    A         B 
# "integer" "integer"

mydt[, C := rowSums(.SD, na.rm=T), .SDcols = c("A","B")]
sapply(mydt, class)
#    A         B         C 
# "integer" "integer" "numeric" 

sapply(mydt, typeof)
#    A         B         C 
# "integer" "integer"  "double" 

1 个答案:

答案 0 :(得分:4)

由于rowSums()将始终返回double类型,您可以将Reduce()+运算符结合使用,以将新列作为integer返回。

mydt[,C:=Reduce(`+`, lapply(.SD, function(x) ifelse(!is.na(x),x,0))),.SDcols = c("A","B")]