data.py 文件源码

def depart(self, G, rng):
        n1 = G.order()
        n2 = rng.binomial(n1, 1.0 / self.k)

        if G.order() <= n2:
            old = G.nodes()
        else:
            old = rng.choice(G.nodes(), n2, replace = False)
            old = old.tolist()

        for v in old:
            self.stats["%s_patient_departed" % G.node[v]["bp"]] += 1
            self.stats["%s_donor_departed" % G.node[v]["bd"]] += 1

        G.remove_nodes_from(old)
        self.stats["departed"] += n2
        return nx.convert_node_labels_to_integers(G)