如何用pandas定义除法函数得到结果？

Question

def set_div(a,b):
    if a==0:
        if b==0:
            return 0
        else:
            return 10**6
    else:
        return b/a

测向：

    a       b
0   0.0     0.0
1   30000.0 0.0
2   0.0     0.0
3   32700.0 10000.0
4   51700.0 0.0

a列和b列的数据类型是float64。你知道吗

df['c'] = df[['a','b']].apply(lambda x:set_div(x[0],x[1])) 

但是它返回KeyError: (0, 'occurred at index a')。你知道吗

我不明白错误，请解释和如何得到我的结果。你知道吗

Answer 1

使用numpy^{}可以直接得到所需的结果，方法如下：

df['c'] = numpy.where(df.a == 0, numpy.where(df.b == 0, 0, 10**6), df.b/df.a)

print(df)

输出为：

       a      b        c
0      0      0  0.00000
1  30000      0  0.00000
2      0      0  0.00000
3  32700  10000  0.30581
4  51700      0  0.00000

如果要使用定义的方法本身，则在调用apply方法时必须添加参数axis=1，以便它按行而不是按列操作。你知道吗

Answer 2

如果要使用自定义函数来处理每一行，需要^{}和axis=1：

df['c'] = df.apply(lambda x :set_div(x['a'],x['b']), axis=1) 

另一个带有^{}的解决方案是矢量化的，因此在大数据帧中有更好的性能：

m1 = df['a'] == 0
m2 = df['b'] == 0
df['c'] = np.select([m1 & ~m2, m1 & m2], [10**6, 0], df['b'] / df['a'])
print (df)
         a        b        c
0      0.0      0.0  0.00000
1  30000.0      0.0  0.00000
2      0.0      0.0  0.00000
3  32700.0  10000.0  0.30581
4  51700.0      0.0  0.00000

性能：

#[50000 rows x 2 columns]
df = pd.concat([df] * 10000, ignore_index=True)

In [113]: %timeit df['c'] = np.select([m1 & ~m2, m1 & m2], [10**6, 0], df['b'] / df['a'])
1.85 ms ± 32.9 µs per loop (mean ± std. dev. of 7 runs, 100 loops each)

In [114]: %timeit df['c1'] = df.apply(lambda x :set_div(x['a'],x['b']), axis=1)
1.01 s ± 17.3 ms per loop (mean ± std. dev. of 7 runs, 1 loop each)