กรองจุดยอดบนคุณสมบัติต่างๆ - Julia

โปรดยกตัวอย่างการทำงานขั้นต่ำในแบบที่เราสามารถคัดลอกและวาง แล้วเริ่มได้ทันที โปรดระบุด้วยว่าปัญหาเกิดขึ้นที่ใดในรหัส ด้านล่างนี้คือตัวอย่างสำหรับสถานการณ์ของคุณ:

Pkg.add("MetaGraphs")

using LightGraphs, MetaGraphs

g = DiGraph(5)
mg = MetaDiGraph(g, 1.0)

add_vertex!(mg)

add_edge!(mg,1,2)
add_edge!(mg,1,3)
add_edge!(mg,1,4)
add_edge!(mg,2,5)
add_edge!(mg,3,5)
add_edge!(mg,5,6)
add_edge!(mg,4,6)

set_props!(mg,3,Dict(:prop1=>1,:prop2=>2))
set_props!(mg,1,Dict(:prop1=>1,:prop2=>0))
set_props!(mg,2,Dict(:prop1=>1,:prop2=>0))
set_props!(mg,4,Dict(:prop1=>0,:prop2=>2))
set_props!(mg,5,Dict(:prop1=>0,:prop2=>2))
set_props!(mg,6,Dict(:prop1=>0,:prop2=>0))

function my_vertex_filter(g::AbstractMetaGraph, v::Integer, prop1, prop2)
  return has_prop(g, v, :prop1) && get_prop(g, v, :prop1) == prop1 &&
         has_prop(g, v, :prop2) && get_prop(g, v, :prop2) == prop2
end

prop1 = 1
prop2 = 2

col = collect(filter_vertices(mg, (g,v)->my_vertex_filter(g,v,prop1,prop2)))
# returns Int[3]

โปรดตรวจสอบ ?filter_vertices --- ซึ่งจะให้คำแนะนำว่าควรเขียนอะไร/อย่างไรเพื่อกำหนดตัวกรองที่คุณกำหนดเอง

แก้ไข สำหรับการกรองขอบ คุณสามารถดูที่ ?filter_edges เพื่อดูว่าคุณต้องการอะไรเพื่อให้ได้รับการกรองขอบ เพิ่มข้อความที่ตัดตอนมาด้านล่างต่อท้ายโซลูชันด้านบนเพื่อรับผลลัพธ์ของคุณ:

function my_edge_filter(g, e, prop1, prop2)
  v = dst(e) # get the edge's destination vertex
  return my_vertex_filter(g, v, prop1, prop2)
end

myedges = collect(filter_edges(mg, (g,e)->my_edge_filter(g,e,prop1,prop2)))
# returns [Edge 1 => 3]

ฉันพบวิธีแก้ปัญหานี้:

function filter_function1(g,prop1,prop2)
  fil1=filter_vertices(g,:prop1,prop1)
  fil2=filter_vertices(g,:prop2,prop2)
  filter=intersect(fil1,fil2)
  return filter
end

ดูเหมือนว่าจะใช้งานได้และค่อนข้างง่ายที่จะนำไปใช้ ฉันไม่รู้ว่าฟังก์ชัน filter_vertices ใช้พลังงานในการคำนวณมากหรือไม่ มิฉะนั้นดูเหมือนว่าจะใช้การวนซ้ำแบบง่าย ๆ เช่นนี้:

function filter_function2(g,prop1,prop2)
  filter=Set{Int64}()
  fil1=filter_vertices(g,:prop1,prop1)
  for f in fil1
      if get_prop(g,f,:prop2)==prop2
          push!(filter,f)
      end
  end
  return filter
end

ฉันเปิดรับคำตอบอื่น ๆ หากคุณมีคำตอบที่หรูหรากว่านี้