使用函数中定义的函数

Question

我尝试在我已经编写的函数中使用一个函数，它可以用于list comprehension和部分函数，但不能用于lambda函数。你知道吗

所以我的功能是：

import numpy as np
from scipy.optimize import minimize
from _functools import partial
from sklearn.metrics import mean_squared_error

arpdau = np.random.randint(0,100,15)

def fitARPDAU(arpdau, max_cohortday, method, par=None):

  valid = {'log', 'power', 'all'}
  if method not in valid:
      raise ValueError("results: method must be one of %r." % valid)

  values = par

  if method == 'log':

    if values == None:
      a = 1
      b = 0
      c = 1
      values = [a, b, c]
      bounds = [(1e-10, None), (1e-10, None), (None, None)]

    def getArpdauFunction(x, values):
      return values[0] * np.log(x + values[1]) + values[2]
  elif method == 'power':

    if values == None:
      a = 1
      b = 0
      c = .5
      d = 0
      values = [a, b, c, d]
      bounds = [(1e-10, None), (None, None), (1e-10, 1), (None, None)]


    def getArpdauFunction(x, values):
      return values[0] * (x + values[1]) ** values[2]+ values[3]

  elif method == 'all':

    log_loss = fitARPDAU(arpdau, max_cohortday, method='log', par=par)
    power_loss = fitARPDAU(arpdau, max_cohortday, method='power', par=par)

    combined_models = [log_loss, power_loss]
    losses = map(lambda x: x[0].fun, combined_models)
    return combined_models[np.argmin(losses)]

  def getLossOptim(values):

    # import ipdb; ipdb.set_trace()
    # arpdau_pred = [getArpdauFunction(x, values) for x in range(max_cohortday)]
    arpdau_pred_1 = map(lambda x: getArpdauFunction(x, values), range(max_cohortday))
    # arpdau_pred_2 = partial(getArpdauFunction, values=values)(range(271))
    return mean_squared_error(arpdau, arpdau_pred_1[:len(arpdau)])

  result = minimize(getLossOptim, values, method='L-BFGS-B', bounds=bounds)

  return result, [getArpdauFunction(x, result.x) for x in range(max_cohortday)], result.x, method, getArpdauFunction

print fitARPDAU(arpdau, 100, method='all', par=None)

在getLossOptim中，部分和列表理解可以工作，但是lambda函数不能工作，这有什么原因吗？你知道吗

lambda函数返回

NameError: global name 'getArpdauFunction' is not defined

谢谢！你知道吗

---

Answer 1

正如我在评论中所说，您的问题与在另一个函数中定义的函数无关。出现错误的原因是函数有时未定义；如果method既不是'log'也不是'power'，那么getArpdauFunction()就永远不会定义。你知道吗

这很容易解决。目前我能看到的最干净的方法是声明几个函数，然后动态地选择一个：

def identity(x): return x
def pp(x): return x+1
def mm(x): return x-1

functions = {
  "++": pp,
  " ": mm
}

default_function = identity

print(functions.get("++", default_function)(2)) # 3
print(functions.get(" ", default_function)(2)) # 1
print(functions.get("non existing function name", default_function)(2)) # 2

如果函数很短（如本例中所示），则可以直接编写：

functions = {
  "++": lambda x: x+1,
  " ": lambda x: x-1
}

Answer 2

这与在另一个函数中定义的函数无关。如果method不是'log'或'power'，那么getArpdauFunction()永远不会被定义。你知道吗

您可能应该为所有内容定义它，然后在method为'log'或'power'时重载它。你知道吗

这与您的问题无关，但是您也不应该使用if x == None。因为None是一个单体，所以使用if x is None更有效、更具python性