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

5q4ezhmt 于 2023-11-18 发布在其他

关注(0)|答案(1)|浏览(169)

我尝试使用一个函数来检查两个向量是否相等。我不知道我是否使用了正确的函数，因为我没有得到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语句。我写的函数正确吗？或者我应该调整什么？

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

型