我有两个 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 > aTrue生成一个包含/False值的数组。
但是,Python 无法确定具有多个元素的 NumPy 数组是否应该是True或False。array([True, False, True])应该如何评估诸如此类的数组?ValueError由于潜在的歧义,提出了A。
相反,您需要检查是否所有的值b > a都是True. 使用 NumPyall()执行此操作:
if (b > a).all():
print 'True'