R data.table，访问赋值函数内的矩阵

Question

我有以下数据表

structure(list(xi = c(1, 1, 1, 2, 2, 2, 3, 3, 3), yi = c(1, 2, 
3, 1, 2, 3, 1, 2, 3), flag = c(0, 0, 0, 0, 0, 0, 0, 0, 0)), .Names = c("xi", 
"yi", "flag"), row.names = c(NA, -9L), class = c("data.table", 
"data.frame"), .internal.selfref = <pointer: 0x11a1a78>)

我还有一个 3x3 矩阵，如下所示。

structure(c(1, 1, 0.4, 1, 0, 0, 1, 0, 0.2), .Dim = c(3L, 3L))

我想为 data.table flag 分配第三列，这样如果矩阵中的元素由 xi 行和 yi 列小于 1，则 flag = 1 else 0。为此我写了一个函数，

func <- function (x, y, m) {
if (m[x, y] < 1) {
    return(1)
}
else {
    return(0)
}
}

但是，如果我尝试

y[,flag := func(xi,yi,m)]

我的标志值始终为 0。有人可以指出我在这里做错了什么吗？ 提前致谢。

Answer 1

您不需要自定义函数...

dt[ , flag := as.integer( m[cbind(xi,yi)] < 1 ) ]

您确实需要注意以正确的方式索引矩阵(使用 cbind(...) 而不是 [,] 形式的索引)。