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])
其中该数组由 a[1]、a[0] 和 a[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]
(这可能是您最初希望它给您的。)