如何使用函数检查两个向量是否相等？

Question

我正在尝试使用函数检查两个向量是否相等。我不知道我是否使用了正确的函数，因为我没有得到 true 或 false 作为返回。这是我的代码：

import numpy as np

x=np.array([1,2,3,4])

y=np.array([1,2,3,4])

def check(x,y):

    if x == y:
        print("They are equal")

当我运行代码时，它不会返回任何内容，因此我假设它没有运行 if 语句。我写的函数正确吗还是应该调整什么？

Answer 1

要检查 NumPy 数组是否相等，可以使用

np.array_equal

。最好练习使用

return

来执行函数，而不是打印结果。

def check(x,y):
    if np.array_equal(x,y):
        return "They are equal"
    return "Not equal"

执行：

print(check(x,y))
# They are equal