Merge pull request #66 from brain-image-library/58-treemap-of-projects-frequency

58 treemap of projects frequency

Author: icaoberg

braininventory/get.py

@@ -2,6 +2,8 @@
 import pandas as pd
 import json
 from datetime import date
+import squarify
+import matplotlib.pyplot as plt
 
 
 def today():
@@ -54,26 +56,30 @@ def __get_contributor(df):
 def __get_affilation(df):
     return df["affiliation"].value_counts().to_dict()
 
+
 def __get_awards(df):
     return df["award_number"].unique()
 
 
 def __get_award_number(df):
     return df["award_number"].value_counts().to_dict()
 
+
 def __get_species(df):
     return df["species"].value_counts().to_dict()
 
 
 def __get_cnbtaxonomy(df):
     return df["cnbtaxonomy"].value_counts().to_dict()
 
+
 def __get_genotypes(df):
     """
     Write documentation here.
     """
     return df["genotype"].unique()
 
+
 def __get_genotype_frequency(df):
     """
     Write documentation here.
@@ -84,6 +90,7 @@ def __get_genotype_frequency(df):
 def __get_generalmodality(df):
     return df["generalmodality"].value_counts().to_dict()
 
+
 def __get_techniques(df):
     """
     Write documentation here.
@@ -97,9 +104,11 @@ def techniques_frequency(df):
     """
     return df["technique"].value_counts().to_dict()
 
+
 def __get_locations(df):
     return df["locations"].value_counts().to_dict()
 
+
 def __get_contributors(df):
     """
     This returns an array of contributor names from the contributorname column.
@@ -128,6 +137,24 @@ def __get_number_of_projects(df):
 
     return len(df["project"].unique())
 
+
+def get_projects_treemap(df):
+    """
+    Created a code for the visualization of projects frequency 
+
+    Input: project values 
+    Output: treemap graph of projects frequency
+    """
+
+    df = df["project"].value_counts().to_dict()
+    sizes_list = list(df.values())
+    names_list = list(df.keys())
+    squarify.plot(sizes_list)
+
+    filename = f'treemap-projects-{datetime.now().strftime("%Y%m%d")}.png'
+    plt.savefig("path/to/save/plot.png")
+
+
 def report():
     # Get today's date
     tdate = date.today()
@@ -153,4 +180,7 @@ def report():
     report["technique"] = __get_technique(df)
     report["locations"] = __get_locations(df)
 
+    # plots
+    get_projects_treemap(df)
+
     return report