python - 条件和的更快和python原生实现（矢量化）

Question

我想将一个 excel SUMIF 公式转换为 python。但是，我使用 Pandas.apply() 实现它的方式会消耗大量行数的时间。有人可以帮助我更好的矢量化实现吗？

def func1(x):
    
    res = data.loc[(( (x['Date'] -data['Date']).dt.days == 1) & (data['ID'] == x['ID'])), ['Stock'].sum()
    return res

data = pd.read_excel(settings['input_file'], sheet_name ="Data", usecols=range(0,10))
data['SUMIF']=data.apply(func1, axis=1)

标签： pythonpandasnumpy