Python: передать дополнительные аргументы вызываемой - PullRequest
Новая
       8

Python: передать дополнительные аргументы вызываемой

0 голосов
/ 29 августа 2018

У меня есть следующий код Python: 

import networkx as nx 

def cost(i, j, d, value1, value2):
    # some operation involving all these values
    return cost


# graph is a networkx graph
# src, dst are integers
# cost is a callable that is passed 3 values automatically i.e. src, dst and d 
# d is a dictionary of edge weights
path = nx.dijkstra_path(graph, src, dst, weight=cost)

Теперь я хочу передать два значения value1 и value2 в функцию cost.

Документация networkx гласит, что weight может быть вызываемым, который принимает ровно 3 аргумента. Но мне нужны value1 и value2 для расчетов. Как это можно сделать?

Редактировать Решение с использованием functools работает хорошо. Однако моя функция в классе выглядит следующим образом: 

import networkx as nx 
import functools
class Myclass:
    def cost(self, i, j, d, value2):
        # some operation involving all these values
        # also need self


    # graph is a networkx graph
    # src, dst are integers
    # cost is a callable that is passed 3 values automatically i.e. src, dst and d 
    # d is a dictionary of edge weights
    # path = nx.dijkstra_path(graph, src, dst, cost)
    cost_partial = functools.partial(cost, self=self, value2=5)
    path = nx.dijkstra_path(graph, src, dst, cost_partial)

При таком подходе nx.dijkstra_path настаивает на присвоении src self. Таким образом, переводчик жалуется, что self присваивается несколько значений. Я нуждаюсь в себе для расчета стоимости.

Ответы [ 2 ]

0 голосов
/ 29 августа 2018

Это в значительной степени зависит от значения на значения value1 и value2. Я бы предложил добавить обертку, вызываемую по networkx: 

def cost_wrapper(i, j, d):
    value1 = 0  # set values as needed
    value2 = 1
    return cost(i, j, d, value1, value2)

И предоставить его сетиx: 

path = nx.dijkstra_path(graph, src, dst, weight=cost_wrapper)

Или просто сделайте их глобальными переменными, а не аргументами.

0 голосов
/ 29 августа 2018

Вам просто нужна функция, которая упаковывает cost. Быстрый способ - использовать functools.partial. 

import functools

def cost(i, j, d, value1, value2):
    'Example function'
    return i, j, d, value1, value2

# This makes a new function which takes all parameters of `cost` except those 
# you've already passed. Here I'm going to set value1 and value2 to 1 and 2 
# respectively.
cost_partial = functools.partial(cost, value1=1, value2=2)

# A new function that only accepts the 3 args i, j, d
cost_partial('i', 'j', 'd')  # --> ('i', 'j', 'd', 1, 2)

# use as
path = nx.dijkstra_path(graph, src, dst, weight=cost_partial)
...