differentialLattice.py 文件源码

def link_export(self):

    from numpy import row_stack

    num = self.num
    links = self.links[:num*10, 0]

    edges = set()
    for i, c in enumerate(self.link_counts[:num,0]):
      for k in range(c):

        j = links[10*i+k]
        if i<j:
          lnk = (i, j)
        else:
          lnk = (j, i)

        if lnk not in edges:
          edges.add(lnk)

    return self.xy[:num,:], row_stack(list(edges))