Python | numpy numpy.ndarray.eq()

简介

numpy.ndarray.__eq__() 方法用于比较两个数组的元素是否相等。

语法

numpy.ndarray.__eq__(y)

参数

• y：数组或标量，用于比较。

返回值

返回一个bool类型的数组，数组的元素对应位置的值为两个数组的对应位置的值是否相等。

示例

import numpy as np

# 例1: 数组比较，结果为布尔类型的数组
a = np.array([1,2,3,4,5])
b = np.array([1,2,3,4,6])
c = np.array([1,2,3,4,5])

print("a == b的结果为: ", a == b)
print("a == c的结果为: ", a == c)

输出结果为:

a == b的结果为:  [ True  True  True  True False]
a == c的结果为:  [ True  True  True  True  True]

在第一个例子中，a 和 b 数组的第五个元素不相等，所以返回 False；在第二个例子中，a 和 c 数组的所有元素都相等，所以返回 True。

# 例2: 数组和标量比较，结果为布尔类型的数组
a = np.array([1,2,3,4,5])
b = 3

print("a == b的结果为: ", a == b)

输出结果为:

a == b的结果为:  [False False  True False False]

在第二个例子中，a 数组的第三个元素等于标量 b，所以返回 True；其余元素和 b 不相等，所以返回 False。

注意事项

• 两个数组必须是维度一样的，如不一样则会提示功能TypeError错误。