r - 如何使用 RcppArmadillo 将距离矩阵的对角线强制为零？

Question

我有以下 Rcpp/RcppArmadillo 函数，它计算矩阵中的相关距离

#include <RcppArmadillo.h>
using namespace Rcpp;
// [[Rcpp::export]]
arma::mat cordiss(NumericMatrix X){  
   int nX = X.nrow(), kX = X.ncol();
   arma::mat XX(X.begin(), nX, kX, false); 
   arma::mat output = (1 - arma::cor(XX.t(), XX.t()))/2;   
   return output.t();   
}

使用 Rcpp::sourceCpp 采购后，我可以执行以下操作:

d <- cordiss(matrix(rnorm(1000),ncol=10))

但是，我在对角线上得到非零值(因为我猜机器精度？)，而我确实需要将这些数字设置为零以进一步处理距离矩阵。这是矩阵类型的问题吗？我该如何解决这个问题，而不必事后将对角线“手动”设置为零(在 Armadillo 或 R 中)，例如:

diag(d) <- 0 

事实上，我想使用一个类似的函数来计算两个不同矩阵之间的距离，我无法预测矩阵的列在哪里相等
谢谢你的帮助!

Answer 1

不可能保证为零，因为任何使用非整数的算法实现都使用 approximations 生效。这样的数字。

最好的办法是手动将对角线设置为零，您可以通过 .diag() 在 Armadillo 中做到这一点。 .例如，output.diag().zeros()