如何动态地将谓词传递给过滤函数?

时间:2013-03-16 11:29:51

标签: python

我有一个名为eval_params的字典,如下所示:

In [1]: eval_params
Out[1]: 
{1: {'CLF': 'EMNB',
  'EM': 'False',
  'FEA': 'words',
  'NGRAM': '1-1',
  'PDS': 'T',
  'PSE': '-',
  'RN': '1dnf',
  'STEM': 'False',
  'UDS': 'He'},
 2: {'CLF': 'EMNB',
  'EM': 'False',
  'FEA': 'words',
  'NGRAM': '1-1',
  'PDS': 'TS',
  'PSE': '-',
  'RN': '1dnf',
  'STEM': 'False',
  'UDS': 'He'},
 3: {'CLF': 'EMNB',
  'EM': 'False',
  'FEA': 'words',
  'NGRAM': '1-1',
  'PDS': 'T',
  'PSE': '-',
  'RN': '1dnf',
  'STEM': 'False',
  'UDS': 'Te'}}

我必须通过某些谓词来过滤这个词典,我使用内置函数filter

In [2]: filter(pred_func, eval_params.iteritems())
Out[2]: 
[(3,
  {'CLF': 'EMNB',
   'EM': 'False',
   'FEA': 'words',
   'NGRAM': '1-1',
   'PDS': 'T',
   'PSE': '-',
   'RN': '1dnf',
   'STEM': 'False',
   'UDS': 'Te'})]    

我的问题是我不想对pred_func中的谓词进行硬编码,而是将它们动态传递给pred_func。当前pred_func看起来像这样:

def pred_func((i, d)):
    # I have to hard-code the parameters to filter by. I don't want that
    return d['PDS'] == 'T' and d['UDS'] = 'Te'

相反,我希望有类似的东西:

def pred_func((i, d), predicates):
    vals = []
    # predicates would be a dictionary such as {'PDS':'T', 'UDS':'Te'}
    for param, val in predicates.iteritems():
        vals.append(d[param] == val)
    return all(vals)

但是,据我所知,我不能将额外的参数传递给谓词函数,那么如何以优雅的方式完成我想要的东西呢?有什么想法吗?

1 个答案:

答案 0 :(得分:4)

创建一个higher-order function [wikipedia],即一个返回函数的函数。例如:

def pred_func(predicates):
    # Define a function inside `pred_func`. This function is a "closure" and 
    # will have access to the `predicates` parameter
    def func((i, d)):
        # You can simply pass a generator to `all`
        return all(d[param] == val for param, val in predicates.iteritems())

    # Return the function
    return func

pred_func现在是一个函数,返回一个接受元组作为参数的函数(就像你原来的pred_func函数一样)。

然后您可以使用以下方法调用该函数:

filter(pred_func(predicates), eval_params.iteritems())