python - 映射具有多个参数的特定函数的最佳 pythonic 方法

Question

假设我有一个方法:

def add_force_coupling(g1, g2, name):
    print {(g1,g2): name}

然后是代码:

l = 3
for g1 in range(l):
    for g2 in range(l):
      name = 'G' + str(g1) + str(g2)
      add_force_coupling(g1, g2, name)

产生结果:

{(0, 0): 'G00'}, {(0, 1): 'G01'}, {(0, 2): 'G02'}, {(1, 0): 'G10'}, ..., {(2, 2): 'G22'}

现在我想通过使用 map() 函数使它更像 pythonic。到目前为止我有:

list1 = sorted(l*range(l))
list2 = l*range(l)
list3 = ["".join(values) for values in zip(l*l*'G',list(map(str,l1)),list(map(str,l2)))]
map( add_force_coupling, list1, list2, list3 )

这正是我想要的，但比标准双循环更难看，因为 int 列表需要在 list3 中转换为 str。也很难理解发生了什么。有没有更好的 pythonic 方法用 map()、zip() 或其他方式重写双循环？

注意:改变方法不是一种选择

Answer 1

不要使用 map()。使用列表理解和 itertools.product() :

from itertools import product

[{(i, j): 'G{}{}'.format(i, j)} for i, j in product(range(l), repeat=2)]

或将您的函数用于左侧表达式:

[add_force_coupling(i, j, 'G{}{}'.format(i, j))
 for i, j in product(range(l), repeat=2)]

令我印象深刻的是，您希望在这里建立一个字典；听写理解可以做到这一点:

{(i, j): 'G{}{}'.format(i, j) for i, j in product(range(l), repeat=2)}

在这里，您可以使用 result[i, j] 解决每个项目。

但是给定索引，嵌套列表将是一种更有效的数据存储机制:

[['G{}{}'.format(i, j) for j in range(l)] for i in range(l)]

并用

处理结果

result[i][j]