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))

您枚举了 vars 的 zip 的元组。

尽管您使用的是 numpy 数组，但我确信有更好的方法可以在不使用 python for 循环的情况下实现同样的目标。

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

Python 联合枚举多个变量

Python joint enumerate with multiple variables

python

loops

enumerate