Merge pull request #60 from brain-image-library/59-general-modality-plot

General Modality Bar Graph

Author: icaoberg

braininventory/get.py

@@ -1,6 +1,7 @@
 import json
 from datetime import date
-
+import urllib.request
+import matplotlib.pyplot as plt
 import humanize
 import matplotlib.pyplot as plt
 import pandas as pd
@@ -14,6 +15,25 @@
 import squarify
 
 
+def __get_general_modality_plot(df):
+    modality_counts = df["generalmodality"].value_counts()
+
+    plt.figure(figsize=(10, 6))
+    color = plt.cm.tab20c.colors
+
+    ax = modality_counts.plot(kind="bar", color=color, edgecolor="black")
+    plt.xlabel("General Modality", fontsize=12)
+    plt.ylabel("Frequency", fontsize=12)
+    plt.title("Frequency of General Modality", fontsize=14)
+    plt.xticks(rotation=45, ha="right", fontsize=10)
+    plt.yticks(fontsize=10)
+
+    plt.tight_layout()
+
+    filename = f'general-modality-{datetime.now().strftime("%Y%m%d")}.png'
+    plt.savefig(filename)
+
+
 def get_random_sample(df):
     """
     Returns a random sample from the dataframe from a dataset with non-zero score.
@@ -826,7 +846,7 @@ def get_projects_treemap(df):
     squarify.plot(sizes_list)
 
     filename = f'treemap-projects-{datetime.now().strftime("%Y%m%d")}.png'
-    plt.savefig("path/to/save/plot.png")
+    plt.savefig(filename)
 
 
 def __get__percentage_of_metadata_version_1(df):