python-3.x - numpy 中的这个操作叫什么？

Question

我一直在浏览 numpy 文档以寻找特定的操作。我会为此使用的词是“覆盖”或“掩码”，但这些词的 NumPy 概念似乎与我的不匹配。

我想取两个数组，一个密集，一个稀疏，然后将它们组合起来:

  [ 1, 2,  3, 4,  5 ] 
X [ N, N, 10, N, 12 ] 
= [ 1, 2, 10, 4, 12 ]

其中 X 是操作，N 是 None、Null、-1 或其他一些特殊字符。

这在 numpy/python3 中是如何实现的？

Answer 1

你可以使用np.where:

# pick special value 
N = -1

dns = [ 1, 2,  3, 4,  5 ] 
sprs = [ N, N, 10, N, 12 ] 

# this is important otherwise the comparison below
# is not done element by element
sprs = np.array(sprs)

# tada!    
np.where(sprs==N,dns,sprs)
# array([ 1,  2, 10,  4, 12])

当使用三个参数调用时 m,a,b where “混合” a 和 b 获取元素从 a 开始，其中 m 为 True，从 b 开始，其中 False。