ค้นหาเพื่อนบ้านทันทีตามกลุ่มโดยใช้ตารางข้อมูลหรือ igraph

ฉันมีdata.table :

groups <- data.table(group = c("A", "B", "C", "D", "E", "F", "G"), 
                     code_1 = c(2,2,2,7,8,NA,5),
                     code_2 = c(NA,3,NA,3,NA,NA,2),
                     code_3 = c(4,1,1,4,4,1,8))

group code_1 code_2 code_3
  A      2     NA      4
  B      2      3      1
  C      2     NA      1
  D      7      3      4
  E      8     NA      4
  F     NA     NA      1
  G      5      2      8

สิ่งที่ฉันต้องการจะทำคือให้แต่ละกลุ่มค้นหาเพื่อนบ้านที่ใกล้เคียงโดยใช้รหัสที่มีอยู่ ตัวอย่างเช่น: กลุ่ม A มีกลุ่มเพื่อนบ้านทันที B, C เนื่องจาก code_1 (code_1 เท่ากับ 2 ในทุกกลุ่ม) และมีกลุ่มเพื่อนบ้านทันที D, E เนื่องจาก code_3 (code_3 เท่ากับ 4 ในกลุ่มเหล่านั้นทั้งหมด)

สิ่งที่ฉันพยายามทำคือสำหรับแต่ละรหัสโดยการตั้งค่าคอลัมน์แรก (กลุ่ม) ตามการแข่งขันดังนี้

groups$code_1_match = list()
for (row in 1:nrow(groups)){

  set(groups, i=row, j="code_1_match", list(groups$group[groups$code_1[row] == groups$code_1]))
}

  group code_1 code_2 code_3          code_1_match
    A      2     NA      4              A,B,C,NA
    B      2      3      1              A,B,C,NA
    C      2     NA      1              A,B,C,NA
    D      7      3      4                  D,NA
    E      8     NA      4                  E,NA
    F     NA     NA      1 NA,NA,NA,NA,NA,NA,...
    G      5      2      8                  NA,G

"kinda" นี้ใช้งานได้ แต่ฉันคิดว่าจะมีตารางข้อมูลมากขึ้นในการทำเช่นนี้ ฉันเหนื่อย

groups[, code_1_match_2 := list(group[code_1 == groups$code_1])]

แต่มันไม่ได้ผล

ฉันไม่มีเคล็ดลับตารางข้อมูลที่ชัดเจนที่จะจัดการกับมันได้หรือไม่

ผลกรณีที่ดีที่สุดของฉันจะมีลักษณะเช่นนี้ (ซึ่งในปัจจุบันจะต้องใช้วิธีการของฉันสำหรับทั้ง 3 คอลัมน์แล้วต่อผลลัพธ์ที่ได้)

group code_1 code_2 code_3    Immediate neighbors
  A      2     NA      4         B,C,D,E
  B      2      3      1         A,C,D,F
  C      2     NA      1         A,B,F
  D      7      3      4           B,A
  E      8     NA      4           A,D
  F     NA     NA      1           B,C
  G      5      2      8           

ใช้igraphรับเพื่อนบ้านระดับ 2 วางโหนดตัวเลขวางโหนดที่เหลือ

library(data.table)
library(igraph)

# reshape wide-to-long
x <- melt(groups, id.vars = "group")[!is.na(value)]

# convert to graph
g <- graph_from_data_frame(x[, .(from = group, to = paste0(variable, "_", value))])

# get 2nd degree neighbours
x1 <- ego(g, 2, nodes = groups$group)

# prettify the result
groups$res <- sapply(seq_along(x1), function(i) toString(intersect(names(x1[[ i ]]),
                                                                   groups$group[ -i ])))

#    group code_1 code_2 code_3        res
# 1:     A      2     NA      4 B, C, D, E
# 2:     B      2      3      1 A, C, D, F
# 3:     C      2     NA      1    A, B, F
# 4:     D      7      3      4    B, A, E
# 5:     E      8     NA      4       A, D
# 6:     F     NA     NA      1       B, C
# 7:     G      5      2      8           

ข้อมูลเพิ่มเติม

นี่คือลักษณะที่ข้อมูลของเรามีลักษณะก่อนแปลงเป็นวัตถุ igraph เราต้องการให้แน่ใจว่า code1 ที่มีค่า 2 แตกต่างจาก code2 ที่มีค่า 2 เป็นต้น

x[, .(from = group, to = paste0(variable, "_", value))]
#     from       to
#  1:    A code_1_2
#  2:    B code_1_2
#  3:    C code_1_2
#  4:    D code_1_7
#  5:    E code_1_8
#  6:    G code_1_5
#  7:    B code_2_3
#  8:    D code_2_3
#  9:    G code_2_2
# 10:    A code_3_4
# 11:    B code_3_1
# 12:    C code_3_1
# 13:    D code_3_4
# 14:    E code_3_4
# 15:    F code_3_1
# 16:    G code_3_8

นี่คือลักษณะเครือข่ายของเรา:

โปรดทราบว่าโหนดจะเชื่อมต่อผ่านA..G code_x_yดังนั้นเราจำเป็นต้องได้รับปริญญาที่ 2ego(..., order = 2)ให้เพื่อนบ้านของเรารวมถึงเพื่อนบ้านระดับที่ 2 และส่งกลับรายการวัตถุ

ในการรับชื่อ:

lapply(x1, names)
# [[1]]
# [1] "A"        "code_1_2" "code_3_4" "B"        "C"        "D"        "E"       
# 
# [[2]]
# [1] "B"        "code_1_2" "code_2_3" "code_3_1" "A"        "C"        "D"        "F"       
# 
# [[3]]
# [1] "C"        "code_1_2" "code_3_1" "A"        "B"        "F"       
# 
# [[4]]
# [1] "D"        "code_1_7" "code_2_3" "code_3_4" "B"        "A"        "E"       
# 
# [[5]]
# [1] "E"        "code_1_8" "code_3_4" "A"        "D"       
# 
# [[6]]
# [1] "F"        "code_3_1" "B"        "C"       
# 
# [[7]]
# [1] "G"        "code_1_5" "code_2_2" "code_3_8"

เพื่อให้ได้ผลลัพธ์ที่ชัดเจนเราจำเป็นต้องลบcode_x_yโหนดและโหนดต้นทาง (โหนดที่ 1)

sapply(seq_along(x1), function(i) toString(intersect(names(x1[[ i ]]), groups$group[ -i ])))
#[1] "B, C, D, E" "A, C, D, F" "A, B, F"    "B, A, E"    "A, D"       "B, C"       ""   

อาจมีวิธีปฏิบัติที่เป็นประโยชน์มากกว่าในการบรรลุเป้าหมายนี้ แต่คุณสามารถทำสิ่งนี้ได้โดยใช้การหลอมละลายและการรวม:

mgrp <- melt(groups, id.vars = "group")[!is.na(value)]
setkey(mgrp, variable, value)
for (i in seq_along(groups$group)) {
  let = groups$group[i]
  set(
    groups, 
    i = i, 
    j = "inei", 
    value = list(mgrp[mgrp[group == let], setdiff(unique(group), let)])
  )
}

groups
#    group code_1 code_2 code_3    inei
# 1:     A      2     NA      4 B,C,D,E
# 2:     B      2      3      1 A,C,D,F
# 3:     C      2     NA      1   A,B,F
# 4:     D      7      3      4   B,A,E
# 5:     E      8     NA      4     A,D
# 6:     F     NA     NA      1     B,C
# 7:     G      5      2      8       

นี่เป็นแรงบันดาลใจจากการละลายของ @ sindri_baldur วิธีนี้:

1. ละลายกลุ่ม
2. ดำเนินการการเข้าร่วมคาร์ทีเซียนด้วยตนเอง
3. วางกลุ่มทั้งหมดที่ตรงกันไว้ด้วยกัน
4. เข้าร่วมกับ DT ดั้งเดิม

library(data.table)
#> Warning: package 'data.table' was built under R version 3.6.2
groups <- data.table(group = c("A", "B", "C", "D", "E", "F", "G"), code_1 = c(2,2,2,7,8,NA,5), code_2 = c(NA,3,NA,3,NA,NA,2), code_3=c(4,1,1,4,4,1,8))

molten_grps = melt(groups, measure.vars = patterns("code"), na.rm = TRUE)

inei_dt = molten_grps[molten_grps,
            on = .(variable, value),
            allow.cartesian = TRUE
            ][,
              .(inei = paste0(setdiff(i.group, .BY[[1L]]), collapse = ", ")),
              by = group]

groups[inei_dt, on = .(group), inei := inei]

groups
#>     group code_1 code_2 code_3       inei
#>    <char>  <num>  <num>  <num>     <char>
#> 1:      A      2     NA      4 B, C, D, E
#> 2:      B      2      3      1 A, C, D, F
#> 3:      C      2     NA      1    A, B, F
#> 4:      D      7      3      4    B, A, E
#> 5:      E      8     NA      4       A, D
#> 6:      F     NA     NA      1       B, C
#> 7:      G      5      2      8

ตามที่กล่าวไว้โดย zx8754 ใช้data.table::meltกับcombnแล้วigraph::as_adjacency_matrix

library(data.table)
df <- melt(groups, id.vars="group", na.rm=TRUE)[,
    if (.N > 1L) transpose(combn(group, 2L, simplify=FALSE)), value][, (1) := NULL]

library(igraph)
as_adjacency_matrix(graph_from_data_frame(df, FALSE))

เอาท์พุท:

7 x 7 sparse Matrix of class "dgCMatrix"
  A B C E D G F
A . 1 1 1 1 1 .
B 1 . 2 . 1 1 1
C 1 2 . . . 1 1
E 1 . . . 1 1 .
D 1 1 . 1 . . .
G 1 1 1 1 . . .
F . 1 1 . . . .

หรือไม่ใช้ igraph

x <- df[, unique(c(V1, V2))]
df <- rbindlist(list(df, data.table(x, x)))
tab <- table(df)   #or xtabs(~ V1 + V2, data=df)
ans <- t(tab) + tab
diag(ans) <- 0L
ans

เอาท์พุท:

   V1
V2  A B C D E F G
  A 0 1 1 1 1 0 1
  B 1 0 2 1 0 1 1
  C 1 2 0 0 0 1 1
  D 1 1 0 0 1 0 0
  E 1 0 0 1 0 0 1
  F 0 1 1 0 0 0 0
  G 1 1 1 0 1 0 0