簡體   English   中英

Python聯合枚舉多個變量

[英]Python joint enumerate with multiple variables

因此,我想同時遍歷兩個列表,並且還需要使用以下代碼進行索引。 兩者對我來說都是直截了當的,但是在第二種情況下是否也可以使用枚舉? 當然,假設兩個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]))

是的,有可能...盡管稍有變化

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

您枚舉varszip的元組。

盡管由於您使用的是numpy數組,但是我敢肯定,有一種更好的方法可以實現相同的目標,而無需使用python for循環。

也改變了變量的名字,因為你usign的自由y在兩個不同的上下文中,可能會造成混亂。

暫無
暫無

聲明:本站的技術帖子網頁,遵循CC BY-SA 4.0協議,如果您需要轉載,請注明本站網址或者原文地址。任何問題請咨詢:yoyou2525@163.com.

 
粵ICP備18138465號  © 2020-2024 STACKOOM.COM