如何利用多键正确筛选data.table中指定条件的行?
解决data.table双键筛选的问题
嗨,我来帮你搞定这个筛选问题!你之前用test[list(c(1,2),c("A","B"))]得到错误结果,是因为这种写法是按位置配对筛选的——它只会找x=1且y="A"、x=2且y="B"这两组配对的行,而不是你想要的“x为1/2 且 y为A/B”的所有组合。
下面给你两种正确的解决方法:
方法1:利用键+交叉连接(CJ)
既然已经用setkey(test,x,y)设置了双键,我们可以用CJ()函数生成所有需要的键组合,再传给data.table进行匹配:
test[CJ(c(1,2), c("A","B"))]
CJ()会生成(1,A)、(1,B)、(2,A)、(2,B)这四组所有可能的键组合,data.table会基于已设置的键快速匹配出对应的行,适合大数据量的场景,性能更优。
方法2:直接用逻辑条件筛选
如果你觉得键的配对逻辑有点绕,也可以直接写清晰的逻辑表达式筛选,不用依赖键:
test[x %in% c(1,2) & y %in% c("A","B")]
这种写法更直观,新手也能一眼看懂,小数据量下使用完全没问题,结果和上面的方法一致。
运行结果示例
不管用哪种方法,最终都会得到符合条件的行:
x y 1: 1 A 2: 1 A 3: 1 B 4: 2 A 5: 2 B
内容的提问来源于stack exchange,提问作者LimingFang
相关产品推荐
相关产品推荐

