Merge pull request #94 from brain-image-library/55-data-visualization-of-generalmodality

Visualization of general modalities

Author: icaoberg

braininventory/get.py

@@ -1384,6 +1384,31 @@ def report():
 
     return report
 
+def create_general_modality_plot(df):
+  # Creates a segmented bar graph that shows the proportion of general modalities over the years. 
+  # Dropped the null values (no creation dates)
+  df[~df['creation_date'].isnull()]
+  df['dates'] = pd.to_datetime(df['creation_date'])
+  df['year'] = df['dates'].dt.year
+  df = df.dropna(subset=['year'])
+  df['year'] = df['year'].astype(int)
+  grouped = df.groupby(df['dates'].dt.year)
+
+  import matplotlib.pyplot as plt
+
+  grouped = df.groupby(['year', 'generalmodality']).size().reset_index(name='count')
+
+  pivot_df = grouped.pivot(index='year', columns='generalmodality', values='count').fillna(0)
+
+  pivot_df.plot(kind='bar', stacked=True)
+
+  plt.title('General Modalities')
+  plt.ylabel('Number of Datasets')
+  plt.xlabel('Year')
+  plt.xticks(rotation=45)
+  plt.tight_layout()
+
+  plt.show()
 
 def create_tree_map(frequency_dict, width, height):
     """
@@ -1409,4 +1434,4 @@ def create_tree_map(frequency_dict, width, height):
     today = date.today()
     output_path = f'treemap-{today.strftime("%Y%m%d")}.png'
     fig.write_image(output_path)
-    fig.show()
+    fig.show()