python - 对代码的小改动给出了 AttributeError(Python、Networkx)

Question

我目前正在做一个项目，该项目涉及使用 networkx 制作一个无向图，其中包含基于 csv 文件的节点位置和具有各种属性的类补丁，以指示它们的位置、状态和每个节点的每个最近邻居的索引.然后在这组节点上执行迁移算法。

我目前想要扩展我的代码，以便我可以拥有更多的节点，因为这个项目本质上是一个天文数字。我使用的是常规欧几里德距离，由于二次缩放，这对于大量人口来说非常慢。因此，我采用了最近邻算法来查看一定距离内的节点。然后我制作了这些的 csv，即每个最近邻居节点的索引的 csv。然后，我尝试在这些节点之间绘制边，以便在生成的图形上执行我的算法。

但是，一旦我进行了此更改，下一小段代码就会出现 AttributeError:“numpy.int64”对象没有属性“pos”。

下面是我的代码，欧几里得距离计算被注释掉了，方便对比。中断的代码块是紧跟在边绘图 block 之后的代码块，它是后面算法的一种设置。我为愚蠢的变量名称道歉 - boba 是最近的邻居节点索引，而 bubu 是绘制了边缘的节点的索引。

我在这里被难住了，因为我不知道这个变化是如何连接的，也不知道为什么会抛出这个错误。

nearand = np.genfromtxt('/Users/Skippy/nearand.csv', delimiter = ',',usecols=np.arange(0, 3)) # 3D cartesian co-ordinates of stars


density = 0.14 #Stellar density per cubic parsec
L = 100 # Size of box

Patches = int(0.056*density*L**3+15)


P_init = 0.0001   # Probability that a patch will be occupied at the beginning
Distance = 10 # An arbitrary parameter to determine which patches are     connected

xcoord = nearand[:,0]
ycoord = nearand[:,1]
zcoord = nearand[:,2]

bub = np.asarray(np.linspace(0, Patches, Patches, dtype = 'int'))
print bub
bobbington = np.asarray(np.genfromtxt('bobbington2.csv', delimiter = ',')) # csv of nearest neighbour node indices
bobbington0 = bobbington[:,0]
bobbington1 = bobbington[:,1]
bobbington2 = bobbington[:,2]
bobbington3 = bobbington[:,3]
bobbington4 = bobbington[:,4]
bobbington5 = bobbington[:,5]
bobbington6 = bobbington[:,6]
bobbington7 = bobbington[:,7]
bobbington8 = bobbington[:,8]
bobbington9 = bobbington[:,9]
bobbington10 = bobbington[:,10]
bobbington11 = bobbington[:,11]
bobbington12 = bobbington[:,12]


class patch:
    def __init__(self,status=0,pos=(0,0,0),boba = (0,0,0,0,0,0,0,0,0,0,0,0,0,),bubu = 0):
    self.status = status
    self.pos = pos
    self.boba = boba
    self.bubu = bubu
def __str__(self):
    return(str(self.status))

G = nx.Graph()

for i in xrange(Patches):

    Stat = 1 if np.random.uniform() < P_init else 0
    Pos  = (xcoord[i], ycoord[i], zcoord[i])
    Bob = (bobbington0[i],bobbington1[i],bobbington2[i],bobbington3[i],bobbington4[i],bobbington5[i],bobbington6[i],bobbington7[i],bobbington8[i],
bobbington9[i],bobbington10[i],bobbington11[i],bobbington12[i])
    Bubu = bub[i]
    G.add_node(patch(Stat,Pos,Bob,Bubu))

#for p1 in G.nodes():
    #for p2 in G.nodes():
        #Dist = np.sqrt((p1.pos[2] - p2.pos[2])**2 + (p1.pos[1]-p2.pos[1])**2+(p1.pos[0]-p2.pos[0])**2)
        #if Dist <= p1.dist:
        #if Dist <= Distance:
            #G.add_edge(p1,p2)

for i in G.nodes():
    edge1= (i.bubu,i.boba[0])
    edge2= (i.bubu,i.boba[1])
    edge3= (i.bubu,i.boba[2])
    edge4= (i.bubu,i.boba[3])
    edge5= (i.bubu,i.boba[4])
    edge6= (i.bubu,i.boba[5])
    edge7= (i.bubu,i.boba[6])
    edge8= (i.bubu,i.boba[7])
    edge9= (i.bubu,i.boba[8])
    edge10= (i.bubu,i.boba[9])
    edge11= (i.bubu,i.boba[10])
    edge12= (i.bubu,i.boba[11])
    edge13= (i.bubu,i.boba[12])

    G.add_edge(*edge1)
    G.add_edge(*edge2)
    G.add_edge(*edge3)
    G.add_edge(*edge4)
    G.add_edge(*edge5)
    G.add_edge(*edge6)
    G.add_edge(*edge7)
    G.add_edge(*edge8)
    G.add_edge(*edge9)
    G.add_edge(*edge10)
    G.add_edge(*edge11)
    G.add_edge(*edge12)
    G.add_edge(*edge13)

pos = {} # THE BIT CAUSING ISSUES
for i in G.nodes():
    pos[i] = i.pos


occup = [n.status for n in G]   # THIS ALSO HAS NO 'status' ATTRIBUTE ERROR

Time = [0]
Occupancy = [np.sum([n.status for n in G])/float(Patches)]

##Then algorithm goes here##

谁能看出是什么问题？

Answer 1

在您的代码中，变量 bub 在我看来是一个 numpy 数组。

在执行 G.add_node(patch(...)) 时创建的每个补丁中，参数 Bubu 被设置为来自 bub 的条目。所以这是一个数字。然后对于那个补丁，self.bubu 是一个数字。

当您添加边时，您是在从 i.bubu 到 i.boba[some index] 添加边。因此，由于 i.bubu 是一个数字，您正在创建从这些数字到其他数字的边。这将导致 networkx 添加这些数字的节点。

所以在进入循环之前 for i in G.nodes(): pos[i] = i.pos G 中的许多节点都是数字而不是补丁。

对于这些数字，引用number.pos 没有意义。这是您为补丁类定义的内容。