如何使用Python从NetworkX中的特定节点属性获取值

时间:2014-04-23 01:18:14

标签: python networkx

我正在处理一个群组项目,我们需要在我们正在处理的图表中创建一个特定节点属性的所有值的列表。每个节点有6个属性,我们只需要一个属性列表中的所有值。

import networkx as nx
import scipy as sp
import matplotlib.pyplot as plt
%matplotlib inline 
import urllib

url = "http://josquin.cti.depaul.edu/~rburke/courses/s14/fmh.graphml"
sock = urllib.urlopen(url)  # open URL
fmh = nx.read_graphml(sock)

for i in fmh:
    if fmh.node[i]['Race'] == 'Asian':
        fmh.add_node(i, RaceN=0)
    elif fmh.node[i]['Race'] == 'Black':
        fmh.add_node(i, RaceN=1)
    elif fmh.node[i]['Race'] == 'Hisp':
        fmh.add_node(i, RaceN=2)
    elif fmh.node[i]['Race'] == 'NatAm':
        fmh.add_node(i, RaceN=3)
    elif fmh.node[i]['Race'] == 'Other':
        fmh.add_node(i, RaceN=4)
    elif fmh.node[i]['Race'] == 'White':
        fmh.add_node(i, RaceN=5)

for i in fmh:
    if fmh.node[i]['Sex'] == 'F':
        fmh.add_node(i, SexN=0)
    elif fmh.node[i]['Sex'] == 'M':
        fmh.add_node(i, SexN=1)

我认为这一切都是相关的。预加载的数据有5个属性,我们添加了第6个,我们只想获取特定属性的值,比如说列出RaceN属性的所有值。

我们的想法是我们有一个整数列表,我们可以在其上调用sp.bincount()函数。

1 个答案:

答案 0 :(得分:6)

您的代码在没有更多信息的情况下无法运行。我想你想要的功能就像函数networkx.get_node_attributes():

In [1]: import networkx as nx

In [2]: G = nx.Graph()

In [3]: G.add_node(1,profit=17)

In [4]: G.add_node(2,profit=42)

In [5]: a = nx.get_node_attributes(G,'profit')

In [6]: a.items()
Out[6]: [(1, 17), (2, 42)]