c – 子集一个向量并对其进行排序

发布时间：2020-12-16 03:24:48 所属栏目：百科来源：网络整理

导读：我正在使用Rcpp软件包,使用C语言简化R包的部件.我是一个C新手(但是很想学习！).我已经实现了一些简单的cpp程序使用优秀的Rcpp – 事实上,这个包激励我学习C … 无论如何,我已经陷入了一个简单的问题,如果我可以修复会帮助很多.我有一个NumericVector我想要子

我正在使用Rcpp软件包,使用C语言简化R包的部件.我是一个C新手(但是很想学习！).我已经实现了一些简单的cpp程序使用优秀的Rcpp – 事实上,这个包激励我学习C …

无论如何,我已经陷入了一个简单的问题,如果我可以修复会帮助很多.我有一个NumericVector我想要子集,然后排序.下面的代码排序整个向量(并且还将处理NAs,这是我需要的).

我的问题是,我想提取这个向量的一部分,排序并将其用于其他处理 – 我该怎么做？例如,对于长度为10的向量,我如何提取和排序元素5:10？

#include <Rcpp.h>
using namespace Rcpp;

// [[Rcpp::export]]
RcppExport SEXP rollP(SEXP x) {
  NumericVector A(x); // the data  
  A = sort_unique(A);  
  return A;
}

我叫R：

sourceCpp( "rollP.cpp")
rollP(10:1)
# [1]  1  2  3  4  5  6  7  8  9 10

解决方法

以下是3种变体：

include <Rcpp.h>
using namespace Rcpp;

// [[Rcpp::export]]
NumericVector rollP(NumericVector A,int start,int end) {
  NumericVector B(end-start+1) ;
  std::copy( A.begin() + start-1,A.begin() + end,B.begin() ) ;
  return B.sort() ;
}

// [[Rcpp::export]]
NumericVector rollP2(NumericVector A,int end) {
  NumericVector B( A.begin() + start-1,A.begin() + end ) ;
  return B.sort() ;
}

// [[Rcpp::export]]
NumericVector rollP3(NumericVector A,int end) {
  NumericVector B = A[seq(start-1,end-1)] ;
  return B.sort() ;
}

开始和结束是指基于1的指数,就好像你从R通过A [开始：结束].

（编辑：李大同）

【声明】本站内容均来自网络，其相关言论仅代表作者个人观点，不代表本站立场。若无意侵犯到您的权利，请及时与联系站长删除相关内容!