使用帶有兩個參數的函數，返回每個參數的平方和

Question

我使用sum_of_squares返回每個參數的平方和。 我為返回參數廣場感到迷茫。

def sum_of_squares(num1,num2):
return sum(int(c) ** 2 for c in str(num1,num2))

Answer 1

str只帶一個參數，您只需刪除它即可使代碼正常工作。

def sum_of_squares(num1, num2):
    return sum(int(c)**2 for c in (num1, num2))

只要將num1和num2塞進一個可迭代的結構中即可。

在這種情況下，更合適的名稱是sum_of_two_squares(num1, num2) 。 使用return num1**2+num2**2對於此方法更整潔。

但是對於更通用的功能sum_of_squares ，我們可以這樣寫：

def sum_of_squares(*args):
        return sum(int(c)**2 for c in args)