How to extract domain from email address with Pandas

Question

I have no idea to extract domain part from email address with pandas. In case if it is '[email protected]' I would like to get 'gmail.com'.

Please give me an idea.

Answer 1

I believe you need split and select second value of lists by indexing:

df = pd.DataFrame({'email':['[email protected]','[email protected]']})

df['domain'] = df['email'].str.split('@').str[1]
#faster solution if no NaNs values 
#df['domain'] = [x.split('@')[1] for x in df['email']]
print (df)
           email     domain
0  [email protected]  gmail.com
1   [email protected]  yahoo.com

Answer 2

This can also be done using lambda function.

df = pd.DataFrame({'email':['[email protected]','[email protected]', '[email protected]']})

df['domain'] = df['email'].apply(lambda x: x.split('@')[1])