0

我有两个 numpy 数组,我试图找到其中更大的数组(元素明智,即所有元素都应该更大)

import numpy as np

a = np.array([4,5,6])
b = np.array([7,8,9])

if b > a:
    print 'True'

但我没有得到所需的输出并出现错误

4

3 回答 3

2

采用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
于 2015-01-31T21:22:33.697 回答
1
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
于 2015-01-31T21:22:17.830 回答
1

b > aTrue生成一个包含/False值的数组。

但是,Python 无法确定具有多个元素的 NumPy 数组是否应该是TrueFalsearray([True, False, True])应该如何评估诸如此类的数组?ValueError由于潜在的歧义,提出了A。

相反,您需要检查是否所有的值b > a都是True. 使用 NumPyall()执行此操作:

if (b > a).all():
    print 'True'
于 2015-01-31T21:26:00.720 回答