在Python中的嵌套列表推导中命名变量？

Question

就像标题所说，有没有什么方法可以命名 Python 中嵌套列表理解中使用的变量（即列表）？

我可以举出一个合适的例子，但我认为这个问题已经足够清楚了。

下面是伪代码的示例：

[... [r for r in some_list if r.some_attribute == Something_from_within_this_list 理解] ... [r for r in some_list if r.some_attribute == Something_from_within_this_list 理解] .. .]

有没有什么方法可以避免这里的重复，并简单地为此临时列表添加一个变量，仅用于列表理解中？

澄清： 列表理解已经运行良好，所以这不是“可以用列表理解来完成”的问题。而且它也比 for 语句的原始形式更快，因此它也不是“for 语句与列表推导式”问题之一。这只是一个通过单独为列表理解内部变量设置变量名来使列表理解更具可读性的问题。只是谷歌搜索我还没有真正找到任何答案。我找到了这个和这个，但这并不是我真正想要的。

Answer 1

根据我对你想做的事情的理解，不，你做不到。

您不能在列表推导式中执行作业，因为列表推导式本质上是这样的形式

[expression(x, y) for x in expression_that_creates_a_container 
                  for y in some_other_expression_that_creates_a_container(x)
                  if predicate(y, x)]

当然还有一些其他情况，但它们都是这样的。请注意，任何地方都不存在用于声明名称分配的空间。因此，除非使用 for my_variable in 语法，否则无法在列表理解的上下文中分配名称。

如果您的列表理解有效，您可以将其发布并查看是否可以简化。基于 itertools 的解决方案通常是 burly 列表推导式的一个很好的替代方案。

Answer 2

我想我完全理解你的意思，并且我想出了一个“部分解决方案”来解决这个问题。该解决方案工作正常，但效率不高。

让我用一个例子来解释一下：

我只是想解决一个毕达哥拉斯三元组，其总和为 1000。解决它的 python 代码只是：

def pythagoreanTriplet(sum):
    for a in xrange(1, sum/2):
        for b in xrange(1, sum/3):
            c = sum - a - b
            if c > 0 and c**2 == a**2 + b**2:
               return a, b, c

但我想以类似函数式编程的风格对其进行编码：

def pythagoreanTriplet2(sum):
    return next((a, b, sum-a-b) for a in xrange(1, sum/2) for b in xrange(1, sum/3) if (sum-a-b) > 0 and (sum-a-b)**2 == a**2 + b**2)

正如在代码中，我计算了3次（sum-ab），我想将结果存储在内部变量中以避免冗余计算。我发现做到这一点的唯一方法是添加另一个带有单个值的循环来声明内部变量：

def pythagoreanTriplet3(sum):
    return next((a, b, c) for a in xrange(1, sum/2) for b in xrange(1, sum/3) for c in [sum-a-b] if c > 0 and c**2 == a**2 + b**2)

它工作正常......但正如我在帖子开头所说，这不是一个有效的方法。将这 3 种方法与 cProfile 进行比较，每种方法所需时间如下：

• 第一种方法：0.077 秒
• 第二种方法：0.087 秒
• 第三种方法：0.109 秒

Answer 3

有些人可能将以下内容归类为“黑客”，但在某些情况下它绝对有用。

f = lambda i,j: int(i==j) #A dummy function (here Kronecker's delta)    
a = tuple(tuple(i + (2+f_ij)*j + (i + (1+f_ij)*j)**2
                for j in range(4)
                for f_ij in (f(i,j),) ) #"Assign" value f(i,j) to f_ij.
          for i in range(4) )
print(a)
#Output: ((0, 3, 8, 15), (2, 13, 14, 23), (6, 13, 44, 33), (12, 21, 32, 93))

如果函数 f 的计算成本很高，则这种方法特别方便。因为它有点不寻常，所以记录“赋值”行可能是个好主意，就像我上面所做的那样。

Answer 4

我只是在这里冒险，因为我不知道你到底想做什么。我只是猜测你试图将超出应有的程度强行塞入单一表达中。不要这样做，只需将子表达式分配给变量即可：

sublist = [r for r in some_list if r.some_attribute == something_from_within_this_list comprehension]
composedlist = [... sublist ... sublist ...]

Answer 5

此功能已在 Python 3.8 中添加（请参阅 PEP 572)，称为“赋值表达式”，运算符为 := 。

文档中的示例：

结果 = [(x, y, x/y) for x in input_data if (y := f(x)) > 0]

stuff = [[y := f(x), x/y] for x in range(5)]