Соответствие положительных и отрицательных чисел в той же группе - R

Question

Я бы хотел сделать следующее:

id calendar_week value
1    1              10
2    2              2
3    2             -2
4    2              3
5    3              10 
6    3             -10

Вывод, который я хочу получить, это список идентификаторов (или строк), которые имеют положительное и отрицательное соответствие для данного calendar_week ->, что означает, что я хочу, например, идентификаторы 2 и 3, потому что есть совпадение -2 до 2 в календарной неделе 2. Мне не нужен идентификатор 4, потому что в календарной неделе 2 нет значения -3 и т. д.

выход:

id calendar_week value 
2    2              2
3    2             -2
5    3              10 
6    3             -10

Answer 1

Может также сделать:

library(dplyr)

df %>%
  group_by(calendar_week, ab = abs(value)) %>%
  filter(n() > 1) %>% ungroup() %>%
  select(-ab)

Вывод:

# A tibble: 4 x 3
     id calendar_week value
  <int>         <int> <int>
1     2             2     2
2     3             2    -2
3     5             3    10
4     6             3   -10

Учитывая ваши дополнительные разъяснения, вы можете сделать:

df %>%
  group_by(calendar_week, value) %>%
  mutate(idx = row_number()) %>%
  group_by(calendar_week, idx, ab = abs(value)) %>%
  filter(n() > 1) %>% ungroup() %>%
  select(-idx, -ab)

В измененном фрейме данных:

  id calendar_week value
1  1             1    10
2  2             2     2
3  3             2    -2
4  3             2     2
5  4             2     3
6  5             3    10
7  6             3   -10
8  7             4    10
9  8             4    10

Это дает:

# A tibble: 4 x 3
     id calendar_week value
  <int>         <int> <int>
1     2             2     2
2     3             2    -2
3     5             3    10
4     6             3   -10

Answer 2

Если, как указано в комментариях, требуется только одно совпадение, вы можете попробовать:

library(dplyr)

df %>%
  group_by(calendar_week, nvalue = abs(value)) %>% 
  filter(!duplicated(value)) %>%
  filter(sum(value) == 0) %>% 
  ungroup() %>%
  select(-nvalue)

     id calendar_week value
  <int>         <int> <int>
1     2             2     2
2     3             2    -2
3     5             3   -10
4     6             3    10

Answer 3

Использование tidyverse:

library(tidyverse)

df %>%
  group_by(calendar_week) %>%
  nest() %>%
  mutate(values = map_chr(data, ~ str_c(.x$value, collapse = ', '))) %>%
  unnest() %>%
  filter(str_detect(values, as.character(-value))) %>%
  select(-values)

Выход:

calendar_week    id value
          <dbl> <int> <dbl>
1             2     2     2
2             2     3    -2
3             3     5    10
4             3     6   -10