Rcpp中M.nrow()返回IntegerVector结果异常为0的技术问询
问题原因与解决办法
这个问题其实是Rcpp中类型隐式转换的坑导致的,我来给你拆解清楚:
核心原因
你的函数声明是返回IntegerVector类型,但实际返回的是一个int类型的变量x(也就是矩阵的行数6)。Rcpp在处理这种不匹配的返回时,会把这个int值当成IntegerVector的初始化长度来创建新向量——而用单个整数初始化IntegerVector时,默认会生成一个对应长度、所有元素为0的空向量。你这里x是6,所以就得到了长度为6的全0向量,也就是你看到的[1] 0 0 0 0 0 0。
两种解决办法
1. 直接修改返回类型为int
如果你的需求只是返回一个单个的行数,直接把函数返回类型改成int即可,Rcpp会自动把这个int值转换成R中长度为1的整数向量:
set.seed(1100) M = matrix(sample(18), nrow = 6) Rcpp::cppFunction('int tccp5(IntegerMatrix M) { int x = M.nrow(); return x;}') tccp5(M) # [1] 6
2. 手动将int包装为IntegerVector
如果一定要返回IntegerVector类型,需要用IntegerVector::create()方法把单个int值包装成向量:
Rcpp::cppFunction('IntegerVector tccp5(IntegerMatrix M) { int x = M.nrow(); return IntegerVector::create(x);}') tccp5(M) # [1] 6
额外提醒
Rcpp的容器类(比如IntegerVector、NumericVector)的构造逻辑要注意:当传入单个整数时,是指定容器的长度;如果要创建包含指定元素的向量,必须用::create()方法或者手动赋值。
内容的提问来源于stack exchange,提问作者AKshayKulkarni
相关产品推荐
相关产品推荐

