问题描述
G=nx.Graph()
G.add_node('N1',n_type=1)
G.add_node('N2',n_type=1)
G.add_node('N3',n_type=2)
G.add_edges_from([('N3','N1'),('N3','N2')])
如果我使用NetworkX的nx.average_neighbor_degree(G)函数,则会得到以下结果。
{'N1': 2.0,'N2': 2.0,'N3': 1.0}
但是,我想仅考虑一种类型的节点来获得平均邻居度。例如,属性取值为1时的平均邻居度应为:
{'N1': 0.0,'N2': 0.0,'N3': 1.0}
这是因为节点N3是唯一具有属性n_type = 1的邻居并且其邻居的平均值为度(N1)+度(N2)/ 2 = 1 + 1/2 = 1
有什么建议吗?
解决方法
检查源代码并稍作调整,我可以提出这个建议。希望能有所帮助。
import networkx as nx
def get_average_nbr_deg_by_type(G,node_type):
avg = {}
for n,deg in G.degree:
nbrs_deg = [d for n,d in G.degree(G[n]) if G.nodes[n]['n_type'] == node_type]
deg = len(nbrs_deg)
if deg == 0:
deg = 1
avg[n] = sum(nbrs_deg) / float(deg)
return avg
get_average_nbr_deg_by_type(G,node_type=1)
# outputs {'N1': 0.0,'N2': 0.0,'N3': 1.0}