爲矩陣命名的索引

Question

我想索引帶有名稱的矩陣。通常的方法給出了錯誤：

NumericMatrix mytest(NumericVector v) { 
    NumericMatrix ans(v.length(), v.length()); 
    rownames(ans) = v; 
    float y = ans("1",0); 
    NumericVector x = ans.row("1"); 
    return (ans); 
} 

我看了看在Matrix.h和矩陣單元測試中RCPP，但沒有找到類似的例子。此郵件列表question也未提供執行此操作的方法。

我可以寫我自己的代碼來索引矩陣，也許使用R的內部C interface？

Answer 1

這是絕不是一個強大的解決方案，但希望一個跳點，哪裏operator()超載處理，因爲行&列索引傳遞的int和string不同的組合：

#include <Rcpp.h> 
// [[Rcpp::plugins(cpp11)]] 

class MyMat : public Rcpp::NumericMatrix { 
    public: 
    MyMat(const Rcpp::NumericMatrix& data_, 
      const std::vector<std::string>& rnames_, 
      const std::vector<std::string>& cnames_) 
    : data(data_), 
     rnames(rnames_), 
     cnames(cnames_) {} 

    double operator()(const std::string& i, const std::string& j) { 
     typedef std::vector<std::string>::const_iterator cit; 
     cit it_i = std::find(rnames.begin(), rnames.end(), i); 
     cit it_j = std::find(cnames.begin(), cnames.end(), j); 

     int idx_i, idx_j; 

     if (it_i != rnames.end()) { 
     idx_i = it_i - rnames.begin(); 
     } else { 
     idx_i = rnames.size(); 
     } 

     if (it_j != cnames.end()) { 
     idx_j = it_j - cnames.begin(); 
     } else { 
     idx_j = cnames.size(); 
     } 

     return data(idx_i, idx_j); 
    } 

    double operator()(const std::string& i, const size_t j) { 
     typedef std::vector<std::string>::const_iterator cit; 
     cit it_i = std::find(rnames.begin(), rnames.end(), i); 

     int idx_i, idx_j; 

     if (it_i != rnames.end()) { 
     idx_i = it_i - rnames.begin(); 
     } else { 
     idx_i = rnames.size(); 
     } 

     if (j <= cnames.size()) { 
     idx_j = j; 
     } else { 
     idx_j = cnames.size(); 
     } 

     return data(idx_i, idx_j); 
    } 

    double operator()(const size_t i, const std::string& j) { 
     typedef std::vector<std::string>::const_iterator cit; 
     cit it_j = std::find(cnames.begin(), cnames.end(), j); 

     int idx_i, idx_j; 

     if (i <= rnames.size()) { 
     idx_i = i; 
     } else { 
     idx_i = rnames.size(); 
     } 

     if (it_j != cnames.end()) { 
     idx_j = it_j - cnames.begin(); 
     } else { 
     idx_j = cnames.size(); 
     } 

     return data(idx_i, idx_j); 
    } 

    double operator()(const int& i, const int& j) { 
     return data(i, j); 
    } 

    private: 
    Rcpp::NumericMatrix data; 
    std::vector<std::string> rnames; 
    std::vector<std::string> cnames; 
}; 

// [[Rcpp::export]] 
void test_MyMat(Rcpp::NumericMatrix m) 
{ 
    std::vector<std::string> rnames = { "a", "b", "c" }; 
    std::vector<std::string> cnames = { "A", "B", "C" }; 
    MyMat mmObj(m,rnames,cnames); 

    Rcpp::Rcout << "(Row 1, Column 1)" << 
    std::endl; 
    Rcpp::Rcout << "(b,B) = " << mmObj("b","B") << 
    std::endl << "(b,1) = " << mmObj("b",1) << 
    std::endl << "(1,B) = " << mmObj(1,"B") << 
    std::endl << "(1,1) = " << mmObj(1,1) << 
    std::endl; 
} 

/*** R 
x <- matrix(1:9,nrow=3) 

test_MyMat(x) 
#(Row 1, Column 1) 
#(b,B) = 5 
#(b,1) = 5 
#(1,B) = 5 
#(1,1) = 5 

x[2,2] 
#[1] 5 
*/ 

Answer 2

我們只支持數字（行）索引。

您可以添加一個「names」屬性並查找其中的索引，和/或添加您自己的訪問器方法。