NumPy布尔数组警告?

时间:2016-08-03 03:31:10

标签: python arrays numpy boolean bitmask

我有几个numpy数组,比方说abc,并创建了一个mask来应用于所有数组。

我试图掩盖它们:

a = a[mask]

其中maskbool数组。值得注意的是,我已经验证了

len(a) = len(b) = len(c) = len(mask)

我收到了一个相当神秘的警告:

FutureWarning: in the future, boolean array-likes will be handled as a boolean array index

1 个答案:

答案 0 :(得分:16)

False == 0,True == 1.如果你的面具是一个列表,而不是一个ndarray,你会得到一些意想不到的行为:

>>> a = np.array([1,2,3])
>>> mask_list = [True, False, True]
>>> a[mask_list]
__main__:1: FutureWarning: in the future, boolean array-likes will be handled as a boolean array index
array([2, 1, 2])

这个数组由[1],[0]和[1]组成,就像

一样
>>> a[np.array([1,0,1])]
array([2, 1, 2])

另一方面:

>>> mask_array = np.array(mask_list)
>>> mask_array
array([ True, False,  True], dtype=bool)
>>> a[mask_array]
array([1, 3])

警告告诉您,a[mask_list]最终将a[mask_array]#include <locale.h> int main() { setlocale(LC_ALL, ""); // ... } 相同(这可能是您希望它首先给您的。)