Python Pandas DataFrame检查字符串是否为其他字符串并填充列

Question

I am learning Python and this might be a noob question:

import pandas as pd

A = pd.DataFrame({"A":["house", "mouse", "car", "tree"]})

check_list = ["house", "tree"]

I want to check rowwise if the string in A is in check_list. The result should be

       A   YESorNO
0  house       YES
1  mouse        NO
2    car        NO
3   tree       YES

Answer 1

Use numpy.where with isin :

import pandas as pd
import numpy as np

A = pd.DataFrame({"A":["house", "mouse", "car", "tree"]})
check_list = ["house", "tree"]

A['YESorNO'] = np.where(A['A'].isin(check_list),'YES','NO')
print (A)
       A YESorNO
0  house     YES
1  mouse      NO
2    car      NO
3   tree     YES

Answer 2

If for some reason you don't want to import numpy,

import pandas as pd

A = pd.DataFrame({"A":["house", "mouse", "car", "tree"]})
check_list = ["house", "tree"]

Here's the one liner:

A['YESorNO'] = ['YES' if x in check_list else 'NO' for x in A['A']]