R - Объединение фреймов данных по двум столбцам с дополнительным условием «между значениями»?

Question

У меня есть два фрейма данных, а именно:

Dataframe A:

code1 code2 element1 from to
c1a    c2a     e1a     1    15
c1a    c2a     e1b     17   50
c1a    c2b     e1c     14   67
c1b    c2c     e1d     1    20
c1b    c2d     e1e    40   60

Dataframe B:

code1 code2 element2 number
c1a    c2a     e2a    7
c1a    c2a     e2b    10
c1a    c2a     e2c    35

Мне в основном нужно присоединиться к ним, если from =< number <= to, чтобы получить что-то вроде:

РЕЗУЛЬТАТ ДАННЫХ

(Фрагмент, у меня недостаточно данных макета. Я хочу это слияние для полных фреймов данных A и B).

code1 code2 element1 element2 from   to   number
c1a    c2a     e1a    e2a      1     15     7
c1a    c2a     e1a    e2b      1     15     10
c1a    c2a     e1b    e2c      17    50     35

Я могу сделать это с помощью for l oop и проверить вручную, но мне было интересно, есть ли более «элегантный» способ сделать это?

Answer 1

Вы можете объединить данные, а затем filter значения, которые находятся в диапазоне.

Вы можете сделать это в dplyr

library(dplyr)
left_join(B, A, by = c('code1', 'code2')) %>% 
    filter(number >= from & number <= to)

#  code1 code2 element2 number element1 from to
#1   c1a   c2a      e2a      7      e1a    1 15
#2   c1a   c2a      e2b     10      e1a    1 15
#3   c1a   c2a      e2c     35      e1b   17 50

или в базе R:

subset(merge(B, A, by = c('code1', 'code2')), number >= from & number <= to)

Answer 2

Здесь используется fuzzyjoin::fuzzy_inner_join. Как я понимаю из ваших выходных данных, помимо критериев для from =< number <= to, вы хотели бы присоединиться по code1 и code2.

1. Присоединиться к code1 и code2 по равенству
2. Присоедините from к number по первому неравенству, т.е. from <= number
3. Присоедините to к number по второму неравенству, то есть number <= to

Особенность fuzzy_join заключается в том, что они выводят все столбцы в обоих фреймах данных.

-

library(fuzzyjoin)
fuzzy_inner_join(
      df_A, df_B,
      by = c(
        "code1" = "code1",
        "code2" = "code2",
        "from" = "number",
        "to" = "number"),
      match_fun = c(
        "code1" = function(l, r) l == r,
        "code2" = function(l, r) l == r,
        "from" = function(l, r) l <= r,
        "to" = function(l, r) r <= l))

# code1.x code2.x element1 from to code1.y code2.y element2 number
# 1     c1a     c2a      e1a    1 15     c1a     c2a      e2a      7
# 2     c1a     c2a      e1a    1 15     c1a     c2a      e2b     10
# 3     c1a     c2a      e1b   17 50     c1a     c2a      e2c     35

данные

df_A <- structure(list(code1 = c("c1a", "c1a", "c1a", "c1b", "c1b"), 
    code2 = c("c2a", "c2a", "c2b", "c2c", "c2d"), element1 = c("e1a", 
    "e1b", "e1c", "e1d", "e1e"), from = c(1L, 17L, 14L, 1L, 40L
    ), to = c(15L, 50L, 67L, 20L, 60L)), class = "data.frame", row.names = c(NA, -5L))

df_B <- structure(list(code1 = c("c1a", "c1a", "c1a"), code2 = c("c2a", 
"c2a", "c2a"), element2 = c("e2a", "e2b", "e2c"), number = c(7L, 
10L, 35L)), class = "data.frame", row.names = c(NA, -3L))