Home > Back-end >  How to extract hashtags from Tweets in pandas data frame?
How to extract hashtags from Tweets in pandas data frame?

Time:03-10

I have a dataset of tweets with several variable (columns) and I want to extract all the hashtags from a tweet (text) and place the result in a new column (hashtags). Below is what I am trying:

import pandas as pd
data = pd.read_csv("Sample.csv", lineterminator='\n')

def hashtags(string):
    Hash = data.text.str.findall(r'#.*?(?=\s|$)')
    return Hash
data['hashtags'] = data['text'].apply(lambda x: hashtags(x))

However, when I run the function hashtags, my notebook is just stuck (not finishing execution or giving any error). My file only have around 10k rows.

Also, if this code run successfully, I am expecting to get something like this:

[#asd, #fer, #gtr]

But I want the resultant column should have only name of the hashtags like [asd, fer, gtr]. Please suggest what change I should do in the code.

I tried to look for solution in previous asked questions but most of them used regular expression and I am looking for a solution using pandas.

Thanks in Advance.

CodePudding user response:

I downloaded some sample twitter data in a .csv from here, https://twitter-sentiment-csv.herokuapp.com/. I've used a slice of the first 10 rows for this example.

def find_tags(row_string):
    # use a list comprehension to find list items that start with #
    tags = [x for x in row_string if x.startswith('#')]
    
    return tags

df = pd.DataFrame({'sentiment': {0: 'neutral',
  1: 'neutral',
  2: 'neutral',
  3: 'neutral',
  4: 'neutral',
  5: 'neutral',
  6: 'neutral',
  7: 'positive',
  8: 'neutral',
  9: 'neutral'},
 'text': {0: 'RT @fakeTakeDump: TRAMS STELARA BICYCLE PINOCHLE JUMBO INDEX SEPTAVALENT TYPEWRITER HOMEBREWING AND ANTI-LOCK HULLO KITTY IN FORTUNE COOKIE…',
  1: 'RT @fauzanzain: Hi warga twitter, sekarang aku lagi cari career coach nih yang punya latar belakang tech recruiter / mid to senior digital…',
  2: 'RT @fakeTakeDump: WOODWORKING THE FORUM SHOPS LIKENESS SPECTROHELIOSCOPE CHEEMS FLAVONOIDS ROCKET IS NEITHER SUGAR DADDY CANNED TUNA HANDMA…',
  3: 'WOODWORKING THE FORUM SHOPS LIKENESS SPECTROHELIOSCOPE CHEEMS FLAVONOIDS ROCKET IS NEITHER SUGAR DADDY CANNED TUNA…',
  4: 'RT @KirkDBorne: Recap of 60 days of #DataScience and #MachineLearning — days 1 through 60: by @NainaChaturved8 \\n———…',
  5: 'Recap of 60 days of #DataScience and #MachineLearning — days 1 through 60:  by… ',
  6: 'RT @IBAConservative: @dax_christensen The truth is out! They can’t hold it back. \\n#CrimesAgainstHumanity \\n#TrudeauTyranny \\n#TrudeauMustResi…',
  7: "RT @drmwarner: As per these children's health organizations, keeping masks on in schools 2wks post March break would have made much more se…",
  8: 'RT @cryptotommy88: TL;DR\\n✅ Collective analytics business \\n✅ Draw power from data science & crowd-sourced knowledge\\n✅ 1st product PFPscore:…',
  9: 'RT @cryptotommy88: TL;DR\\n✅ Collective analytics business \\n✅ Draw power from data science & crowd-sourced knowledge\\n✅ 1st product PFPscore:…'},
 'user': {0: 'BotDuran',
  1: 'ezash',
  2: 'BlkHwk0ps',
  3: 'fakeTakeDump',
  4: 'RobotProud',
  5: 'KirkDBorne',
  6: 'cloudcnworld',
  7: 'NeuroTeck',
  8: 'BIGwinCutiejoy8',
  9: 'luckbigw1n'}})

df['split'] = df['text'].str.split(' ')

df['tags'] = df['split'].apply(lambda row : find_tags(row))
# replace # as requested in OP, replace for new lines and \ as needed.
df['tags'] = df['tags'].apply(lambda x : str(x).replace('#', '').replace('\\n', ',').replace('\\', '').replace("'", ""))

Output df['tags']:

0                                []
1                                []
2                                []
3                                []
4    [DataScience, MachineLearning]
5    [DataScience, MachineLearning]
6                                []
7                                []
8                                []
9                                []
Name: tags, dtype: object
  • Related