数量C-Api阵列_equal

这是的代码 array_equal :

def array_equal(a1, a2):
    try:
        a1, a2 = asarray(a1), asarray(a2)
    except:
        return False
    if a1.shape != a2.shape:
        return False
    return bool(asarray(a1 == a2).all())

正如你所看到的,它不是 c-api 级别函数。在确保两个输入都是数组并且形状匹配之后,它执行一个元素 == 测试,然后 all .

这对于浮动不可靠。整数和布尔都可以。

c-api中可能有某种相等函数,但它的克隆可能不是您所需要的。

PyArray_CountNonzero(PyArrayObject* self)

可能是一个很好的功能。我记得从之前的代码中发现 PyArray_Nonzero 使用它来确定要分配和返回的数组的大小。您可以给它一个对象来比较2个数组的元素(在给定 dtype ),然后测试非零计数。

或者,您可以构造自己的迭代器,一旦它得到一对不相等的元素,就会立即退出。使用 nditer 以获得全阵列广播功率。