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

Last update on May 02 2025 12:52:22 (UTC/GMT +8 hours)

26. Extract Mentions (@) from Tweets

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

For more Practice: Solve these Related Problems:

Write a Pandas program to extract words beginning with '@' from a tweet column using regex and then output them as a list.
Write a Pandas program to create a new column with all user mentions extracted from a DataFrame column containing tweets.
Write a Pandas program to filter a text column for mentions, then count how many times each user is mentioned.
Write a Pandas program to extract and standardize all '@' mentions from a column and then verify by listing unique mentions.

Go to:

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.

Python Code Editor:

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