value matching function在R
中非常有用。但根據我的理解,它不足以支持兩個或高維度的輸入。在R中,匹配矩陣行或列的函數
例如,假設x
和y
是相同的列數的矩陣,並且我想匹配的x
行的y
行。 'R'函數調用match(x,y)
不這樣做。列表輸入發生同樣的不足。
我已經實現了我自己的版本,名爲matchMat(xMat, yMat)
(附後),但我想知道你對這個任務有什麼解決方案。
matchMat = function(xMat, uMat, dimn=1) {
ind = rep(-1, dim(xMat)[dimn])
id = 1 : dim(uMat)[dimn]
for (i in id) {
e = utilSubMat(i, uMat, dimn)
isMatch = matchVect(e, xMat, dimn)
ind[isMatch] = i
}
return(ind)
}
matchVect = function(v, xMat, dimn) {
apply(xMat, dimn, function(e) {
tf = e == v
all(tf)
})
}
unittest_matchMat = function() {
dimn = 1
uMat = matrix(c(1, 2, 2, 3, 3, 4, 4, 5), ncol=2, byrow=T)
ind = sample(dim(uMat)[1], 10, replace=T)
print(ind)
xMat = uMat[ind, ]
rst = matchMat(xMat, uMat, dimn)
print(rst)
stopifnot(all(ind == rst))
xMat2 = rbind(c(999, 999), xMat, c(888, 888))
rst2 = matchMat(xMat2, uMat, dimn)
print(rst2)
stopifnot(all(c(-1, ind, -1) == rst2))
print('pass!')
}