Python联合枚举多个变量

Question

所以我想同时迭代两个列表，并且我还需要索引，如以下代码所示。两者对我来说都很简单，但是在第二种情况下是否也可以使用枚举？当然，假设两个 var 数组的长度相同。

import numpy as np

def _eqn(y, variables, sign=-1.0):
    f = 0
    for i,x in enumerate(variables):
        f += x/(1+0.06)**(i+1)
    return float(sign*(y + f))


_eqn(1,np.array([1,1]))

def _eqn2(y, vars1, vars2, sign=-1.0):
    f = 0
    n = len(vars1)
    for x,y,i in zip(vars1, vars2, range(n)):
        f += (x+y)/(1+0.06)**(i+1)
    return float(sign*(y + f))


_eqn2(1,np.array([1,1]),np.array([1,1]))

Answer 1

是的，有可能......尽管略有变化

def _eqn2(y, vars1, vars2, sign=-1.0):
    f = 0
    for i, (v1,v2) in enumerate(zip(vars1, vars2)):
        f += (v1+v2)/(1+0.06)**(i+1)
    return float(sign*(y + f))

您枚举zip您的 的元组vars。

尽管由于您正在使用numpy数组，但我确信有更好的方法可以在不使用 pythonfor循环的情况下实现相同的目标。

还冒昧地更改了变量的名称，因为您y在两个不同的上下文中使用并且可能会造成混淆。