Question

我有一个numpy一维数组c，应该填充内容 a + b。我首先使用a + b在设备上执行PyOpenCL。

我想使用c切片在python中快速确定结果数组numpy的正确性。

这就是我目前所拥有的

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

但我收到错误：

ValueError：具有多个元素的数组的真值是不明确的。使用a.any（）或a.all（）

但似乎a.any或a.all只会确定值是否为0.

如果我想测试numpy数组temp中的所有定标器是否等于numpy数组c中的每个值，我该怎么办？< / p>

Answer 1

为什么不直接使用NumPy函数中的numpy.array_equal(a1, a2)^[docs]？

Answer 2

如果np.array数据类型为浮点数，则

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'

Answer 3

您可以在比较结果上致电any：if np.any(a+b != c):或等效if np.all(a+b == c):。 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可能更快，因为它不必创建整个布尔结果数组。

测试numpy数组中的所有值是否相等

3 个答案: