Python Pandas IF ELSE

Question

将感谢对此的任何帮助，试图做我的第一个Pandas IF ELSE语句，但我正在努力与语法…

if g['Operator'] == 100151:    
g['floor']=g['y'].mean() 
elif g['Operator'] == 20137: 
g['floor']=g['y'].mean() 
elif g['Operator'] == 152: 
g['floor']=g['y'].mean() 
else: 
g['floor']=g['y'].mean()/2

谢谢Gav

Answer 1

你可以这样处理:

import numpy as np
g['floor'] = np.where(g['Operator'].isin([100151, 20137, 152]), g['y'].mean(), g['y'].mean() / 2)

检查文档的(基本矢量化if - else),和型号