37

R には明らかな単純な関数が欠けているようです: psum. 別の名前で存在しますか、それともパッケージのどこかにありますか?

x = c(1,3,NA,5)
y = c(2,NA,4,1)

min(x,y,na.rm=TRUE)    # ok
[1] 1
max(x,y,na.rm=TRUE)    # ok
[1] 5
sum(x,y,na.rm=TRUE)    # ok
[1] 16

pmin(x,y,na.rm=TRUE)   # ok
[1] 1 3 4 1
pmax(x,y,na.rm=TRUE)   # ok
[1] 2 3 4 5
psum(x,y,na.rm=TRUE)
[1] 3 3 4 6                             # expected result
Error: could not find function "psum"   # actual result

私はそれ+がすでに似ていることを理解していますpsumが、どうNAですか？ 

x+y                      
[1]  3 NA NA  6        # can't supply `na.rm=TRUE` to `+`

追加する場合はありpsumますか？または私は何かを逃しました。

この質問は、この質問のフォローアップです:
Using :=in data.table to sum the values of two columns in R, ignoring NAs

4

3 に答える 3

21

前の質問に対する@JoshUlrichのコメントに続いて、

psum <- function(...,na.rm=FALSE) { 
    rowSums(do.call(cbind,list(...)),na.rm=na.rm) }

編集：Sven Hohensteinから：

psum2 <- function(...,na.rm=FALSE) { 
    dat <- do.call(cbind,list(...))
    res <- rowSums(dat, na.rm=na.rm) 
    idx_na <- !rowSums(!is.na(dat))
    res[idx_na] <- NA
    res 
}

x = c(1,3,NA,5,NA)
y = c(2,NA,4,1,NA)
z = c(1,2,3,4,NA)

psum(x,y,na.rm=TRUE)
## [1] 3 3 4 6 0
psum2(x,y,na.rm=TRUE)
## [1] 3 3 4 6 NA

n = 1e7
x = sample(c(1:10,NA),n,replace=TRUE)
y = sample(c(1:10,NA),n,replace=TRUE)
z = sample(c(1:10,NA),n,replace=TRUE)

library(rbenchmark)
benchmark(psum(x,y,z,na.rm=TRUE),
          psum2(x,y,z,na.rm=TRUE),
          pmin(x,y,z,na.rm=TRUE), 
          pmax(x,y,z,na.rm=TRUE), replications=20)

##                          test replications elapsed relative 
## 4  pmax(x, y, z, na.rm = TRUE)           20  26.114    1.019 
## 3  pmin(x, y, z, na.rm = TRUE)           20  25.632    1.000 
## 2 psum2(x, y, z, na.rm = TRUE)           20 164.476    6.417
## 1  psum(x, y, z, na.rm = TRUE)           20  63.719    2.486

Sven のバージョン (間違いなく正しいバージョン) はかなり遅いですが、それが問題になるかどうかは明らかにアプリケーションによって異なります。インライン/Rcpp バージョンをハックしたい人はいますか?

これが存在しない理由については、わかりませんが、R-core がこのような追加を行うようになったことを幸運に思います*misc。

r-devel に関する Matthew のフォローアップ スレッドはこちらです (確認しているようです):
r-devel: それぞれ na.rm を使用する pmin と pmax があります。

于 2012-10-29T14:39:05.917 に答える