Обратный вызов удаления/удаления узла Networkx

Question

Заданный ориентированный сетевой график. Я хотел бы иметь функцию, например «abc()», для каждого узла, который удален из графика. Я просмотрел документацию networkx, но не нашел такой функции обратного вызова.

То, что я считал:

1. Добавление вызова abc() к (Разрушитель) метод __del__() объекта, связанного с узлом. Тем не менее, помимо ловушек использования __del__() (см., Например, here), это не работает, если какие-либо ссылки на объект узла существуют где-то в памяти.
2. Подкласс класса networkx.DiGraph() и переопределение метода remove_node(). Недостаток: для этого потребуется переопределить все методы, удаляющие узел, например. remove_nodes_from (есть ли еще?)
3. Поскольку реализация графика gridx основана на словарях, это может быть решением как-то «подключить» функцию del этого словаря. Однако вмешательство этого в сетевую систему кажется неуместным.

Что является самым простым способом реализации функции обратного вызова, которая вызывается каждый раз, когда удаляется узел networkx?

Answer 1

В этом случае вам нужно только подклассифицировать два метода. Вот пример из NetworkX импорта Digraph

class RemoveNodeDiGraph(DiGraph): 
    def remove_node(self, n): 
     DiGraph.remove_node(self, n) 
     print("Remove node: %s"%n) 

    def remove_nodes_from(self, nodes): 
     for n in nodes: 
      self.remove_node(n) 

if __name__=='__main__': 
    G = RemoveNodeDiGraph() 
    G.add_node('foo') 
    G.add_nodes_from('bar',weight=8) 
    G.remove_node('b') 
    G.remove_nodes_from('ar') 

Это не будет столь же быстро, как и оригинальные методы в классе орграфа (особенно remove_nodes_from()), но если у вас нет очень больших графов это не будет, вероятно, быть значительным. Если вам нужна более высокая производительность, вы можете скопировать код непосредственно из этих методов вместо вызова суперкласса.