python - 在 python 中跟踪索引数组时的意外行为

Question

我在使用 while 循环获取与从一个数组到另一个数组的跟踪量相关的索引量时遇到了一些意外行为。

我的输入代码是:

from numpy import array
from numpy.ma import where as mwhere
from numpy.ma import masked_array, masked
start_pts = array([False, True, False, False, False, False, True, False, False, False, True])
active    = array([True, True, True, True, True, True, True, True, True, True, True])
spline_index = array([0, 0, 0, 0, 1, 1, 1, 2, 2, 2, 2])
pt0     = array([1, 0, 3, 0, 6, 4, 0, 8, 9, 10, 0])
m_a_pt0 = masked_array(pt0, start_pts)

for i in range(3):
    # start pt
    pt = mwhere((spline_index[active] == i) & (start_pts[active] == 1))
    print pt
    while pt[0]:
        pt = mwhere((spline_index[active] == i) & (m_a_pt0[active] == pt[0]))
        print i, pt

我期待的输出是:

(array([1]),)
0 (array([0]),)
0 (array([3]),)
0 (array([2]),)
0 (array([], dtype=int64,)
(array([6]),)
1 (array([4]),)
1 (array([5]),)
1 (array([], dtype=int64),)
(array([10]),)
2 (array([9]),)
2 (array([8]),)
2 (array([7]),)
2 (array([], dtype=int64),)

但是当 pt = array([0],) 时 while 循环退出，我得到:

(array([1]),)
0 (array([0]),)
(array([6]),)
1 (array([4]),)
1 (array([5]),)
1 (array([], dtype=int64),)
(array([10]),)
2 (array([9]),)
2 (array([8]),)
2 (array([7]),)
2 (array([], dtype=int64),)

但是如果我在以下位置测试一个值:

test = (array([0]),)
print mwhere((spline_index[active] == 0) & (m_a_pt0[active] == test[0]))

这将返回带有数组的预期元组:

(array([3]),)

为什么当数组 ([0]) 的值仍然存在时我的 while 循环过早退出？

Answer 1

长度为 1 的 Numpy 数组具有其单个元素的真值。参见例如 this question .如果要检查数组是否为空，请检查其长度是否为零。