如果条件为真,则为每一行设置新字段

时间:2019-06-04 17:02:36

标签: python pandas

我正在尝试向数据集添加新字段。如果给定列表中有特定字段,则新值应为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

请,建议我可能会出错的地方

0 个答案:

没有答案