Python單線程條件返回for循環

Question

我該怎麼做到這一點？

def get_sum(a,b): 
    return sum(for i in range (a,b)) if a != b else a 

我嘗試了幾個不同的東西，但與for循環我無法得到它的工作。

謝謝！

錯誤：

File "", line 2 return sum(for i in range (a,b)) if a != b else a ^SyntaxError: invalid syntax

Answer 1

編輯： OP明確了get_sum功能的意圖，我也相應調整我的答案：

. . . find the sum of all the numbers between including them too and return it. If both numbers are equal return a or b .

---

您代表達應該採取的表格i for i in range(a, b)（注意附加i）：

def get_sum(a, b): 
    return sum(i for i in range(min(a, b), max(a, b) + 1)) 

更短的版本將放棄發電機完全構建，因爲range s爲迭代：

def get_sum(a, b): 
    return sum(range(min(a, b), max(a, b) + 1)) 

然而，我不能告訴你方法的目的到底是什麼。 。 。是get_sum(10, 11)真的應該返回與get_sum(10, 10)相同的結果嗎？

Answer 2

隨着：

return sum(for i in range (a,b)) if a != b else a 

你可能嘗試做以下：

("return sum(range (a,b)) if a != b else a")