0

我想M在 Rcpp 中将矩阵四舍五入到任意精度。在 R 中这样做特别容易:

M <- matrix(rnorm(4), 2, 2)
M
           [,1]      [,2]
[1,] 0.04463484 0.1455878
[2,] 1.77416096 1.0787835

round(M,2)
     [,1] [,2]
[1,] 0.04 0.15
[2,] 1.77 1.08

这在 Rcpp / C++ 中被证明是有点挑战的。

有一个round()函数,但是不幸的是它只能四舍五入到最接近的整数。出于输出目的,例如"%.2f"格式可用于四舍五入到两位小数。如果要在进一步计算中使用四舍五入的数字,则可以通过玩弄来将单个元素四舍五入到任意精度floorfroundf以及ceilingf使用手动调整的不同缩放因子的函数,请参阅此处的讨论和建议的解决方案。

因此,一种可能的前进方式是将上述方法应用于矩阵的每个元素(或更有效地应用于每一列)M。这似乎不必要地复杂,我想知道你们中的一个人是否有一个更有效/优雅的解决方案,用于在 Rcpp 中将矩阵舍入为任意精度。

4

2 回答 2

4

F. Privé 有一个技术上正确的答案。但是,就像他之前的 OP 一样,它错过了 Rcpp Sugar 功能已经完全一样:

R> Rcpp::cppFunction("NumericVector mr(NumericVector x,int d) {return round(x,d);}")
R> set.seed(42)    
R> x <- runif(5)     
R> x    
[1] 0.914806 0.937075 0.286140 0.830448 0.641746     
R> mr(x, 2)    
[1] 0.91 0.94 0.29 0.83 0.64     
R> mr(x, 0)          
[1] 1 1 0 1 1          
R> mr(x, 7)        
[1] 0.914806 0.937075 0.286139 0.830448 0.641745   
R>  

混淆(如果有的话)是认为参数的默认值是位数的唯一允许值。当然,事实并非如此。digits

于 2020-01-26T14:55:48.580 回答
2

您可以使用例如自己实现它

#include <Rcpp.h>
using namespace Rcpp;

// [[Rcpp::export]]
NumericVector myround(const NumericVector& A, int digits = 0) {
  NumericVector B = clone(A);
  std::size_t K = A.size();
  for (std::size_t k = 0; k < K; k++) {
    B[k] = ::Rf_fround(A[k], digits);
  }
  return B;
}

在 R 中:

> (x <- runif(10))
 [1] 0.5050331 0.8921151 0.4566404 0.5828360 0.6931808 0.9296267 0.3091896 0.4077148 0.9563310
[10] 0.6905403
> myround(x)
 [1] 1 1 0 1 1 1 0 0 1 1
> myround(x, 2)
 [1] 0.51 0.89 0.46 0.58 0.69 0.93 0.31 0.41 0.96 0.69
> (M <- matrix(rnorm(4), 2, 2))
           [,1]     [,2]
[1,] -1.0852162 1.793925
[2,] -0.1912413 1.170089
> myround(M, 2)
      [,1] [,2]
[1,] -1.09 1.79
[2,] -0.19 1.17
于 2020-01-26T08:44:43.843 回答