如何迭代地将 2 个列表中的元素应用到新函数中？ [复制]

Question

假设我有 2 个列表，其中第一个列表的第 i 个元素对应于第二个列表的第 i 个元素。如何迭代地将 2 个列表中的元素应用到不同的函数中？

def GetLists(n):
    List1 = []
    List2 = []
    n = int(input("how many terms: "))
    for i in range(1,n):
        val1 = float(input("what is val1: "))
        val2 = float(input("what is the corresponding val2: "))
        List1.append(val1)
        List2.append(val2)
    return List1, List2

def newfunc(ListA, ListB, var):
    # take i-th elements from lists A and B where (a,b) are the i-th elements of A and B
    # want output = sum(a * var ** b) for index in len(A) if len(A) == len(B)

最pythonic的方法是什么？如果可能的话，我想这样做而不导入外部模块。

编辑：我检查了其他解决方案。 “重复”答案需要导入模块；我试图不这样做。此外，我正在尝试执行返回输出而不是打印值的操作，这使 zip 的使用复杂化，超出了重复答案中显示的级别。

Answer 1

从列表 A 和 B 中获取第 i 个元素，其中 (a,b) 是 A 和 B 的第 i 个元素

如果 len(A) == len(B)，则希望输出 = sum(a * var ** b) 用于 len(A) 中的索引

这就是你要找的吗？它将两个相同长度的列表压缩在一起，计算f(a, b, i) = a[i] * var ** b[i] 为一些常数var 和每个i 其中0 <= i < len(a)。然后它返回总和。

def process(list_a, list_b, var):
    if len(list_a) != len(list_b):
        raise ValueError('Lists are not equal in length')

    def combine():
        for a, b in zip(list_a, list_b):
            yield a * var ** b

    return sum(combine())

print(process([5, 2, 3], [2, 2, 3], 10))

输出

3300

这个输出是(1 * 10 ** 2) + (2 * 10 ** 2) + (3 * 10 ** 3)的结果。

编辑

上述方法将组合逻辑（您问题的重点）与求和逻辑分离。另一种可以说是更 Pythonic（根据您的要求）并且更短的替代方法是使用 generator expression，正如 cmets 中针对此答案所建议的那样：

def process(list_a, list_b, var):
    if len(list_a) != len(list_b):
        raise ValueError('Lists are not equal in length')
    
    return sum(a * var ** b for a, b in zip(list_a, list_b))

print(process([1, 2, 3], [2, 2, 3], 10))

基本上，sum 中的表达式充当了我在之前方法中定义的 生成器函数 combine 的匿名替换。