Merge pull request #68 from brain-image-library/67-genotype-tree-map

Genotype Tree Map

Author: icaoberg

braininventory/get.py

@@ -850,6 +850,19 @@ def get_projects_treemap(df):
 
 
 def __get_modalities(df):
+    """
+    Get the counts of different modalities from the DataFrame.
+
+    This function takes a pandas DataFrame as input and extracts the counts of different modalities
+    from the 'generalmodality' column of the DataFrame. It returns a dictionary where the keys
+    represent the unique modalities, and the values represent their respective counts.
+
+    Parameters:
+        df (pandas.DataFrame): The input DataFrame containing the 'generalmodality' column.
+
+    Returns:
+        dict: A dictionary with modalities as keys and their corresponding counts as values.
+    """
     return (df["generalmodality"].value_counts()).to_dict()
 
 

setup.py

@@ -20,3 +20,38 @@
     ],
     python_requires=">=3.6",
 )
+import pandas as pd
+import urllib.request
+
+url = 'https://download.brainimagelibrary.org/inventory/daily/reports/today.json'
+file_path, _ = urllib.request.urlretrieve(url)
+df = pd.read_json(file_path)
+
+#get affiliations
+affiliations = df['genotype'].value_counts().to_dict()
+affiliations
+
+!pip install kaleido
+
+import plotly.graph_objects as go
+from datetime import date
+
+def create_tree_map(frequency_dict):
+    labels = list(frequency_dict.keys())
+    values = list(frequency_dict.values())
+
+    fig = go.Figure(go.Treemap(
+        labels=labels,
+        parents=[''] * len(labels),
+        values=values,
+        textinfo='label+value'
+    ))
+
+    fig.update_layout(title='Genotypes')
+
+    today = date.today()
+    output_path = f'treemap-{today.strftime("%Y%m%d")}.png'
+    fig.write_image(output_path)
+    fig.show()
+
+create_tree_map(affiliations)