modusa 0.3.92__tar.gz → 0.3.93__tar.gz

{modusa-0.3.92 → modusa-0.3.93}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: modusa
-Version: 0.3.92
+Version: 0.3.93
 Summary: A modular signal analysis python library.
 Author-Email: Ankit Anand <ankit0.anand0@gmail.com>
 License: MIT

{modusa-0.3.92 → modusa-0.3.93}/pyproject.toml

@@ -16,7 +16,7 @@ dependencies = [
 ]
 requires-python = ">=3.10"
 readme = "README.md"
-version = "0.3.92"
+version = "0.3.93"
 
 [project.license]
 text = "MIT"

{modusa-0.3.92 → modusa-0.3.93}/src/modusa/__init__.py

@@ -1,11 +1,11 @@
 from modusa.utils import excp, config
 
 #=====Giving access to plot functions to plot multiple signals.=====
-from modusa.tools import plot_dist, fig
+from modusa.tools import dist_plot, hill_plot, fig
 #=====
 
 from modusa.tools import play, convert, record, save
 from modusa.tools import download
 from modusa.tools import load, load_ann
 
-__version__ = "0.3.92"
+__version__ = "0.3.93"

{modusa-0.3.92 → modusa-0.3.93}/src/modusa/tools/__init__.py

@@ -9,4 +9,4 @@ from .ann_loader import load_ann
 from .audio_recorder import record
 
 from .plotter import Fig as fig
-from .plotter import plot_dist
+from .plotter import dist_plot, hill_plot

{modusa-0.3.92 → modusa-0.3.93}/src/modusa/tools/plotter.py

@@ -712,10 +712,153 @@ class Fig:
 		fig = self._fig
 		fig.savefig(path, bbox_inches="tight")
 		
+#======== Distribution ==========
+def hill_plot(*args, labels=None, xlabel=None, ylabel=None, title=None, widths=0.7, bw_method=0.3, jitter_amount=0.1, side='upper', show_stats=True, ax=None):
+		"""
+		A plot to see distribution of different groups
+		along with statistical markers.
+		
+		Parameters
+		----------
+		*args : array-like
+				- Data arrays for each group.
+		labels : list of str, optional
+				- Labels for each group (y-axis).
+		side : str, 'upper' or 'lower'
+				Which half of the violin to draw (upper or lower relative to y-axis).
+		show_stats : bool, optional
+				Whether to show mean and median markers.
+		xlabel : str, optional
+				Label for x-axis.
+		ylabel : str, optional
+				Label for y-axis.
+		ax : matplotlib axes, optional
+				Axes to plot on.
+		title : str, optional
+				Plot title.
+		widths : float, optional
+				Width of violins.
+		bw_method : float, optional
+				Bandwidth method for KDE.
+		jitter_amount : float, optional
+				Amount of vertical jitter for strip points.
+		"""
+	
+		created_fig = False
+		if ax is None:
+			fig, ax = plt.subplots(figsize=(8, len(args) * 1.5))
+			created_fig = True
+			
+		n = len(args)
+	
+		# Default labels/colors
+		if labels is None:
+			labels = [f"Group {i}" for i in range(1, n+1)]
+			
+		colors = plt.cm.tab10.colors
+		if len(colors) < n:
+			colors = [colors[i % len(colors)] for i in range(n)] # Repeat colors
+			
+		# --- Half-violin ---
+		parts = ax.violinplot(args, vert=False, showmeans=False, showmedians=False, widths=widths, bw_method=bw_method)
+	
+		# Remove the default bar lines from violin plot
+		for key in ['cbars', 'cmins', 'cmaxes', 'cmedians']:
+			if key in parts:
+				parts[key].set_visible(False)
+					
+		# Clip violin bodies to show only upper or lower half
+		for i, pc in enumerate(parts['bodies']):
+			verts = pc.get_paths()[0].vertices
+			y_center = i + 1  # Center y-position for this violin
+		
+			if side == 'upper':
+					verts[:, 1] = np.maximum(verts[:, 1], y_center)
+			else:  # 'lower'
+					verts[:, 1] = np.minimum(verts[:, 1], y_center)
+		
+			pc.set_facecolor(colors[i])
+			pc.set_edgecolor(colors[i])
+			pc.set_linewidth(1.5)
+			pc.set_alpha(0.3)
+			
+		# --- Strip points with jitter ---
+		for i, x in enumerate(args, start=1):
+			x = np.array(x)
+			jitter = (np.random.rand(len(x)) - 0.5) * jitter_amount
+			y_positions = np.full(len(x), i) + jitter
+			ax.scatter(x, y_positions, color=colors[i-1], alpha=0.6, s=25, edgecolor="white", linewidth=0.8, zorder=2)
+			
+		# --- Statistical markers on violin distribution curve ---
+		if show_stats:
+			for i, (pc, x) in enumerate(zip(parts['bodies'], args), start=1):
+				x = np.array(x)
+				median_val = np.median(x)
+				mean_val = np.mean(x)
+			
+				# Get the violin curve vertices
+				verts = pc.get_paths()[0].vertices
+			
+				# Find y-position on violin curve for median
+				median_mask = np.abs(verts[:, 0] - median_val) < (np.ptp(x) * 0.01)
+				if median_mask.any():
+					median_y = np.max(verts[median_mask, 1]) if side == 'upper' else np.min(verts[median_mask, 1])
+				else:
+					median_y = i + widths/2 if side == 'upper' else i - widths/2
+					
+				# Find y-position on violin curve for mean
+				mean_mask = np.abs(verts[:, 0] - mean_val) < (np.ptp(x) * 0.01)
+				if mean_mask.any():
+					mean_y = np.max(verts[mean_mask, 1]) if side == 'upper' else np.min(verts[mean_mask, 1])
+				else:
+					mean_y = i + widths/2 if side == 'upper' else i - widths/2
+					
+				# Triangle offset from curve
+				triangle_offset = 0.05
+			
+				# Mean marker - triangle below curve pointing up
+				ax.scatter(mean_val, mean_y - triangle_offset, marker='^', s=30, 
+									facecolor=colors[i-1], edgecolor='none', 
+									linewidth=0, zorder=6,
+									label='Mean' if i == 1 else '')
+			
+				# Mean value text - below the triangle
+				ax.text(mean_val, mean_y - triangle_offset - 0.05, f'mean: {mean_val:.2f}', ha='center', va='top', fontsize=8, fontweight='bold', color=colors[i-1], zorder=7)
+			
+				# Median marker - triangle above curve pointing down
+				ax.scatter(median_val, median_y + triangle_offset, marker='v', s=30, 
+									facecolor=colors[i-1], edgecolor='none', 
+									linewidth=0, zorder=6,
+									label='Median' if i == 1 else '')
+			
+				# Median value text - above the triangle
+				ax.text(median_val, median_y + triangle_offset + 0.05, f'median: {median_val:.2f}', ha='center', va='bottom', fontsize=8, fontweight='bold', color=colors[i-1], zorder=7)
+					
+		# --- Labels & formatting ---
+		ax.set_yticks(range(1, n + 1))
+		ax.set_yticklabels(labels, fontsize=10)
+		if side == "lower":
+			ax.set_ylim(0.2, n + 0.5)
+		else:
+			ax.set_ylim(0.5, n + 0.5) 
+			
+		# Style improvements
+		ax.spines['top'].set_visible(False)
+		ax.spines['right'].set_visible(False)
+		ax.grid(axis='x', alpha=0.3, linestyle='--', linewidth=0.5)
+	
+		if xlabel:
+			ax.set_xlabel(xlabel, fontsize=11, fontweight='bold')
+		if ylabel:
+			ax.set_ylabel(ylabel, fontsize=11, fontweight='bold')
+		if title:
+			ax.set_title(title, fontsize=13, fontweight='bold', pad=15)
+			
+		plt.tight_layout()
+		return fig
 
 
-#======== Plot distribution ===========
-def plot_dist(*args, ann=None, xlim=None, ylim=None, ylabel=None, xlabel=None, title=None, legend=None, show_hist=True, npoints=200, bins=30):
+def dist_plot(*args, ann=None, xlim=None, ylim=None, ylabel=None, xlabel=None, title=None, legend=None, show_hist=True, npoints=200, bins=30):
 		"""
 		Plot distribution.