基准比较

Question

我有数据框，例如：

   productid   ordernum   
   p1          10
   p2          20
   p3          30 
   p4          5
   p5          20
   p6          8

我想添加另一个名为groupid的列，它按顺序将产品分组在一起，一旦sum（ordernum）达到30，就分配一个新的组ID，例如结果应该是

 productid   ordernum  groupid   
   p1          10        1
   p2          20        1 
   p3          30        2
   p4          5         3
   p5          20        3 
   p6          8         3

循环非常容易，如何在不循环的情况下实现这一目标？

Answer 1

使用c++编写的for简短Rcpp循环怎么样？这个小函数采用numeric向量，即您的ordernum列和threshold参数（您想要从中开始新ID的累积和），并返回长度相等的ID向量到输入向量。应该相对较快地运行，因为它是for中的c++循环。下面的代码片段将为您安装Rcpp，如果您尚未安装它并将编译该函数准备使用。只需复制并粘贴到R ...

if( !require(Rcpp) ) install.packages("Rcpp"); require(Rcpp)
Rcpp::cppFunction( ' NumericVector grpid( NumericVector x , int threshold ){
  int n = x.size();
  NumericVector out(n);
  int tot = 0;
  int id = 1;
  for( int i = 0; i < n; ++i){
    tot += x[i];
    out[i] = id;
    if( tot >= threshold ){
      id += 1;
      tot = 0;
    }
  }
  return out;
}')

然后使用该函数就像使用任何其他R函数一样使用它，提供相关的参数：

df$groupid <- grpid( df$ordernum , 30 )
#  productid ordernum groupid
#1        p1       10       1
#2        p2       20       1
#3        p3       30       2
#4        p4        5       3
#5        p5       20       3
#6        p6        8       3

基准比较

OP请求我将Rcpp循环与基础R for循环进行基准测试。这是代码和结果。在100,000个产品ID的载体上速度提高约400倍：

set.seed(1)
x <- sample(30,1e5,repl=T)
for.loop <- quote({
    tot <- 0 
    id <- 1
    out <- numeric(length(x))
    for( i in 1:length(x) ){
        tot <- tot + x[i]
        out[i] <- id
        if( tot >= 30 ){
            tot <- 0
            id <- id + 1
        }
    }
})

rcpp.loop <- quote( out <- grpid(x,30))

require( microbenchmark )
print( bm , unit = "relative" , digits = 2 , "median" )
Unit: relative
            expr min  lq median  uq max neval
 eval(rcpp.loop)   1   1      1   1   1    50
  eval(for.loop) 533 462    442 428 325    50

如何在不循环的情况下添加组ID？

1 个答案:

基准比较