Numpy проверьте, если массив все ноль
>>> all_zeros = not np.any(a)
Obnoxious Oryx
>>> all_zeros = not np.any(a)