同一组中正负匹配-R

时间:2019-03-14 09:57:29

标签: r dataframe dplyr

我想做以下事情:

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

我想要的输出是ID(或行)的列表,该列表对给定的calendar_week具有负匹配的正值->这意味着我想要例如ID 2和3,因为匹配项为-2到日历周2中的2。我不需要id 4,因为在日历周2中没有-3值,依此类推。

输出:

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

3 个答案:

答案 0 :(得分:1)

使用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

答案 1 :(得分:1)

如果如评论中所述,只需进行一次匹配,您可以尝试:

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

答案 2 :(得分: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