python之添加可能包含 'None' 条目的数组

我有一个关于添加 numpy 数组的问题。 假设我已经定义了一个函数

def foo(a,b): 
    return a+b 

接受两个相同形状的数组并简单地返回它们的和。 现在，我必须处理某些条目可能为 None 的情况。 我想处理那些对应于 float(0) 的条目，这样

[1.0,None,2.0] + [1.0,2.0,2.0]  

加起来

[2.0,2.0,4.0] 

您能为我提供一个已经实现的解决方案吗？

TIA

请您参考如下方法:

我建议 numpy.nan_to_num :

>>> np.nan_to_num(np.array([1.0,None,2.0], dtype=np.float)) 
array([ 1.,  0.,  2.]) 

然后，

>>> def foo(a,b): 
...         return np.nan_to_num(a) + np.nan_to_num(b) 
... 
>>> foo(np.array([1.0,None,2.0], dtype=np.float), np.array([1.0,2.0,2.0], dtype=np.float)) 
array([ 2.,  0.,  4.])