条件组合:Numpy数组

3
我试图将几个numpy条件组合在一起,但结果不是我期望的那样。
import numpy as np
ke, ku = np.arange(30, dtype=int), np.arange(50, dtype=int)
KE, KU = np.meshgrid(ke, ku, indexing='ij')

现在,以下内容可以正常工作:
>>> (KU > 0)[:5, :5] 
Out[28]: 
array([[False,  True,  True,  True,  True],
       [False,  True,  True,  True,  True],
       [False,  True,  True,  True,  True],
       [False,  True,  True,  True,  True],
       [False,  True,  True,  True,  True]], dtype=bool)

>>> (KE > 0)[:5, :5] 
Out[29]: 
array([[False, False, False, False, False],
       [ True,  True,  True,  True,  True],
       [ True,  True,  True,  True,  True],
       [ True,  True,  True,  True,  True],
       [ True,  True,  True,  True,  True]], dtype=bool)

>>> ((KE > 0) & (KU > 0))[:5, :5] 
Out[31]: 
array([[False, False, False, False, False],
       [False,  True,  True,  True,  True],
       [False,  True,  True,  True,  True],
       [False,  True,  True,  True,  True],
       [False,  True,  True,  True,  True]], dtype=bool)

然而,在下一个案例中,情况就不太一样了:
>>> (KE + KU > 1)[:5, :5]
Out[24]: 
array([[False, False,  True,  True,  True],
       [False,  True,  True,  True,  True],
       [ True,  True,  True,  True,  True],
       [ True,  True,  True,  True,  True],
       [ True,  True,  True,  True,  True]], dtype=bool)

>>> ((KE + KU > 1) & KE > 0)[:5, :5]
Out[19]: 
array([[False, False, False, False, False],
       [False,  True,  True,  True,  True],
       [False, False, False, False, False],
       [ True,  True,  True,  True,  True],
       [False, False, False, False, False]], dtype=bool)

我预期的结果是什么 - 为什么没有发生?
array([[False, False, False, False, False],
       [False,  True,  True,  True,  True],
       [ True,  True,  True,  True,  True],
       [ True,  True,  True,  True,  True],
       [ True,  True,  True,  True,  True]], dtype=bool)
1个回答

4
这个案例对我来说看起来是正确的:(KE + KU > 1)[:5, :5] 第二种情况只是一个括号问题:
((KE + KU > 1) & (KE > 0))[:5, :5]

网页内容由stack overflow 提供, 点击上面的
可以查看英文原文,
原文链接