Python 如何比较两个数组元素的平方和？

我在做katas（practices）来练习我在Python中的学习。这是卡塔：

给定两个整数数组

a

，

b

，两个数组的长度

>=1

，创建一个程序 如果

a

中每个元素的平方和为 严格大于

b

中每个元素的平方和

这是我尝试的代码：

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)

---

您将问题描述复制错误。它是

a

的平方和，以及

b

的立方体和（它对两个数组进行了不同的计算）。以下方面应起作用：

def array_madness(a,b):
    return sum(i**2 for i in a) > sum(i**3 for i in b)

---

你提到立方体，不是应该是**3吗？@Luv对不起，我修好了。它是方形的。我不确定问题/错误是什么？@lhd我已经发布了我试图克服的挑战的url。我还安排了我必须通过的考试。当您输入我迄今为止所获得的代码时，它返回一个错误，该错误表示：True应该等于false非常感谢。我现在明白了。