np.where在pandas中灵活应用如excel的if

import pandas as pd
from  numpy import where as iif
d = {
        '客户名称':['A','B','A','B'],
        '产品编码':['x', 'y', 'x','y'],
        '发货数量':[10,20,30,40],
        '发货日期':['20220101','20220331','20220530','20221231'],
        }
df=pd.DataFrame(d)


df['new1'] = iif(df.客户名称=='A',df.发货数量,0)
df['new2'] = iif(df.发货日期>'20220401',df.发货数量,0)

看到很多小伙伴,对numpy.where的应用不是很熟悉.其实这个函数在pandas中,可以当作excel中的if灵活应用在整列的判断和计算中.上面代码很简单,对照excel的功能想想,哈哈,就这么简单,相信聪明的你一看就秒懂😊