将R split()函数转换为C++
将R split()函数转换为C++
提问于 2021-08-20 07:52:57
考虑R中的可重现示例：
test <- c(1:12)
> test
[1] 1 2 3 4 5 6 7 8 9 10 11 12预期结果：
test.list <- split(test, gl(2, 3))
> test.list
$`1`
[1] 1 2 3 7 8 9
$`2`
[1] 4 5 6 10 11 12我正在尝试用C++编写等价的代码，以生成并返回从test.list得到的两个向量。请注意，在C++中，我正处于尴尬的新手阶段。
回答 2
Stack Overflow用户
回答已采纳
发布于 2021-08-20 12:31:04
我们可以使用@jignatius的很好的答案，并使它成为一个R可调用的函数。为了简单起见，我把它放在NumericVector上；我们这里有一大堆答案，它们显示了如何根据运行时负载在NumericVector和IntegerVector之间切换。
代码
#include <Rcpp.h>
// [[Rcpp::export]]
Rcpp::List mysplit(Rcpp::NumericVector nums, int n, int size) {
std::vector<std::vector<double>> result(n);
int i = 0;
auto beg = nums.cbegin();
auto end = nums.cend();
while (beg != nums.cend()) {
//get end iterator safely
auto next = std::distance(beg, end) >= size ? beg + size : end;
//insert into result
result[i].insert(result[i].end(), beg, next);
//advance iterator
beg = next;
i = (i + 1) % n;
}
Rcpp::List ll;
for (const auto&v : result)
ll.push_back(v);
return ll;
}
/*** R
testvec <- 1:12
mysplit(testvec, 2, 3)
*/输出
> Rcpp::sourceCpp("~/git/stackoverflow/68858728/answer.cpp")
> testvec <- 1:12
> mysplit(testvec, 2, 3)
[[1]]
[1] 1 2 3 7 8 9
[[2]]
[1] 4 5 6 10 11 12
> 原始问题中有一个小错误，因为我们不需要调用gl()；只需要两个标量。
Stack Overflow用户
发布于 2021-08-20 10:28:46
试试这个，它创建了一个矢量向量，其中包含交替分块的源元素：
#include <iostream>
#include <vector>
template<typename T>
std::vector<std::vector<T>> split(std::vector<T> nums, int n, int size)
{
std::vector<std::vector<T>> result(n);
int i = 0;
auto beg = nums.cbegin();
auto end = nums.cend();
while (beg != nums.cend()) {
//get end iterator safely
auto next = std::distance(beg, end) >= size ? beg + size : end;
//insert into result
result[i].insert(result[i].end(), beg, next);
//advance iterator
beg = next;
i = (i + 1) % n;
}
return result;
}
int main()
{
std::vector<int> vnums = { 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12 };
auto vectors = split(vnums, 2, 3);
for (const auto& v : vectors)
{
for (auto num : v) {
std::cout << num << " ";
}
std::cout << std::endl;
}
}页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接：
https://stackoverflow.com/questions/68858728
复制相关文章
相似问题
腾讯云开发者
