3

我在 R 代码中使用了很多"%within[]%" <- function(x,y){x>=y[1] & x<=y[2]}(意思x是在紧凑集中y),但我很确定它非常慢。你有更快的东西吗?它需要适用于>定义的所有内容。

编辑:x可以是一个向量和y一个升序的 2 个元素向量...

EDIT2:奇怪的是,没有人(据我所知)编写了一个rOperator实现快速C运算符的包,例如%w/i[]%, %w/i[[%, ...

EDIT3:我意识到我的问题太笼统了,因为假设x,y会修改任何结果,我认为我们应该关闭它,感谢您的输入。

4

4 回答 4

6
"%within[]%" <- function(x,y){x>=y[1] & x<=y[2]}

x <- 1:10
y <- c(3,5)

x %within[]% y
"%within[]2%" <- function(x,y) findInterval(x,y,rightmost.closed=TRUE)==1
x %within[]2% y

library(microbenchmark)

microbenchmark(x %within[]% y,x %within[]2% y)

Unit: microseconds
             expr   min    lq median    uq    max
1  x %within[]% y 1.849 2.465 2.6185 2.773 11.395
2 x %within[]2% y 4.928 5.544 5.8520 6.160 37.265

x <- 1:1e6
microbenchmark(x %within[]% y,x %within[]2% y)

Unit: milliseconds
             expr      min       lq   median       uq      max
1  x %within[]% y 27.81535 29.60647 31.25193 56.68517 88.16961
2 x %within[]2% y 20.75496 23.07100 24.37369 43.15691 69.62122

这可能是 Rcpp 的工作。

于 2013-02-21T13:09:44.597 回答
3

你可以通过一个简单的 Rcpp 实现来获得小的性能提升:

library(Rcpp)
library(microbenchmark)

withinR <- function(x,y) x >= y[1] & x <= y[2]
cppFunction("LogicalVector withinCpp(const NumericVector& x, const NumericVector& y) {
  double min = y[0], max = y[1];

  int n = x.size();
  LogicalVector out(n);

  for(int i = 0; i < n; ++i) {
    double val = x[i];
    if (NumericVector::is_na(val)) {
      out[i] = NA_LOGICAL;
    } else {
      out[i] = val >= min & val <= max;
    }

  }
  return out;
}")

x <- sample(100, 1e5, rep = T)

stopifnot(all.equal(withinR(x, c(25, 50)), withinCpp(x, c(25, 50))))

microbenchmark(
  withinR(x, c(25, 50)),
  withinCpp(x, c(25, 50))
)

在我的计算机上,C++ 版本的速度大约快 4 倍。如果您想使用更多 Rcpp 技巧,您可能还可以进一步调整它,但这似乎已经相当快了。即使是 R 版本也需要非常频繁地调用才能成为瓶颈。

# Unit: microseconds
#                      expr  min   lq median   uq  max
# 1 withinCpp(x, c(25, 50))  635  659    678 1012 27385
# 2   withinR(x, c(25, 50)) 1969 2031   2573 2954 4082
于 2013-02-21T14:50:26.073 回答
1

好吧,我不知道这是否可以被认为是缓慢的,但这里有一些基准:

R> within <- function(x,y){return(x>=y[1] & x<=y[2])}
R> microbenchmark(within(2,c(1,5)))
Unit: microseconds
               expr   min     lq median    uq    max neval
 within(2, c(1, 5)) 2.667 2.8305 2.9045 2.969 15.818   100

R> within2 <- function(x,y) x>=y[1] & x<=y[2]
R> microbenchmark(within2(2,c(1,5)))
Unit: microseconds
                expr   min     lq median    uq    max neval
 within2(2, c(1, 5)) 2.266 2.3205  2.398 2.483 12.472   100

R> microbenchmark(2>=1 & 2<=5)
Unit: nanoseconds
            expr min    lq median  uq  max neval
 2 >= 1 & 2 <= 5 781 821.5    850 911 5701   100

因此,按照 Konrad Rudolph 的建议,省略 似乎可以return加快速度。但是不写函数要快得多。

于 2013-02-21T13:02:25.523 回答
1

x如果包含许多值,基于树的结构会提供更好的性能。如果您可以将要求限制为数值,则有 2 个选项

整数的区间树的实现可以在 Bioconductor 包IRanges中找到。

默认情况下,RSQLite 正在编译启用 rtrees 的嵌入式 SQLite。这可以与任何数值一起使用。

于 2013-02-21T13:06:33.280 回答