notebook.py

#!/usr/bin/env python
# coding: utf-8

# ## 1. Local and global thought patterns
# <p>While we might not be Twitter fans, we have to admit that it has a huge influence on the world (who doesn't know about Trump's tweets). Twitter data is not only gold in terms of insights, but <strong><em>Twitter-storms are available for analysis in near real-time</em></strong>. This means we can learn about the big waves of thoughts and moods around the world as they arise. </p>
# <p>As any place filled with riches, Twitter has <em>security guards</em> blocking us from laying our hands on the data right away ⛔️ Some  authentication steps (really straightforward) are needed to call their APIs for data collection. Since our goal today is learning to extract insights from data, we have already gotten a green-pass from security ✅ Our data is ready for usage in the datasets folder — we can concentrate on the fun part! 🕵️‍♀️🌎
# <br>
# <br>
# <img src="https://assets.datacamp.com/production/project_760/img/tweets_influence.png" style="width: 300px">
# <hr>
# <br>Twitter provides both global and local trends. Let's load and inspect data for topics that were hot worldwide (WW) and in the United States (US) at the moment of query  — snapshot of JSON response from the call to Twitter's <i>GET trends/place</i> API.</p>
# <p><i><b>Note</b>: <a href="https://developer.twitter.com/en/docs/trends/trends-for-location/api-reference/get-trends-place.html">Here</a> is the documentation for this call, and <a href="https://developer.twitter.com/en/docs/api-reference-index.html">here</a> a full overview on Twitter's APIs.</i></p>

# In[81]:


# Loading json module
import json

# Load WW_trends and US_trends data into the the given variables respectively
WW_trends = json.loads(open('datasets/WWTrends.json').read())
US_trends = json.loads(open('datasets/USTrends.json').read())

# Inspecting data by printing out WW_trends and US_trends variables
print('WW_trends:'),print('\n')
print(WW_trends)

print('\n'),print('='*50),print('\n')

print('US_trends:'),print('\n')
print(US_trends)


# In[82]:


get_ipython().run_cell_magic('nose', '', '\n# One or more tests of the student\'s code\n# The @solution should pass the tests\n# The purpose of the tests is to try to catch common errors and\n# to give the student a hint on how to resolve these errors\n\ndef test_pandas_loaded():\n    assert \'json\' in globals(), \\\n    \'Did you import the json module?\'\n    \ndef test_WW_trends_correctly_loaded():\n    correct_ww_trends = json.loads(open(\'datasets/WWTrends.json\').read())\n    assert correct_ww_trends == WW_trends, "The variable WW_trends should contain the data in WWTrends.json."\n\ndef test_US_trends_correctly_loaded():\n    correct_us_trends = json.loads(open(\'datasets/USTrends.json\').read())\n    assert correct_us_trends == US_trends, "The variable WW_trends should contain the data in USTrends.json."')


# ## 2. Prettifying the output
# <p>Our data was hard to read! Luckily, we can resort to the <i>json.dumps()</i> method to have it formatted as a pretty JSON string.</p>

# In[83]:


# Pretty-printing the results. First WW and then US trends.

print("WW trends:")
print(json.dumps(WW_trends,indent=1))

print("\n","\n", "US trends:")
print(json.dumps(US_trends,indent=1))


# In[84]:


get_ipython().run_cell_magic('nose', '', '# %%nose needs to be included at the beginning of every @tests cell\n\n# Not sure what to check here\n\n# One or more tests of the students code. \n# The @solution should pass the tests.\n# The purpose of the tests is to try to catch common errors and to \n# give the student a hint on how to resolve these errors.\n\ndef strip_comment_lines(cell_input):\n    """Returns cell input string with comment lines removed."""\n    return \'\\n\'.join(line for line in cell_input.splitlines() if not line.startswith(\'#\'))\n\nlast_input = strip_comment_lines(In[-2])\n\ndef test_info_command():\n    assert \'json.dumps(WW_trends\' in last_input, \\\n        "We expected the json.dumps method with the correct input object."\n    assert \'json.dumps(US_trends\' in last_input, \\\n         "We expected the json.dumps method with the correct input object."')


# ## 3.  Finding common trends
# <p>🕵️‍♀️ From the pretty-printed results (output of the previous task), we can observe that:</p>
# <ul>
# <li><p>We have an array of trend objects having: the name of the trending topic, the query parameter that can be used to search for the topic on Twitter-Search, the search URL and the volume of tweets for the last 24 hours, if available. (The trends get updated every 5 mins.)</p></li>
# <li><p>At query time <b><i>#BeratKandili, #GoodFriday</i></b> and <b><i>#WeLoveTheEarth</i></b> were trending WW.</p></li>
# <li><p><i>"tweet_volume"</i> tell us that <i>#WeLoveTheEarth</i> was the most popular among the three.</p></li>
# <li><p>Results are not sorted by <i>"tweet_volume"</i>. </p></li>
# <li><p>There are some trends which are unique to the US.</p></li>
# </ul>
# <hr>
# <p>It’s easy to skim through the two sets of trends and spot common trends, but let's not do "manual" work. We can use Python’s <strong>set</strong> data structure to find common trends — we can iterate through the two trends objects, cast the lists of names to sets, and call the intersection method to get the common names between the two sets.</p>

# In[85]:


# Extracting all the WW trend names from WW_trends
world_trends = set([trend['name'] for trend in WW_trends[0]['trends']])

# Extracting all the US trend names from US_trends
us_trends = set([trend['name'] for trend in US_trends[0]['trends']]) 

# Getting the intersection of the two sets of trends
common_trends = world_trends.intersection(us_trends)

# Inspecting the data
print(world_trends, "\n")
print(us_trends, "\n")
print (len(common_trends), "common trends:", common_trends)


# In[86]:


get_ipython().run_cell_magic('nose', '', "\n# One or more tests of the students code. \n# The @solution should pass the tests.\n# The purpose of the tests is to try to catch common errors and to \n# give the student a hint on how to resolve these errors.\n\ndef test_ww_trends():\n    correct_world_trends = set([trend['name'] for trend in WW_trends[0]['trends']])\n    assert world_trends == correct_world_trends, \\\n    'The variable world_trends does not have the expected trend names.'\n   \ndef test_us_trends():\n    correct_us_trends = set([trend['name'] for trend in US_trends[0]['trends']])\n    assert us_trends == correct_us_trends, \\\n    'The variable us_trends does not have the expected trend names.'\n\ndef test_common_trends():\n    correct_common_trends = world_trends.intersection(us_trends)\n    assert common_trends == correct_common_trends, \\\n    'The variable common_trends does not have the expected common trend names.'")


# ## 4. Exploring the hot trend
# <p>🕵️‍♀️ From the intersection (last output) we can see that, out of the two sets of trends (each of size 50), we have 11 overlapping topics. In particular, there is one common trend that sounds very interesting: <i><b>#WeLoveTheEarth</b></i> — so good to see that <em>Twitteratis</em> are unanimously talking about loving Mother Earth! 💚 </p>
# <p><i><b>Note</b>: We could have had no overlap or a much higher overlap; when we did the query for getting the trends, people in the US could have been on fire obout topics only relevant to them.</i>
# <br>
# <img src="https://assets.datacamp.com/production/project_760/img/earth.jpg" style="width: 500px"></p>
# <div style="text-align: center;"><i>Image Source:Official Music Video Cover: https://welovetheearth.org/video/</i></div>
# <hr>
# <p>We have found a hot-trend, #WeLoveTheEarth. Now let's see what story it is screaming to tell us! <br>
# If we query Twitter's search API with this hashtag as query parameter, we get back actual tweets related to it. We have the response from the search API stored in the datasets folder as <i>'WeLoveTheEarth.json'</i>. So let's load this dataset and do a deep dive in this trend.</p>

# In[87]:


# Loading the data
tweets = json.loads(open('datasets/WeLoveTheEarth.json').read())

# Inspecting some tweets
tweets[0:2]


# In[88]:


get_ipython().run_cell_magic('nose', '', '# %%nose needs to be included at the beginning of every @tests cell\n\n# One or more tests of the student\'s code\n# The @solution should pass the tests\n# The purpose of the tests is to try to catch common errors and\n# to give the student a hint on how to resolve these errors\n\ndef test_tweets_loaded_correctly():\n    correct_tweets_data = json.loads(open(\'datasets/WeLoveTheEarth.json\').read())\n    assert correct_tweets_data == tweets, "The variable tweets should contain the data in WeLoveTheEarth.json."\n    ')


# ## 5. Digging deeper
# <p>🕵️‍♀️ Printing the first two tweet items makes us realize that there’s a lot more to a tweet than what we normally think of as a tweet — there is a lot more than just a short text!</p>
# <hr>
# <p>But hey, let's not get overwhemled by all the information in a tweet object! Let's focus on a few interesting fields and see if we can find any hidden insights there. </p>

# In[89]:


# Extracting the text of all the tweets from the tweet object
texts = [tweet['text'] for tweet in tweets]

# Extracting screen names of users tweeting about #WeLoveTheEarth
names = [user_mention['screen_name'] for tweet in tweets for user_mention in tweet['entities']['user_mentions']]

# Extracting all the hashtags being used when talking about this topic
hashtags = [hashtag['text'] for tweet in tweets for hashtag in tweet['entities']['hashtags']]

# Inspecting the first 10 results
print (json.dumps(texts[0:10], indent=1),"\n")
print (json.dumps(names[0:10], indent=1),"\n")
print (json.dumps(hashtags[0:10], indent=1),"\n")


# In[90]:


get_ipython().run_cell_magic('nose', '', "# %%nose needs to be included at the beginning of every @tests cell\n\n# One or more tests of the student's code\n# The @solution should pass the tests\n# The purpose of the tests is to try to catch common errors and\n# to give the student a hint on how to resolve these errors\n\ndef test_extracted_texts():\n    correct_text = [tweet['text'] for tweet in tweets ]\n    assert texts == correct_text, \\\n    'The variable texts does not have the expected text data.'\n   \ndef test_extracted_names():\n    correct_names = [user_mention['screen_name'] \n                         for tweet in tweets \n                             for user_mention in tweet['entities']['user_mentions']]\n    assert correct_names == names, \\\n    'The variable names does not have the expected user names.'\n    \ndef test_extracted_hashtags():\n    correct_hashtags = [hashtag['text'] \n                            for tweet in tweets\n                                 for hashtag in tweet['entities']['hashtags']]\n    assert correct_hashtags == hashtags, \\\n    'The variable hashtags does not have the expected hashtag data.'")


# ## 6. Frequency analysis
# <p>🕵️‍♀️ Just from the first few results of the last extraction, we can deduce that:</p>
# <ul>
# <li>We are talking about a song about loving the Earth.</li>
# <li>A lot of big artists are the forces behind this Twitter wave, especially Lil Dicky.</li>
# <li>Ed Sheeran was some cute koala in the song — "EdSheeranTheKoala" hashtag! 🐨</li>
# </ul>
# <hr>
# <p>Observing the first 10 items of the interesting fields gave us a sense of the data. We can now take a closer look by doing a simple, but very useful, exercise — computing frequency distributions. Starting simple with frequencies is generally a good approach; it helps in getting ideas about how to proceed further.</p>

# In[91]:


# Importing modules
from collections import Counter

# Counting occcurrences/ getting frequency dist of all names and hashtags
for item in [names, hashtags]:
    c = Counter(item) 
    # Inspecting the 10 most common items in c
    print (c.most_common(10), "\n")


# In[92]:


get_ipython().run_cell_magic('nose', '', '# %%nose needs to be included at the beginning of every @tests cell\n\n# One or more tests of the student\'s code\n# The @solution should pass the tests\n# The purpose of the tests is to try to catch common errors and\n# to give the student a hint on how to resolve these errors\n\ndef test_counter():\n    for item in [names, hashtags]: correct_counter = Counter(item)  \n    assert c == correct_counter, \\\n    "The variable c does not have the expected values."')


# ## 7. Activity around the trend
# <p>🕵️‍♀️ Based on the last frequency distributions we can further build-up on our deductions:</p>
# <ul>
# <li>We can more safely say that this was a music video about Earth (hashtag 'EarthMusicVideo') by Lil Dicky. </li>
# <li>DiCaprio is not a music artist, but he was involved as well <em>(Leo is an environmentalist so not a surprise to see his name pop up here)</em>. </li>
# <li>We can also say that the video was released on a Friday; very likely on April 19th. </li>
# </ul>
# <p><em>We have been able to extract so many insights. Quite powerful, isn't it?!</em></p>
# <hr>
# <p>Let's further analyze the data to find patterns in the activity around the tweets — <b>did all retweets occur around a particular tweet? </b><br></p>
# <p>If a tweet has been retweeted, the <i>'retweeted_status'</i>  field gives many interesting details about the original tweet itself and its author. </p>
# <p>We can measure a tweet's popularity by analyzing the <b><i>retweet<em>count</em></i></b> and <b><i>favoritecount</i></b> fields. But let's also extract the number of followers of the tweeter  —  we have a lot of celebs in the picture, so <b>can we tell if their advocating for #WeLoveTheEarth influenced a significant proportion of their followers?</b></p>
# <hr>
# <p><i><b>Note</b>: The retweet_count gives us the total number of times the original tweet was retweeted. It should be the same in both the original tweet and all the next retweets. Tinkering around with some sample tweets and the official documentaiton are the way to get your head around the mnay fields.</i></p>

# In[93]:


# Extracting useful information from retweets
retweets = [(tweet['retweet_count'], tweet['retweeted_status']['favorite_count'], tweet['retweeted_status']['user']['followers_count'], tweet['retweeted_status']['user']['screen_name'], tweet['text']) for tweet in tweets if 'retweeted_status' in tweet]


# In[94]:


get_ipython().run_cell_magic('nose', '', '# %%nose needs to be included at the beginning of every @tests cell\n\n# One or more tests of the student\'s code\n# The @solution should pass the tests\n# The purpose of the tests is to try to catch common errors and\n# to give the student a hint on how to resolve these errors\n\ndef test_retweets():\n    correct_retweets = [\n             (tweet[\'retweet_count\'], \n              tweet[\'retweeted_status\'][\'favorite_count\'],\n              tweet[\'retweeted_status\'][\'user\'][\'followers_count\'],\n              tweet[\'retweeted_status\'][\'user\'][\'screen_name\'],\n              tweet[\'text\']) \n            \n            for tweet in tweets \n                if \'retweeted_status\' in tweet\n           ]\n    assert correct_retweets == retweets, \\\n    "The retweets variable does not have the expected values. Check the names of the extracted field and their order."')


# ## 8. A table that speaks a 1000 words
# <p>Let's manipulate the data further and visualize it in a better and richer way — <em>"looks matter!"</em></p>

# In[95]:


# Importing modules
import matplotlib.pyplot as plt
import pandas as pd

# Create a DataFrame and visualize the data in a pretty and insightful format
df = pd.DataFrame(data=retweets,columns=['Retweets','Favorites', 'Followers', 'ScreenName', 'Text']).groupby(['ScreenName','Text','Followers']).sum().sort_values(by=['Followers'],ascending=False)

df.style.background_gradient()


# In[96]:


get_ipython().run_cell_magic('nose', '', '# %%nose needs to be included at the beginning of every @tests cell\n\n# One or more tests of the student\'s code\n# The @solution should pass the tests\n# The purpose of the tests is to try to catch common errors and\n# to give the student a hint on how to resolve these errors\n    \ndef strip_comment_lines(cell_input):\n    """Returns cell input string with comment lines removed."""\n    return \'\\n\'.join(line for line in cell_input.splitlines() if not line.startswith(\'#\'))\n\nlast_input = strip_comment_lines(In[-2])\n\n\ndef test_df_creation_command():\n    assert \'retweets\' in last_input, \\\n        "The input data for DataFrame creation is not as expected."\n    assert \'columns=\' in last_input, \\\n        "The columns parameter is missing."\n    assert \'groupby\' in last_input, \\\n        "The groupby method is missing."\n    assert \'sum()\' in last_input, \\\n        "The sum method is missing."\n    assert \'sort_values\' in last_input, \\\n        "The sort_values method is missing."\n    assert \'ascending\' in last_input, \\\n        "The ascending parameter is missing."\n    \ndef test_dataframe():\n    correct_dataframe = pd.DataFrame(retweets, columns=[\'Retweets\',\'Favorites\', \'Followers\', \'ScreenName\', \'Text\']).groupby(\n    [\'ScreenName\',\'Text\',\'Followers\']).sum().sort_values(by=[\'Followers\'], ascending=False)\n    assert correct_dataframe.equals(df), \\\n    "The created dataframe does not match the expected dataframe."')


# ## 9. Analyzing used languages
# <p>🕵️‍♀️ Our table tells us that:</p>
# <ul>
# <li>Lil Dicky's followers reacted the most — 42.4% of his followers liked his first tweet. </li>
# <li>Even if celebrities like Katy Perry and Ellen have a huuge Twitter following, their followers hardly reacted, e.g., only 0.0098% of Katy's followers liked her tweet. </li>
# <li>While Leo got the most likes and retweets in terms of counts, his first tweet was only liked by 2.19% of his followers. </li>
# </ul>
# <p>The large differences in reactions could be explained by the fact that this was Lil Dicky's music video. Leo still got more traction than Katy or Ellen because he played some major role in this initiative.</p>
# <hr>
# <p>Can we find some more interesting patterns in the data? From the text of the tweets, we could spot different languages, so let's create a frequency distribution for the languages.</p>

# In[97]:


# Extracting language for each tweet and appending it to the list of languages
tweets_languages = []
for tweet in tweets: 
    tweets_languages.append(tweet['lang'])

# Plotting the distribution of languages
get_ipython().run_line_magic('matplotlib', 'inline')
plt.hist(tweets_languages)


# In[98]:


get_ipython().run_cell_magic('nose', '', '# %%nose needs to be included at the beginning of every @tests cell\n\n# One or more tests of the student\'s code\n# The @solution should pass the tests\n# The purpose of the tests is to try to catch common errors and\n# to give the student a hint on how to resolve these errors\n\ndef test_tweet_languages():\n    correct_tweet_languages = []\n    for tweet in tweets: correct_tweet_languages.append(tweet[\'lang\'])\n    assert correct_tweet_languages == tweets_languages, \\\n    "The tweets_languages variable does not have the expected values."\n    \nlast_value = _\n\ndef test_plot_exists():\n    assert type(last_value) == type(plt.hist(tweets_languages)), \\\n    \'A plot was not the last output of the code cell.\'\n\ndef strip_comment_lines(cell_input):\n    """Returns cell input string with comment lines removed."""\n    return \'\\n\'.join(line for line in cell_input.splitlines() if not line.startswith(\'#\'))\n\nlast_input = strip_comment_lines(In[-2])\n\ndef test_plot_command():\n    assert \'plt.hist(tweets_languages)\' in last_input, \\\n        "We expected the plt.hist() method in your input."   ')


# ## 10. Final thoughts
# <p>🕵️‍♀️ The last histogram tells us that:</p>
# <ul>
# <li>Most of the tweets were in English.</li>
# <li>Polish, Italian and Spanish were the next runner-ups. </li>
# <li>There were a lot of tweets with a language alien to Twitter (lang = 'und'). </li>
# </ul>
# <p>Why is this sort of information useful? Because it can allow us to get an understanding of the "category" of people interested in this topic (clustering). We could also analyze the device type used by the Twitteratis, <code>tweet['source']</code>, to answer questions like, <strong>"Does owning an Apple compared to Andorid influences people's propensity towards this trend?"</strong>. I will leave that as a <strong>further exercise</strong> for you!</p>
# <p><img src="https://assets.datacamp.com/production/project_760/img/languages_world_map.png" style="width: 500px"></p>
# <hr>
# <p><span style="color:#41859e">
# What an exciting journey it has been! We started almost clueless, and here we are.. rich in insights. </span></p>
# <p><span style="color:#41859e">
# From location based comparisons to analyzing the activity around a tweet to finding patterns from languages and devices, we have covered a lot today — let's give ourselves <b>a well-deserved pat on the back!</b> ✋
# </span>
# <br><br></p>
# <div style="text-align: center;color:#41859e"><b><i>Magic Formula = Data + Python + Creativity + Curiosity</i></b></div>
# <p><img src="https://assets.datacamp.com/production/project_760/img/finish_line.jpg" style="width: 500px"></p>

# In[99]:


# Congratulations!
print("High Five!!!")


# In[100]:


get_ipython().run_cell_magic('nose', '', '# %%nose needs to be included at the beginning of every @tests cell\n\ndef test_nothing_task_10():\n    assert True, "Nothing to test."')