使用pandas的case语句

Question

我需要输入来解决这个用例

    import pandas as pd
    data = {'name': ['A', 'B', 'C', 'D', 'E'], 
            'score': [1,2,3,4,5],
            'status': ['Active','Inactive','Active','Inactive','Active']
            }
    df = pd.DataFrame(data)

    #create a new column (check_flag) set to 1 where status = Active and score in ( 3 and 5) else 0
    df['check_flag']= 

    #keep the records where status = Active and score in ( 3 and 5)

由于

Answer 1

我希望以下解决方案有帮助

df['check_flag'] = ((df['status']== 'Active') & (df['score']>=3))

这将创建一个新列。

Answer 2

谢谢Paras！

我在下面尝试了它也有效

df.loc [（df [＆＃39; status＆＃39;] ==＆＃39; Active＆＃39;）＆amp; （df [＆＃39;得分＆＃39;]。isin（[3,5]）），＆＃39; checkflag＆＃39;] = 1