python - 基于特定数据框列处理异常

Question

我有一个数据框如下:

现在我想将 x 除以 y 值而不给我异常(exception)。例如，当我将 3 除以 2 时，它应该得到 1.5，而当我将 3 除以 0 时，它应该得到零。为此，我编写了一个异常函数

def divide(x,y):
    try:
        result = x/y
        print (result)
    except ZeroDivisionError:
        print (0)

我现在想在数据框中创建一个新列并应用此函数。到目前为止我用过:

df['num']  = df.apply(divide)

但这并没有给我所需的结果。有人可以帮忙

Answer 1

你也可以试试:

def divide(df):
    if(df['y']!=0):
        return df['x']/df['y']
    else:
        return 0
df['num']=df.apply(divide,axis=1)

但我建议改用这个:

import numpy as np
df['num']=df['x']/df['y']
df.replace(np.inf,0)