Início / coding / Perguntas / 79024828
Accepted
noNameTed
Asked: 2024-09-26 05:31:23 +0800 CST

Existe uma maneira rápida/inteligente de retornar um vetor lógico se os elementos de um vetor estiverem em pelo menos um intervalo?

  • 772

Suponha que você tenha um vetor numérico xe um data frame dfcom colunas starte stop. Existe uma maneira inteligente de retornar um vetor lógico com comprimento igual a xindicando se x está em pelo menos um intervalo definido por startou stop?

O caso real com o qual estou trabalhando tem length(x)>> nrow(df). A maneira ingênua de fazer isso seria usar um loop for, mas eu esperava algo mais elegante e que rodasse rápido.

 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 respostas

  1. Best Answer

    Talvez você possa usar outercomo abaixo

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

    ou você pode usar %inrange%dedata.table

    > library(data.table)

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

  2. Talvez isso seja muito inteligente, mas resolve seu problema. (@ThomasIsCoding's é muito semelhante, mas um pouco mais compacto...)

    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]

    Ainda mais rápido seria usar Rcpppara escrever código que fizesse um loop e um curto-circuito (ou seja, retornasse TRUE assim que a condição fosse atendida uma vez para um dado x, em vez de passar por todas as comparações)

    • 6

  3. Editar: Como o usuário ThomasIsCoding gentilmente aponta, há data.table::inrange()um que é vetorizado:

    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

relate perguntas