我正在尝试向数据集添加新字段。如果给定列表中有特定字段,则新值应为1;如果另一个列表中存在特定字段,则新值应为2。
这是我的代码:
data = [['ann', 10], ['nick', 15], ['juli', 14]]
final = pd.DataFrame(data, columns = ['index', 'Age'])
final['reward']=0
list1=['ann']
list2=['sam','nick']
def rew(df):
for i in df['value']:
if i in list1:
df['reward']=1
if i in list2:
df['reward']=2
return df['reward']
final['reward']=final.apply (rew,axis=1)
expected result is:
index Age reward
ann 10 1
nick 15 2
juli 14 0
请,建议我可能会出错的地方