Home > front end >  Dataframe column match with list and create new columns
Dataframe column match with list and create new columns

Time:02-09

I have data frame DF column A , check how much List a=["enter", 'Enter', 'Click', 'click'] available in data frame column, and create another column B on the same names. i want to my data frame to look like this :

DF=       A
       "enter Username"
       "click UserBox"
       "Enter Password"
       "Click login"
       "click checkbox"
       "Enter Phonenumber"

OUTPUT :

DF=       A                         B
       "enter Username"             enter
       "click UserBox"              click
       "Enter Password"             Enter
       "Click login"                Click
       "click checkbox"             click
       "Enter Phonenumber"          Enter 

CodePudding user response:

Use Series.apply with set intersection using operator &:

In [1003]: s = set(a)
In [994]: df['B'] = df['A'].apply(lambda x: (set(x.split()) & s).pop() if len((set(x.split()) & s)) > 0 else '')

In [995]: df
Out[995]: 
                   A      B
0     enter Username  enter
1      click UserBox  click
2     Enter Password  Enter
3        Click login  Click
4     click checkbox  click
5  Enter Phonenumber  Enter
  •  Tags:  
  • Related