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

假设我有一个方法:

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() 或其他方式重写双循环？

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

请您参考如下方法:

不要使用 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]