重载numpy数组的“==”运算符

Question

我正在用Python定义一个需要检查的函数

if a==b:
  do.stuff()

原则上，a和b可以是numpy数组或整数，我希望我的实现对这一点具有健壮性。但是，要检查numpy数组的相等性，需要将布尔值与all（）一起追加，这将在a和b为整数时中断代码

是否有一种简单的方法来编码相等性测试，使它无论a和b是整数还是numpy数组都能工作

Answer 1

这个对数组和整数（数字）都有效的方法怎么样

if np.array_equal(a,b):
    do.stuff()