我在 codewars.com 做 katas(練習)來練習我在 Python 中的學習。這是卡塔:給定兩個長度為 的整數數組a,創建一個程序,如果 in 每個元素的平方和嚴格大于in 每個元素的平方和,則返回。b>= 1Trueab這是我嘗試過的代碼:def array_madness(a,b): a = sum(i**2 for i in a) b = sum(i**2 for i in b) return True if a > b else False這是我需要通過的測試:test.assert_equals(array_madness([4, 5, 6], [1, 2, 3]),True)test.assert_equals(array_madness( [1, 2, 3],[4, 5, 6]),False)
1 回答

白衣非少年
TA貢獻1155條經驗 獲得超0個贊
您錯誤地復制了問題描述。它是 的平方和a
,以及 的立方和b
(它對兩個數組進行不同的計算)。以下應該有效:
def array_madness(a,b): return sum(i**2 for i in a) > sum(i**3 for i in b)
添加回答
舉報
0/150
提交
取消