簡體   English   中英

如何使用igraph和python添加對象屬性條件的邊

[英]How to add edges that are conditional to object attribute using igraph and python

我的圖表的節點是具有一些方法的實例。 我已正確添加節點,但現在我想檢查,如果這些節點具有相同的group_id,則為add_edges。 我正在使用igraph

import igraph as ig

def making_graph(researchers):
    g = ig.Graph()
    for each in researchers:
        for i in range(len(each)):
            g.add_vertex(each[i])
            g.vs[i]['researcher_id'] = each[i].get_researcher_id()
            g.vs[i]['name'] = each[i].get_name()
            g.vs[i]['sex'] = each[i].get_sex()

    g.add_edges() # if researchers have the same group_id, add edge
    return g

研究員的課程是

class Researcher:

    def __init__(self, group_id, research_id, name, tit, sex,
             tot_nac_2011, tot_nac_2014, tot_int_2011, tot_int_2014, tot_bbl_2011, tot_bbl_2014):

        self.group_id = group_id
        self.research_id = research_id
        self.name = name
        self.tit = tit
        self.sex = sex
        self.producao = [tot_nac_2011, tot_nac_2014, tot_int_2011, tot_int_2014, tot_bbl_2011, tot_bbl_2014]

    def get_group_id(self):
        return self.group_id

    def get_researcher_id(self):
        return self.research_id

    def get_sex(self):
        return self.sex

當我打電話給making_graph我會傳遞一份包含研究人員名單的清單

編輯。 這也行不通。 為什么?

def making_graph(researchers):
    g = ig.Graph()
    for each in researchers:
        for i in range(len(each)):
            g.add_vertex(each[i])
            g.vs[i]['researcher_id'] = each[i].get_researcher_id()
            g.vs[i]['name'] = each[i].get_name()
            g.vs[i]['sex'] = each[i].get_sex()
            for other in researchers:
                for j in range(len(other)):
                    if each[i].get_group_id() == other[j].get_group_id():
                        g.add_edge([(g.vs[i], g.vs[j])])
return g

現在,似乎這確實有效:

def making_graph(researchers):
    g = ig.Graph()
    for each in researchers:
        for i in range(len(each)):
            g.add_vertex(each[i])
            g.vs[i]['group_id'] = each[i].get_group_id()
            g.vs[i]['researcher_id'] = each[i].get_researcher_id()
            g.vs[i]['name'] = each[i].get_name()
            g.vs[i]['sex'] = each[i].get_sex()
    for v in g.vs:
        for w in g.vs:
            if v['group_id'] == w['group_id'] and v != w and g.are_connected(v, w) is False:
                g.add_edge(v, w)
    return g

暫無
暫無

聲明:本站的技術帖子網頁,遵循CC BY-SA 4.0協議,如果您需要轉載,請注明本站網址或者原文地址。任何問題請咨詢:yoyou2525@163.com.

 
粵ICP備18138465號  © 2020-2024 STACKOOM.COM