想不通这个卡塔

Question

我正在 CodeWars 中处理这个 Kata，我通过了除一项测试外的所有测试，但它没有说明错误是什么。描述说要检查两个列表的多重性。列表 a2 是列表 a1 的正方形列表。我需要在 a1 中检查 a2 中所有项目的平方根，如果数组为空、None 或平方根不在 a1 中，则返回 false。

我希望得到您可能有的任何提示，这些提示不仅涉及功能（使其更加 Pythonic），还涉及错误处理。我错过了什么？为什么不通过？提前致谢。

def comp(a1, a2):
    if a1 == [] or a2 == []:
        return False
    if a1 == None or a2 == None:
        return False
    pos_count = 0
    neg_count = 0
    for num in set(a2):
        if num**0.5 in set(a1):
            pos_count += 1
        else:
            neg_count += 1
    if neg_count > 0:
        return False
    else:
        return True

Answer 1

您可以改进作为评论发布的代码：

def comp(a1, a2):
    if a1 is None or a2 is None:
        return False
    num_squared = []
    for num1 in a1:
        num_squared.append(num1**2)
    if set(a2) == set(num_squared):
        return True
    else:
        return False

您可以使用理解表达式构建序列。您可以通过将正方形附加到一个空列表来构建一个列表，但您可以这样做：

num_squared = [num ** 2 for num in a1]

您创建一个临时的set 对象set(num_squared)。但是，您可以以与列表相同的方式构建集合：

num_squared = {num ** 2 for num in a1}

如果你有一个布尔表达式然后返回 True 否则 False 你可以用表达式本身替换它：

return set(a2) == set(num_squared)

应用所有这些更改，您可以获得：

def comp(a1, a2):
    if a1 is None or a2 is None:
        return False
    else:
        return set(a2) == {num ** 2 for num in a1}

编辑：请注意：

空列表在布尔表达式中计算为False：

>>> bool([])
False

另外，None 评估 False：

>>> bool(None)
False