减去两个数组,只保留另一个没有数据的值

时间:2014-08-06 12:37:43

标签: python arrays numpy

我一直试图减去(并组合乘法等)两个Numpy数组,这样得到的数组只会在这样的位置留下值,而另一个数组没有数据。

如果我有矩阵a和b,a-b会给c:

a = np.array([0,2,3,0])
b = np.array([1,0,3,0])
c = np.array([0,2,0,0])

我已经尝试将b乘以一个非常大的数字,但后来我无法弄清楚如何摆脱负值。还有一个数组a和b的缺失值为-999。

非常感谢帮助!谢谢!

1 个答案:

答案 0 :(得分:5)

这个怎么样?

>>> a = np.array([0,2,3,0])
>>> b = np.array([1,0,3,0])
>>> a[b!=0] = 0
>>> a
array([0, 2, 0, 0])