Python：将一些参数传递给功能配置

如何解决Python：将一些参数传递给功能配置

我的问题类似于this one。想象一下我想使用pairwise_distances中的函数sklearn.metrics.pairwise。

def network_analysis(G):
    pairwise_distances(G.nodes,G.nodes,metric=my_metric)
    # blah blah
    #

我有

 def my_metric(x,y,G):
     return networkx.shortest_path(G,x,y)* (G.nodes[x]['weight']-G.nodes[y]['weight'])

但是我不知道如何在pairwise_distances调用中将G传递给我的指标。

附带说明，这是Networkx一直存在的问题，因为当节点由数字标识时，对于需要使用节点属性的每个功能，我都必须通过整个网络。

解决方法

假设您无法更改pairwise_distances，请尝试使用functools.partial。这使您可以执行以下操作：

import functools

def network_analysis(G_input):
    new_metric = functools.partial(my_metric,G = G_input)
    pairwise_distances(G_input.nodes,G_input.nodes,metric = new_metric)
    

新指标的行为如下：

def new_metric(x,y):
    return my_metric(x,y,G)  # G has already been passed in via functools.partial