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的{{1}}元组。

虽然你正在使用vars数组，但我确信有更好的方法可以在不使用python numpy循环的情况下实现同样的目标。

也可以自由地更改变量的名称，因为你在两个不同的环境中使用了for并且可能会让人感到困惑。

Python联合枚举多个变量

1 个答案: