python - 根据python中的逻辑数组设置数组的值

Question

我有一个数组 a=np.array([ -1, 20, 34, 4, 5, -60, -7, 8]) ，我想用这种方式测试2个条件:p=np.logical_or(a<=5, a>=25) .以这种方式，我有一个逻辑数组 (True-False, 0-1) .现在我想将满足 p 的 True 条件的数组 a 的值设置为预定义的值，例如 NaN。

p
array([ True, False,  True,  True,  True,  True,  True, False], dtype=bool)

我要:

value=array([NaN, 20, NaN, NaN, NaN, NaN, NaN, 8])

我怎样才能做到？

Answer 1

使用numpy.where :

>>> np.where((a<=5) | (a>=25), np.nan, a)
array([ nan,  20.,  nan,  nan,  nan,  nan,  nan,   8.])
>>> p = np.logical_or(a<=5, a>=25)
>>> np.where(p, np.nan, a)
array([ nan,  20.,  nan,  nan,  nan,  nan,  nan,   8.])

或者使用 a 创建一个新的 float 类型的数组，并使用 p 将 np.nan 分配给它的索引:

>>> a_ = a.astype(float)
>>> a_[p] = np.nan
>>> a_
array([ nan,  20.,  nan,  nan,  nan,  nan,  nan,   8.])