编写Rcpp函数来检测NumericMatrix是否有任何NA值

#include <Rcpp.h>
using namespace Rcpp;

// [[Rcpp::export]]
bool hasNA(NumericMatrix mat) {
  int nrow = mat.nrow();
  int ncol = mat.ncol();
  
  for (int i = 0; i < nrow; i++) {
    for (int j = 0; j < ncol; j++) {
      if (NumericVector::is_na(mat(i, j))) {
        return true;
      }
    }
  }
  
  return false;
}

# 安装并加载Rcpp包
install.packages("Rcpp")
library(Rcpp)

# 编译C++代码
sourceCpp(code = '
  #include <Rcpp.h>
  using namespace Rcpp;
  
  // [[Rcpp::export]]
  bool hasNA(NumericMatrix mat) {
    int nrow = mat.nrow();
    int ncol = mat.ncol();
    
    for (int i = 0; i < nrow; i++) {
      for (int j = 0; j < ncol; j++) {
        if (NumericVector::is_na(mat(i, j))) {
          return true;
        }
      }
    }
    
    return false;
  }
')

# 创建一个NumericMatrix
mat <- matrix(c(1, 2, NA, 4, 5, 6), nrow = 2)

# 调用C++函数检测NA值
hasNA(mat)