Advertisement
Guest User

seaborn jupyter shit

a guest
Sep 30th, 2017
91
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
Python 1.92 KB | None | 0 0
  1. import csv
  2. import os.path
  3. import matplotlib.pyplot as plt
  4. import matplotlib.patches as patches
  5. import seaborn as sns
  6. %matplotlib inline
  7.  
  8. #data loading
  9. data = ....
  10. ------------------------------
  11.  
  12. def visualizeComedyByOtherGenres(data):
  13.     keysToNotInclude = ['Id', 'Duration', 'Language', 'Country', 'Rating', 'Target', "Poster"]
  14.     keys = [key for key in data[0].keys() if not key in keysToNotInclude]
  15.     isComedyKey = 'Target'
  16.     plotValues = dict.fromkeys(keys, 0)
  17.     for item in data:
  18.         if item[isComedyKey] == "1":
  19.             for key in keys:
  20.                 if item[key] == "1":
  21.                     plotValues[key] += 1
  22.     xValues = []
  23.     yValues = []
  24.     for key, value in plotValues.items():
  25.         xValues.append(key)
  26.         yValues.append(value)
  27.     sns.set_context("notebook", font_scale=10)
  28.     gr = sns.barplot(x=xValues, y=yValues)
  29.     plt.setp(gr.get_xticklabels(), rotation=90)
  30.     plt.title("By Genres")
  31.     plt.rcParams['figure.figsize']=(70,40)
  32.  
  33. visualizeComedyByOtherGenres(data)    
  34.  
  35. ----------------------------------
  36.  
  37. def visualizeByLanguage(data):
  38.     languageKey = "Language"
  39.     langToNumComedies = {}
  40.     for item in data:
  41.         language = item[languageKey]
  42.         numMovies, numComedies = langToNumComedies.get(language, (0,0))
  43.         numMovies += 1
  44.         if(item['Target'] == "1"):
  45.             numComedies += 1
  46.         langToNumComedies[language] = (numMovies, numComedies)
  47.    
  48.     xValues = list(langToNumComedies.keys())
  49.     yValues = []
  50.     for key in xValues:
  51.         numMovies, numComedies = langToNumComedies[key]
  52.         yValues.append(numComedies/numMovies)
  53.     plt.rcParams['figure.figsize']=(75,30)
  54.     gr = sns.barplot(x=xValues, y=yValues)
  55.     sns.set_context("notebook", font_scale=10)
  56.     plt.setp(gr.get_xticklabels(), rotation=90)
  57.     plt.title("By Language(Normalized)")
  58.  
  59.  
  60. visualizeByLanguage(data)
  61.  
  62. ------------------------------
  63. etc
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement