我有两个 numpy 数组,我试图找到其中更大的数组(元素明智,即所有元素都应该更大)
import numpy as np
a = np.array([4,5,6])
b = np.array([7,8,9])
if b > a:
print 'True'
但我没有得到所需的输出并出现错误
我有两个 numpy 数组,我试图找到其中更大的数组(元素明智,即所有元素都应该更大)
import numpy as np
a = np.array([4,5,6])
b = np.array([7,8,9])
if b > a:
print 'True'
但我没有得到所需的输出并出现错误
采用np.all()
In [1]: import numpy as np
In [2]: a = np.array([4,5,6])
In [3]: b = np.array([7,8,9])
In [4]: np.all(b > a)
Out[4]: True
if all(b>a):
print 'True'
对于多维数组,请使用:
if np.all(b>a):
print 'True'
但是all()
对于一维数组来说更快,并且如果您的数组非常大,可能会很有用:
>>> timeit('a = np.array([4,5,6]); b = np.array([7,8,9]); all(a>b)',number=100000,setup='import numpy as np')
0.34104180335998535
>>> timeit('a = np.array([4,5,6]); b = np.array([7,8,9]); np.all(a>b)',number=100000,setup='import numpy as np')
0.9201719760894775
b > a
True
生成一个包含/False
值的数组。
但是,Python 无法确定具有多个元素的 NumPy 数组是否应该是True
或False
。array([True, False, True])
应该如何评估诸如此类的数组?ValueError
由于潜在的歧义,提出了A。
相反,您需要检查是否所有的值b > a
都是True
. 使用 NumPyall()
执行此操作:
if (b > a).all():
print 'True'