在地图函数中除以 0

Question

我想知道如何在 map 函数中处理除以 0 错误（在 Python 2.7 下）。

不使用map，我得到

def my_func(a, b):
    return a / b

a = pandas.DataFrame([1, 1])
b = pandas.DataFrame([1, 0])

my_func(a, b)
Out[]: 
          0
0  1.000000
1       inf

但我在使用map 时得到不同的结果：

map(my_func, a, b)
Out[]:
File "<ipython-input-12-83b4adbfd8de>", line 3, in func
return a / b
ZeroDivisionError: integer division or modulo by zero

我该如何处理这个错误？

Answer 1

你可以这样试试：

def my_func(a, b):
    if b != 0: return a / b
    else: return np.inf

或通过以下方式捕获警告：

import warnings
warnings.filterwarnings("error")

def my_func(a, b):
    try:
        return a / b
    except:
        return np.inf