在生成器表达式中使用“状态变量”

Question

好吧，在我的应用程序中，我基本上需要使用“列表中所有先前项的总和”，但是列表不仅仅是一个列表，而是一个元数据生成器（其中第三个索引是“违规”索引）。

举个例子（显然不是有效的代码，但是应该可以证明这个想法）：

def calculate_minimal_charge_rate(self) -> float:
    o = self.orbit
    shadow_time = o.generate_shadow_light_time_list()
    P = self.getIdlePowerConsumption()
    a = 0
    return max(a += ind[1]; P * ind[2].period / \
            (ind[2].period - a) for ind in shadow_time)

或一个最小的工作示例（请注意，这很愚蠢，因为itertools的另一个功能可以为我解决这个问题）

def calculate_minimal_charge_rate(self) -> float:
    o = self.orbit
    shadow_time = o.generate_shadow_light_time_list()
    P = self.getIdlePowerConsumption()
    a = 0
    return max(a += ind[1]; a for ind in shadow_time)

现在显而易见的“解决方案”是使用for循环：

def calculate_minimal_charge_rate(self) -> float:
    o = self.orbit
    shadow_time = o.generate_shadow_light_time_list()
    P = self.getIdlePowerConsumption()
    a = 0
    max_power = 0
    for ind in shadow_time:
        a += ind[1] 
        preq = P * ind[2].period / \
                (ind[2].period - a)
        if preq > max_power:
            max_power = preq
    return max_power

但是，对于这样一个简单的“东西”来说，这不是太冗长吗？ 出于完整性考虑，这是orbit.generate_shadow_light_time_list() ：

def generate_shadow_light_time_list(self):
    """
    Returns total max time in shadow
    Iterativelly calls max_time_in_shadow for each orbit until star is found
    """
    o = self
    try:
        while o.parent.brightness <= 0:
            t = o.max_time_in_shadow()
            yield (o.period - t, t, o)
            o = o.parent.orbit
    except AttributeError:
        return
    return

Answer 1

由于您尝试在生成器表达式中累积状态，因此需要一个单独的对象来为您维护状态。 就维护和可读性而言，当前的for循环解决方案可能是最简单的方法。 另一种选择是编写自己的生成器，该生成器将保持所需的状态并将其传递给max 。 感谢@ patrick-haugh建议此解决方案：

def calculate_minimal_charge_rate(self) -> float:
    def get_that_quantity():
        a = 0
        for item in shadow_time:
            a += item[1]
            p = item[2].period
            yield power * p / (p - a)

    power = self.getIdlePowerConsumption()
    shadow_time = self.orbit.generate_shadow_light_time_list()
    return max(get_that_quantity())

希望查看此代码可以使您确信没有充分的理由使用它。 唯一的优点是您基本上是通过生成器将大部分for循环传递给max ，而不是自己实现max的逻辑。 您的案例足够专业，在这种情况下，您不太可能从混淆中获得额外收益。

更新

通过将参数移到使用生成器的函数中，我消除了将参数传递给生成器的需要。 可以说这使实现与for循环相提并论，因为它没有引入任何新代码，但确实删除了max逻辑。