Найти соседа соседа - PullRequest
Новая
       12

Найти соседа соседа

3 голосов
/ 30 сентября 2019

У меня есть df, перечисляющий количество областей (df$area) и областей, с которыми они граничат (df$next_area). Исходя из этого я хочу получить аналогичный df, но с соседом его соседа. Я написал следующее, которое работает, но кажется чрезвычайно запутанным. Было ли лучшее решение? 

library(dplyr)
library(tidyr)

df <-     data.frame(area=c("A","A","B","B","C","C","C","D"),next_area=c("B","C","A"    ,"C","A","B","D","C") )
    df <- df %>% group_by(area) %>% 
  summarize(next_area = list(sort(unique(as.character(next_area)))))
    df$next_area_exploded <- df$next_area
    for(i in 1:nrow(df)){
      for(j in 1:length(df$next_area[[i]])){
        df$next_area_exploded[[i]][j] <-         list(df$next_area_exploded[[which(df$area==df$next_area[[i]][j])]])
  }
}
df$next_area_exploded <- lapply(df$next_area_exploded, function(x)         unique(unlist(x)))
for(i in 1:nrow(df)){
  df$next_next_area[[i]] <- df$next_area_exploded[[i]]    [!df$next_area_exploded[[i]] %in% df$next_area[[i]]]
  df$next_next_area[[i]] <- df$next_next_area[[i]][!df$next_next_area[[i]]     %in% df$area[[i]]]
  }
df <- df %>% unnest(next_next_area) %>% 
  group_by(area) %>% 
  mutate(col=paste0(seq_along(area),".add")) %>%
  spread(key=col, value=next_next_area)
df$next_area<-NULL; df$next_area_exploded<-NULL 
df_final <- df %>% gather(a,next_next,c(names(df)    [grepl(".add",names(df))])) %>% select(-a) %>% filter(!is.na(next_next))

Ответы [ 2 ]

5 голосов
/ 30 сентября 2019

Вы можете думать об этом как о графике, и для каждого узла найти все остальные узлы, которые находятся на расстоянии 2: 

library(igraph)

df <-  data.frame(area=c("A","A","B","B","C","C","C","D"),
                  next_area=c("B","C","A","C","A","B","D","C") )

g = graph_from_data_frame(df)

distances(g) %>%
    as_tibble(rownames = 'area') %>%
    gather(-area, key = 'next_next_area', value = 'distance') %>%
    filter(distance == 2)

Вывод: 

# A tibble: 4 x 3
  area  next_next_area distance
  <chr> <chr>             <dbl>
1 D     A                     2
2 D     B                     2
3 A     D                     2
4 B     D                     2
1 голос
/ 30 сентября 2019

Вот мой подход. Я получил всю потенциальную область соседа своего соседа и выбор области, а не начала и соседа. 

df %>% 
  inner_join(df %>% 
               rename(next_area = area, 
                      next_next_area = next_area),
             by = "next_area") %>% 
  filter(area != next_next_area) %>% 
  group_by(area) %>% 
  filter(! next_next_area %in% next_area) %>% 
  ungroup()
...