2015-04-06 65 views
2

我必須定義一個函數,它包含一個字符串列表。推,流行和窺視。正確添加

def digit_stack(commands): 
    stack = [] 
    sums = 0 
    for i in commands: 
     if "PUSH" in i: 
      for n in i: 
       if n.isdigit(): 
        stack.append(int(n)) 
     return stack 

然而

digit_stack("PUSH 3", "PUSH 4") == [3] 

爲什麼它只是附加的第一推?

回答

3

return一次迭代後:

for i in commands: 
    if "PUSH" in i: 
     for n in i: 
      if n.isdigit(): 
       stack.append(int(n)) 
return stack # move outside the loop 
+1

OMG,這是尷尬。謝謝 – Juanvulcano

+0

@Juanvulcano,不用擔心,容易犯錯誤。 –

+1

@Tim Castelijns,歡呼聲。 –