SymPy求和时如何让迭代器随x取值逐次调用next()递增

问题根源 #

你当前代码的核心问题是：callFunc(next_expr())在定义logic表达式时就已经执行完毕，仅调用了一次next()获取固定值。后续SymPy进行求和或替换x时，所有项复用的都是这个固定值，导致结果全为相同数值。

解决代码 #

要实现每个x取值时调用一次next()，必须让next()的调用延迟到SymPy计算每一项时执行。我们可以通过SymPy的自定义函数来实现这一点：

from sympy import symbols, summation, Function
import random

x = symbols('x')

class DynamicIterator:
    def __init__(self, func, a, b):
        self.func = func
        self.counter = 0
        self.a = a
        self.b = b
        self.can_advance = True

    def __iter__(self):
        return self

    def __next__(self):
        if not self.can_advance:
            raise StopIteration
        value = self.func(self.counter, self.a, self.b)
        self.counter += 1
        return value

def generate_combinations(n, a, c):
    range_size = c - a + 1
    l = a + (n % range_size)
    return l

# 创建迭代器实例
expressions = DynamicIterator(generate_combinations, 1, 2**256 + 1)

def next_expr():
    return next(expressions)

# 定义SymPy自定义函数，每次求值时触发next()调用
class NextExpr(Function):
    @classmethod
    def eval(cls):
        return next_expr()

def formula109(l, integer_func):
    # 延迟获取迭代值，直到计算当前项时才调用next()
    integer_val = integer_func()
    random_number = random.randint(integer_val, integer_val + 1) + l
    random_number2 = random_number - l
    return random_number2

# 传入自定义函数对象，而非提前计算的固定值
logic = formula109(x, NextExpr)
# 先小范围测试，验证生效后再改回900000
total = summation(logic, (x, 1, 10))
print(total)

# 查看每个x对应的数值
terms = [logic.subs(x, k) for k in range(1, 10)]
print(terms)

核心修改说明 #

• 自定义SymPy函数：新增NextExpr类继承自sympy.Function，其eval方法会在SymPy计算每一项时自动调用，确保每个x对应一次next()调用。
• 延迟求值：修改formula109接收函数对象而非固定值，避免提前执行next()，让迭代值的获取与项的计算绑定。
• 简化冗余代码：移除了原代码中无用的callFunc函数及无关的SymPy导入，精简结构。

运行修改后的代码，每个x对应的迭代值会逐次递增，求和结果及各项数值将符合预期。

内容的提问来源于stack exchange，提问作者user27394478