将循环打包到函数中以禁用变量的python方法？

Question

我不知道我应该去多详细，所以请要求详细说明，如果这是太简洁了。你知道吗

有没有一种方法可以把for a,b,c in product(d['a'],d['b'],d['c']):打包成语法糖，这样我就只需要为这个循环本身键入一次静音变量a,b,c？你知道吗

像这样的事情可能是？你知道吗

my_for_loop('a','b','c'): 
  API_Call1(a)
  API_Call2(b,c)

而不是

for a,b,c in product(d['a'],d['b'],d['c']):
  API_Call1(a)
  API_Call2(b,c)

my_for_loop会是什么样子？我对如何从概念上处理这个问题有点迷茫。你知道吗

更多细节：

我有一个API，它要求为某些列表的笛卡尔积的每个单元格调用它。我使用乘积函数来避免嵌套循环。假设我们有一个列表1和一个列表2，它可以通过以下方式完成

from itertools import product
for a,b in product(list1,list2):
  API_Call(a,b)

我创建了一个dictionary_of_lists={'a':list1,'b':list2,'c':list3...} 能够这样写

for a,b in product(dictionary_of_lists['a'],dictionary_of_lists['b']):
  API_Call(a,b)

for c,b in product(dictionary_of_lists['c'],dictionary_of_lists['b']):
  API_Call(c,b)

for e,f,g,h in product(dictionary_of_lists['e'],dictionary_of_lists['f'],dictionary_of_lists['g'],dictionary_of_lists['h'],):
  API_Call1(e,f,g,h)
  API_Call2(e,h)
...

因此，循环创建的变量基本上是在API调用中使用的，它们是静音的，否则，它们的名称无关紧要。这些调用有很多，而且有一些复杂的逻辑围绕着它们。所以我想保持循环本身的简单性，如果我需要更改变量，我不必在每个循环的三个位置更改它们。你知道吗

my_for_loop('a','b'):
  API_Call(a,b)

my_for_loop('b','c'):
  API_Call(c,b)

my_for_loop('e','f','g','h'):
  API_Call1(e,f,g,h)
  API_Call2(e,h)
...

附加： 我已经简化了一些事情，但却惊讶地发现其中隐藏着模棱两可的地方：-）

感谢所有的答案到目前为止！你知道吗

这是一个很好的建议，有D产品包装。我有一个确实，只是不想抢先你的建议。你知道吗

变量名对于代码逻辑是静音的，但是为了维护代码，它们有一些含义。所以它们不能由一个字母组成。你知道吗

为了进一步澄清：我希望避免在“for…”部分、对dproduct wrapper的调用和API调用中使用变量名三次。两次-在对包装器的调用和在API中调用都是可以的，因为它反映了逻辑。你知道吗

下面是一个更详细的代码示例。你知道吗

def dproduct(d, keys):
  subset_d = dict((k, d[k]) for k in keys if k in d)
  return product(*[subset_d.values()])

for foo, bar in dproduct(d, ['foo','bar',]):
  some logic here
    if API_Call1(foo,bar) == 123: 
       some other stuff, API_Call6(quux,baz,)         
  some more stuff and a call to another dproduct
  for quux, sl, baz in dproduct(d, ['quux','sl','baz',]):
    blah blah, API_Call2(quux,sl,baz) 
    other stuff

 for pa, mf in dproduct(d, ['pa','mf',]):
  API_Call4(pa,mf)

 for quux, sl, baz in dproduct(d, ['quux','sl','baz',]):
  further logic
    if API_Call1(quux, sl, baz) == 342: some other stuff
  some more stuff and a call to another dproduct
  for pa,mf in dproduct(d, ['pa','mf',]):
    API_Call3(pa,mf)

Answer 1

您不能轻易地将变量插入到本地名称空间中（可能可以但不应该没有正当理由）。所以使用字典来保存命名的值。你知道吗

我使用了operator.itemgetter函数来获取各种API调用所需的函数，并将产品函数包装在生成器中。你知道吗

from operator import itemgetter
from itertools import product

class DictCaller(dict):
    def __call__(self, fn, *args):
        fn(*map(itemgetter(self), args))

def my_product(d, *args):
    for xs in product(*map(itemgetter(d), args)):
        yield DictCaller(zip(args, xs))

for caller in my_product(*'abc'):
    caller(API_CALL, *'ab')
    caller(API_CALL1, *'bc')

Answer 2

首先，您可以这样编写product包装器：

def dproduct(d, keys):
    return product(*(d[key] for key in keys))

for a, b, c in dproduct(d, 'abc'):
    API_Call1(a)
    API_Call2(b, c)

（请注意，您可以用operator.itemgetter而不是genexpr编写相同的内容；这取决于您发现哪个更可读。）

当然，我是在利用这样一个事实，即所有键都有一个字符名，而字符串是其字符的一个iterable。如果你的真名不是一个字符，你必须稍微详细一点：

for a, b, c in dproduct(d, ('spam', 'eggs', 'beans')):

…此时您可能需要考虑在参数中使用*args而不是args，或者，如果您不介意使用hacky：

def dproduct(d, keys):
    return product(*(d[key] for key in keys.split()))

for a, b, c in dproduct(d, 'spam eggs beans'):

---

然后你可以更进一步，在你的电话周围写包装。例如，如果生成值的dict而不是值的元组，则可以使用这些dict，方法与使用原始列表的dict相同：

def dproduct(d, keys):
    for vals in product(*(d[key] for key in keys)):
        yield dict(zip(keys, vals))
def call1(d, keys):
    return API_Call1(*(d[key] for key in keys))
def call2(d, keys):
    return API_Call2(*(d[key] for key in keys))

for vals in dproduct(d, 'abc'):
    call1(vals, 'a')
    call2(vals, 'bc')

---

或者，如果您的API_Call*函数可以采用关键字而不是位置参数，则可以使事情变得更简单、更清晰：

def dproduct(d, keys):
    for vals in product(*(d[key] for key in keys)):
        yield dict(zip(keys, vals))
def dselect(d, keys):
    return {key: d[key] for key in keys}
for vals in dproduct(d, 'abc'):
    API_Call1(**dselect(vals, 'ab'))
    API_Call2(**dselect(vals, 'c'))

---

如果不能使用关键字参数，并且有许多API_Call*函数，则可以更进一步地动态生成包装器：

def apiwrap(apicall):
    @functools.wraps(apicall)
    def wrapper(d, keys):
        return apicall(*(d[key] for key in keys))
    return wrapper
apicall1 = apiwrap(API_Call1)
apicall2 = apiwrap(API_Call2)
# etc.

虽然如果你有很多这样的东西，你可能首先会想把它们放在一个列表或者一个字典里

---

如果您想变得太聪明，甚至可以根据API函数的签名动态拆分元组：

def dispatch(d, keys, *calls):
    for vals in product(*(d[key] for key in keys)):
        it = iter(vals)
        for call in calls:
            args = islice(it, len(signature(call).parameters))
            call(*args)
dispatch(d, 'abc', API_Call1, API_Call2)

（如果您的函数体非常小，那么您可能希望通过在函数顶部执行argcounts = [len(signature(call).parameters for call in calls]，然后每次在内部循环中使用zip(calls, argcounts)而不是使用inspect来加快速度。）

---

不管怎样，如果不了解你的程序，很难说你到底能做些什么，你应该做些什么。这些想法中的大多数通常都不是很像python，即使它们在某些特殊情况下可能有用。你知道吗

但是无论哪种方式，它们都应该成为你可以很容易地做的事情的例子，而不必陷入涉及locals和globals或getframe的可怕的黑客攻击。你知道吗