Creating new dataframe column using string filter of other column

Question

Below is the dataframe with column name 'Address'. I want to create a separate column 'City' with specific string using filter from Address column.

df1
Serial_No  Address 
1          India Gate Delhi
2          Delhi Redcross Hospital
3          Tolleyganj Bus Stand Kolkata
4          Kolkata Howrah
5          Katra Jammu

Below is the script that I am using

descr = []
col = 'City'
for col in df:
    if np.series(df[col]= df[df[col].str.contains('Delhi ', na=False)]:
        desc = 'Delhi'
    elif np.series(df[col]= df[df[col].str.contains('Kolkata ', na=False)]:
        desc = 'Kolkata'
    else:
        desc = 'None'

Below is the intended output

df1
Serial_No  Address                        City
1          India Gate Delhi               Delhi
2          Delhi Redcross Hospital        Delhi
3          Tolleyganj Bus Stand Kolkata   Kolkata
4          Kolkata Howrah                 Kolkata
5          Katra Jammu                    None

BENY · Accepted Answer · 2021-05-12 04:26:07Z

2

Let us try str.extract

df['new'] = df.Address.str.extract(('(Delhi|Kolkata)'))[0]

answered May 12, 2021 at 4:26

BENY

324k22 gold badges176 silver badges250 bronze badges

Sign up to request clarification or add additional context in comments.

Comments

Vivs · Accepted Answer · 2021-05-12 04:48:16Z

1

Try this

import pandas as pd
df1=pd.DataFrame([[1,'India Gate Delhi'],[2,'Delhi Redcross Hospital'],[3,'Tolleyganj Bus Stand Kolkata'],[4,'Kolkata Howrah'],[5,'Katra Jammu']],columns=['Serial_No','Address'])
print(df1)

def f(df1):
    if 'Delhi' in df1['Address']:
        val = 'Delhi'
    elif 'Kolkata' in df1['Address']:
        val = 'Kolkata'
    else:
        val = 'None'
    return val
df1['City'] = df1.apply(f, axis=1)
print(df1)

answered May 12, 2021 at 4:48

Vivs

4854 silver badges11 bronze badges

Collectives™ on Stack Overflow

Creating new dataframe column using string filter of other column

2 Answers 2

Comments

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

2 Answers 2

Comments

Comments

Your Answer

Sign up or log in

Post as a guest

Related