主页 / coding / 问题 / 79024828
Accepted
noNameTed
Asked: 2024-09-26 05:31:23 +0800 CST

如果向量的元素至少位于一个区间内，是否有一种快速/巧妙的方法来返回逻辑向量？

  • 772

假设您有一个向量数字向量和一个包含列和的x数据框。有没有一种巧妙的方法可以返回一个长度等于的逻辑向量，指示 x 是否至少位于由或定义的一个区间内？dfstartstopxstartstop

我处理的实际案例是length(x)>> nrow(df)。最简单的方法是使用 for 循环，但我希望有更优雅、运行速度更快的方法。

 x <- 1:10
df <- data.frame(start = c(0, 4.5, 6), stop = c(1, 5.5, 8.5))
z <- rep(FALSE, length(x))

for(i in 1:nrow(df)){
  z <- z | (df$start[i] <= x & x <= df$stop[i])
}

x[z] # 1 5 6 7 8

3 个回答

  1. Best Answer

    也许你可以使用outer下面这样的

    > with(df, x[rowMeans(outer(x, start, `>=`) & outer(x, stop, `<=`)) > 0])
[1] 1 5 6 7 8

    或者，你可以%inrange%使用data.table

    > library(data.table)

> x[x %inrange% df]
[1] 1 5 6 7 8
    • 6

  2. 这可能太聪明了，但确实解决了你的问题。（@ThomasIsCoding 的非常相似，但稍微紧凑一些......）

    x_gt_start <- outer(x,df$start, ">=")
x_lt_stop <- outer(x,df$stop, "<=")
between <- rowSums(x_gt_start & x_lt_stop) > 0 ## or: apply(..., 1, any)
x[between]

    更快的方法是使用Rcpp循环和短路编写代码（即，只要满足给定的条件就返回 TRUE x，而不是进行所有的比较）

    • 6

  3. 编辑：正如用户 ThomasIsCoding 指出的那样，它是data.table::inrange()矢量化的：

    x = seq(10)
df = data.frame(start = c(0, 4.5, 6), stop = c(1, 5.5, 8.5))
x[data.table::inrange(x, df$start, df$stop)]
#> [1] 1 5 6 7 8
    • 5

相关问题