R中向量的子集（）

Question

我基于subset()编写了以下函数，我觉得很方便：

ss <- function (x, subset, ...) 
{
    r <- eval(substitute(subset), data.frame(.=x), parent.frame())
    if (!is.logical(r)) 
        stop("'subset' must be logical")
    x[r & !is.na(r)]
}

所以，我可以写：

ss(myDataFrame$MyVariableName, 500 < . & . < 1500)

而不是

myDataFrame$MyVariableName[ 500 < myDataFrame$MyVariableName 
                                & myDataFrame$MyVariableName < 1500]

这似乎是其他人可能已经开发出解决方案的东西 - 包括我可能错过的核心 R 中的一些东西。已经有什么了吗？

Answer 1

我意识到 Ken 提供的解决方案比仅在范围内选择项目更通用（因为它应该适用于任何逻辑表达式）但这确实提醒我 Greg Snow 在他的教学演示包中有比较中缀运算符：

library(TeachingDemos)
x0 <- rnorm(100)
x0[ 0 %<% x0 %<% 1.5 ]

Answer 2

感谢肯分享。

你可以使用：

x <- myDataFrame$MyVariableName; x[x > 100 & x < 180] 

您的代码可能需要较少的输入，但如果您共享代码，则代码对其他人的概括性较差。我自己也有一些类似的节省时间的功能，但要谨慎使用它们，因为它们可能会减慢您的代码（额外步骤），并且在您与其他人共享文件时还要求您还包含该功能的代码。

比较写作长度。几乎相同的长度：

ss(mtcars$hp, 100 < . & . < 180)
x <- mtcars$hp; x[x > 100 & x < 180] 

比较 1000 次复制的时间。

library(rbenchmark)
benchmark(
       tyler = x[x > 100 & x < 180],
       ken = ss(mtcars$hp, 100 <. & . < 180),
 replications=1000)

   test replications elapsed relative user.self sys.self user.child sys.child
2   ken         1000    0.56 18.66667      0.36     0.03         NA        NA
1 tyler         1000    0.03  1.00000      0.03     0.00         NA        NA

所以我想这取决于您是否需要速度和/或可共享性与便利性。如果它只适合您在一个小型数据集上使用，我会说它很有价值。

编辑：新基准测试

> benchmark(
+     tyler = {x <- mtcars$hp; x[x > 100 & x < 180]}, 
+     ken = ss(mtcars$hp, 100 <. & . < 180), 
+     ken2 = ss2(mtcars$hp, 100 <. & . < 180),
+     joran = with(mtcars,hp[hp>100 & hp< 180 ]), 
+  replications=10000)

   test replications elapsed  relative user.self sys.self user.child sys.child
4 joran        10000    0.83  2.677419      0.69     0.00         NA        NA
2   ken        10000    3.79 12.225806      3.45     0.02         NA        NA
3  ken2        10000    0.67  2.161290      0.35     0.00         NA        NA
1 tyler        10000    0.31  1.000000      0.20     0.00         NA        NA