w3resource

Pandas: Extract word mention someone in tweets using @ from the specified column of a given DataFrame

Pandas: String and Regular Expression Exercise-26 with Solution

Write a Pandas program to extract word mention someone in tweets using @ from the specified column of a given DataFrame.

Sample Solution:

Python Code :

import pandas as pd
import re as re
pd.set_option('display.max_columns', 10)
df = pd.DataFrame({
    'tweets': ['@Obama says goodbye','Retweets for @cash','A political endorsement in @Indonesia', '1 dog = many #retweets', 'Just a simple #egg']
    })
print("Original DataFrame:")
print(df)
def find_at_word(text):
    word=re.findall(r'(?<=@)\w+',text)
    return " ".join(word)

df['at_word']=df['tweets'].apply(lambda x: find_at_word(x))
print("\Extracting @word from dataframe columns:")
print(df)

Sample Output:

Original DataFrame:
                                  tweets
0                    @Obama says goodbye
1                     Retweets for @cash
2  A political endorsement in @Indonesia
3                 1 dog = many #retweets
4                     Just a simple #egg
\Extracting @word from dataframe columns:
                                  tweets    at_word
0                    @Obama says goodbye      Obama
1                     Retweets for @cash       cash
2  A political endorsement in @Indonesia  Indonesia
3                 1 dog = many #retweets           
4                     Just a simple #egg

Python Code Editor:

Have another way to solve this solution? Contribute your code (and comments) through Disqus.

Previous: Write a Pandas program to extract hash attached word from twitter text from the specified column of a given DataFrame.
Next: Write a Pandas program to extract only number from the specified column of a given DataFrame.

What is the difficulty level of this exercise?

Test your Programming skills with w3resource's quiz.



Become a Patron!

Follow us on Facebook and Twitter for latest update.

It will be nice if you may share this link in any developer community or anywhere else, from where other developers may find this content. Thanks.

https://www.w3resource.com/python-exercises/pandas/string/python-pandas-string-exercise-26.php