Updated code to print pie charts

Author: the-javapocalypse

main.py

@@ -1,5 +1,7 @@
 import sys,tweepy,csv,re
 from textblob import TextBlob
+import matplotlib.pyplot as plt
+
 
 class SentimentAnalysis:
 
@@ -113,6 +115,7 @@ def DownloadData(self):
  print(str(snegative) + "% people thought it was strongly negative")
  print(str(neutral) + "% people thought it was neutral")
 
+ self.plotPieChart(positive, wpositive, spositive, negative, wnegative, snegative, neutral, searchTerm, NoOfTerms)
 
 
  def cleanTweet(self, tweet):
@@ -121,12 +124,23 @@ def cleanTweet(self, tweet):
 
  # function to calculate percentage
  def percentage(self, part, whole):
- return 100 * float(part) / float(whole)
+ temp = 100 * float(part) / float(whole)
+ return format(temp, '.2f')
 
+ def plotPieChart(self, positive, wpositive, spositive, negative, wnegative, snegative, neutral, searchTerm, noOfSearchTerms):
+ labels = ['Positive [' + str(positive) + '%]', 'Weakly Positive [' + str(wpositive) + '%]','Strongly Positive [' + str(spositive) + '%]', 'Neutral [' + str(neutral) + '%]',
+ 'Negative [' + str(negative) + '%]', 'Weakly Negative [' + str(wnegative) + '%]', 'Strongly Negative [' + str(snegative) + '%]']
+ sizes = [positive, wpositive, spositive, neutral, negative, wnegative, snegative]
+ colors = ['yellowgreen','lightgreen','darkgreen', 'gold', 'red','lightsalmon','darkred']
+ patches, texts = plt.pie(sizes, colors=colors, startangle=90)
+ plt.legend(patches, labels, loc="best")
+ plt.title('How people are reacting on ' + searchTerm + ' by analyzing ' + str(noOfSearchTerms) + ' Tweets.')
+ plt.axis('equal')
+ plt.tight_layout()
+ plt.show()
 
 
 
 if __name__== "__main__":
  sa = SentimentAnalysis()
  sa.DownloadData()
-