python - 如何处理 numpy 数组中的 ZeroDivisionError

Question

我有两个大的 numpy 数组，我需要将它们分开。

由于我在 Python 32 位上工作，并且数组太大，为了避免 FloatingPointError，我正在做，例如:

x = numpy.array([...],dtype=object)
y = numpy.array([...].dtype=object)

问题是，在数组 y 中，一些元素可以是 0.0

那么，我的问题是，如何处理计算 x/y 并避免 ZeroDivisionError。

例如，如果 y 中的元素为 0，我想强制执行 nan。因此，对于数组中的特定元素，计算 x/y=nan

Answer 1

使用屏蔽数组，

x = numpy.array([...],dtype=object)
y = numpy.array([...].dtype=object)

x_m = numpy.ma.array(x, mask=(x==0))
y_m = numpy.ma.array(y, mask=(y==0))

print(x_m/y_m)