numpy配列内のすべての値が等しいかどうかのテスト

Question

a + bの内容で満たされると思われるnumpyの1次元配列cがあります。私はPyOpenCLを使って最初にデバイスでa + bを実行しています。

numpyスライスを使用して、Pythonで結果配列cの正確性を素早く判断したいと思います。

これは私が現在

def python_kernel(a, b, c): 
    temp = a + b 
    if temp[:] != c[:]: 
     print "Error" 
    else: 
     print "Success!" 

を持っているものである。しかし、私はエラーを取得：

ValueError: The truth value of an array with more than one element is ambiguous. Use a.any() or a.all()

をしかし、a.anyかa.allだけの値が0

ないかどうかを判断します思えます

numpy配列tempのすべてのスケーラーがすべての値と等しいかどうかをテストする場合はどうすればよいですか？ numpy配列c？

Answer 1

なぜNumPyの機能のnumpy.array_equal(a1, a2)^[docs]を使用しないのですか？

Answer 2

if np.any(a+b != c):または同等の場合はif np.all(a+b == c):の比較結果にanyを呼び出します。 a+b != cはdtype=boolの配列を作成し、次にanyはその配列を参照して、メンバーがTrueであるかどうかを調べます。それは全体のブール結果の配列を作成する必要はありませんので、

>>> import numpy as np 
>>> a = np.array([1,2,3]) 
>>> b = np.array([4,5,2]) 
>>> c = a+b 
>>> c 
array([5, 7, 5]) # <---- numeric, so any/all not useful 
>>> a+b == c 
array([ True, True, True], dtype=bool) # <---- BOOLEAN result, not numeric 
>>> all(a+b == c) 
True 

すべてのことを言って、しかし、Amber's solutionはおそらく高速です。 np.arrayデータ型が浮動小数点数の場合は、

Answer 3

np.allcloseが適しています。 np.array_equalが正常に動作しない場合があります。例えば：

import numpy as np 
def get_weights_array(n_recs): 
    step = - 0.5/n_recs 
    stop = 0.5 
    return np.arange(1, stop, step) 

a = get_weights_array(5) 
b = np.array([1.0, 0.9, 0.8, 0.7, 0.6]) 

結果：

>>> a 
array([ 1. , 0.9, 0.8, 0.7, 0.6]) 
>>> b 
array([ 1. , 0.9, 0.8, 0.7, 0.6]) 
>>> np.array_equal(a, b) 
False 
>>> np.allclose(a, b) 
True 

>>> import sys 
>>> sys.version 
'2.7.3 (default, Apr 10 2013, 05:13:16) \n[GCC 4.7.2]' 
>>> np.version.version 
'1.6.2'