如何有效地将布尔numpy数组转换为阈值布尔数组？

2023年12月22日 243次阅读

我正在使用
Python 2.7和NumPy来处理大型布尔值数组.

我有一个数组A,就像这样：

>>> A
array([[[False, False, True, True, True],
        [False, False, False, True, True],
        [False, False, True, True, True],
        [False, False, False, True, True],
        [False, False, False, False, True]],

       [[False, True, True, True, True],
        [False, True, True, True, True],
        [False, False, True, True, True],
        [False, True, True, True, True],
        [False, False, True, True, True]]])

我必须将它转换为这样的布尔数组：

>>> B
array([[[True, False, True, True, True],
        [True, True, False, True, True],
        [True, False, True, True, True],
        [True, True, False, True, True],
        [True, True, True, False, True]],

       [[False, True, True, True, True],
        [False, True, True, True, True],
        [True, False, True, True, True],
        [False, True, True, True, True],
        [True, False, True, True, True]]])

因此,我们的想法是每行的最后一个False值应该保留,任何其他值都应该为True.
我需要创建它,以便将其用作另一个数组的掩码.

有没有办法使用NumPy而不使用for循环(这很慢)？

最佳答案 您也可以使用xor运算符^来实现此目的.只需将数据“leftshift”一个,然后向右添加True值,然后xor new和old数组：

A = np.array([[False, False, True, True, True],
              [False, False, False, True, True],
              [False, False, True, True, True],
              [False, False, False, True, True],
              [False, False, False, False, True]])

X = np.hstack((A[:,1:], 
               np.array(np.ones((A.shape[0], 1)), dtype=np.bool))))
>>> array([[False, True, True, True, True],
           [False, False, True, True, True],
           [False, True, True, True, True],
           [False, False, True, True, True],
           [False, False, False, True, True]])

np.invert(A ^ X)
>>> array([[True, False, True, True, True],
           [True, True, False, True, True],
           [True, False, True, True, True],
           [True, True, False, True, True],
           [True, True, True, False, True]])

仅当剩下所有False值且仅后跟True值时,此方法才有效.

点赞