Numpy > vector > if (n == (vector)0) @ CのNumpy実装 > np.count_nonzero()を使う

動作環境

GeForce GTX 1070 (8GB)
ASRock Z170M Pro4S [Intel Z170chipset]
Ubuntu 16.04.4 LTS desktop amd64
TensorFlow v1.7.0
cuDNN v5.1 for Linux
CUDA v8.0
Python 3.5.2
IPython 6.0.0 -- An enhanced Interactive Python.
gcc (Ubuntu 5.4.0-6ubuntu1~16.04.4) 5.4.0 20160609
GNU bash, version 4.3.48(1)-release (x86_64-pc-linux-gnu)
scipy v0.19.1
geopandas v0.3.0
MATLAB R2017b (Home Edition)
ADDA v.1.3b6
gnustep-gui-runtime v0.24.0-3.1
PyMieScatt v1.7.0

C実装

if (n == (Vector)0)             // triangle is degenerate

これをNumpyでチェックするにはどうするか?

参考

np.count_nonzero()を使うとのこと。

code

test_numpy_array_all_zero_180512.py

import numpy as np

vecU = np.array([3.0, 1.0, 4.0])


def check_all_zero(avec):
    return (np.count_nonzero(avec)==0)


res = check_all_zero(vecU)
print(res)

res = check_all_zero(vecU-vecU)
print(res)

run

$ python3 test_numpy_array_all_zero_180512.py 
False
True