[REF] plot_mod_heatmap param changes

Author: rmarkello

netneurotools/plotting.py

@@ -75,7 +75,7 @@ def sort_communities(consensus, communities):
 def plot_mod_heatmap(data, communities, *, inds=None, edgecolor='black',
                      ax=None, figsize=(6.4, 4.8), xlabels=None, ylabels=None,
                      xlabelrotation=90, ylabelrotation=0, cbar=True,
-                     square=True, ticklabels=None,
+                     square=True, xticklabels=None, yticklabels=None,
                      mask_diagonal=True, **kwargs):
     """
     Plots `data` as heatmap with borders drawn around `communities`
@@ -105,8 +105,9 @@ def plot_mod_heatmap(data, communities, *, inds=None, edgecolor='black',
         provided. Default : xlabelrotation: 90, ylabelrotation: 0
     square : bool, optional
         Setting the matrix with equal aspect. Default: True
-    ticklabels : list, optional
-        List of labels for each area. Default: None
+    {x,y}ticklabels : list, optional
+        Incompatible with `{x,y}labels`. List of labels for each entry (not
+        community) in `data`. Default: None
     cbar : bool, optional
         Whether to plot colorbar. Default: True
     mask_diagonal : bool, optional
@@ -120,6 +121,10 @@ def plot_mod_heatmap(data, communities, *, inds=None, edgecolor='black',
         Axis object containing plot
     """
 
+    for t, l in zip([xticklabels, yticklabels], [xlabels, ylabels]):
+        if t is not None and l is not None:
+            raise ValueError('Cannot set both {x,y}labels and {x,y}ticklabels')
+
     # get indices for sorting consensus
     if inds is None:
         inds = sort_communities(data, communities)
@@ -188,11 +193,13 @@ def plot_mod_heatmap(data, communities, *, inds=None, edgecolor='black',
                 ax.set_yticklabels(labels=ylabels, rotation=ylabelrotation)
                 ax.tick_params(left=False, bottom=False)
 
-    if ticklabels is not None:
-        labels_ind = [ticklabels[i] for i in inds]
-        ax.set_xticks(np.arange(len(labels_ind))+0.5)
-        ax.set_yticks(np.arange(len(labels_ind))+0.5)
+    if xticklabels is not None:
+        labels_ind = [xticklabels[i] for i in inds]
+        ax.set_xticks(np.arange(len(labels_ind)) + 0.5)
         ax.set_xticklabels(labels_ind, rotation=90)
+    if yticklabels is not None:
+        labels_ind = [yticklabels[i] for i in inds]
+        ax.set_yticks(np.arange(len(labels_ind)) + 0.5)
         ax.set_yticklabels(labels_ind)
 
     return ax