Initial commit.

This is Incenp.org's PyPlot, a Python helper library for plotting
tasks.

.gitignore

@@ -0,0 +1,7 @@
+.project
+.pydevproject
+.settings
+build
+dist
+*.egg-info
+__pycache__

incenp/__init__.py

@@ -0,0 +1 @@
+__path__ = __import__('pkgutil').extend_path(__path__, __name__)

incenp/plotting/__init__.py

@@ -0,0 +1 @@
+__version__ = '0.1.0'

incenp/plotting/scatter.py

@@ -0,0 +1,82 @@
+# -*- coding: utf-8 -*-
+# Incenp.Plotting - Incenp.org's plotting helper library
+# Copyright © 2020 Damien Goutte-Gattat
+#
+# This program is free software; you can redistribute it and/or modify
+# it under the terms of the GNU General Public License as published by
+# the Free Software Foundation; either version 3 of the License, or
+# (at your option) any later version.
+#
+# This program is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
+# GNU General Public License for more details.
+#
+# You should have received a copy of the GNU General Public License
+# along with this program. If not, see <http://www.gnu.org/licenses/>.
+
+"""Module description."""
+
+from .util import xdistr
+
+
+def scatterplot_subtrack(ax, data, n_track, n_subtrack, max_subtrack, color, width=.7, min_sep=.1):
+    offset = n_track * max_subtrack + n_subtrack
+    xs = xdistr(data.values, width, center=True, min_sep=min_sep, offset=offset)
+    ax.plot(xs, data.values, color + '.')
+    ax.hlines(data.mean(), offset - .5, offset + .5, color)
+    
+    
+def scatterplot(ax, data, columns, subtrackcolumns=False, tracks=[None], trackname=0, subtracks=[None], subtrackname=1,
+                colors='rgb', width=.7, min_sep=.1):
+    
+    if isinstance(columns, list):
+        if subtrackcolumns:
+            n = len(columns) + 1
+            labels = tracks
+        else:
+            n = len(subtracks) + 1
+            labels = columns
+    else:
+        n = len(subtracks) + 1
+        labels = tracks
+        
+    i = 0
+    
+    if isinstance(columns, list) and not subtrackcolumns:
+        for column in columns:
+            j = 0
+            
+            for subtrack in subtracks:
+                if subtrack is None:
+                    subset = data.loc[:, column].dropna()
+                else:
+                    level = subtrackname if data.index.nlevels > 1 else None
+                    subset = data.xs(subtrack, level=level).loc[:, column].dropna()
+                scatterplot_subtrack(ax, subset, i, j, n, colors[j % n], width, min_sep)
+                j += 1
+            i += 1
+        
+    else:
+        for track in tracks:
+            j = 0
+            
+            if isinstance(columns, list) and subtrackcolumns:
+                for column in columns:
+                    subset = data.xs(track, level=trackname).loc[:, column].dropna()
+                    scatterplot_subtrack(ax, subset, i, j, n, colors[j % n], width, min_sep)
+                    j += 1
+                i += 1
+            else:
+                for subtrack in subtracks:
+                    indexer = [track, subtrack] if subtrack else [track]
+                    level = [trackname, subtrackname] if subtrack else [trackname]
+                    
+                    subset = data.xs(indexer, level=level).loc[:, columns].dropna()
+                    scatterplot_subtrack(ax, subset, i, j, n, colors[j % n], width, min_sep)
+                    j += 1
+                i += 1
+                
+    ax.set_xticks([(.5 * (n - 2)) + n * i for i in range(len(labels))])
+    ax.set_xticklabels(labels)
+    

incenp/plotting/util.py

@@ -0,0 +1,77 @@
+# -*- coding: utf-8 -*-
+# Incenp.Plotting - Incenp.org's plotting helper library
+# Copyright © 2020 Damien Goutte-Gattat
+#
+# This program is free software; you can redistribute it and/or modify
+# it under the terms of the GNU General Public License as published by
+# the Free Software Foundation; either version 3 of the License, or
+# (at your option) any later version.
+#
+# This program is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
+# GNU General Public License for more details.
+#
+# You should have received a copy of the GNU General Public License
+# along with this program. If not, see <http://www.gnu.org/licenses/>.
+
+"""Miscellaneous utility functions for plotting tasks."""
+
+
+def xdistr(values, width, offset=0, even_max=10, center=False, min_sep=1):
+    """Distribute coordinates around an axis.
+    
+    Given a list of values, return an array of x coordinates so that
+    the plotted values are equally distributed.
+    
+    :param values: The list of values to distribute
+    :param width: The distance along which to distribute the values
+    :param offset: An offset to apply to each returned coordinates
+    :param even_max: If the number of values on a given rank exceeds
+        this parameter, they will be put closer together
+    :param center: Distribute coordinates on both sides of offset
+    :param min_sep: If two values differ by less than this parameter,
+        they will be considered to belong on the same rank
+    :return: An array of x coordinates
+    """
+    
+    dist = {}
+    spent = {}
+    xcoords = []
+    
+    values = [int(i / min_sep) for i in values]
+    
+    # Get the distribution of values
+    for v in values:
+        if v in dist:
+            dist[v] += 1
+        else:
+            dist[v] = 1
+        spent[v] = 0
+        
+    # Now proceeds
+    for v in values:
+        
+        # Compute step to distribute the values on the specified width
+        step = width / max(even_max, dist[v])
+        
+        if not center:
+            x = offset + step * spent[v]
+        elif spent[v] % 2 == 0:
+            # If we've already drawn an even number of points,
+            # draw the next one on the right of offset
+            x = offset + (step / 2) * spent[v]
+        else:
+            # Draw the next point on the left of offset
+            x = offset - (step / 2) * (spent[v] + 1)
+            
+        # If there is an even number of points in that rank,
+        # shift all points by a half-step
+        if dist[v] % 2 == 0:
+            x += step / 2
+            
+        spent[v] += 1
+        xcoords.append(x)
+        
+    return xcoords
+        

setup.py

@@ -0,0 +1,32 @@
+# -*- coding: utf-8 -*-
+# Incenp.Plotting - Incenp.org's plotting helper library
+# Copyright © 2020 Damien Goutte-Gattat
+#
+# This program is free software; you can redistribute it and/or modify
+# it under the terms of the GNU General Public License as published by
+# the Free Software Foundation; either version 3 of the License, or
+# (at your option) any later version.
+#
+# This program is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
+# GNU General Public License for more details.
+#
+# You should have received a copy of the GNU General Public License
+# along with this program. If not, see <http://www.gnu.org/licenses/>.
+
+"""Module description."""
+
+from setuptools import setup, find_packages
+from incenp.plotting import __version__
+
+setup(
+    name='incenp.plotting',
+    version=__version__,
+    description='Incenp.org´s plotting helper library',
+    author='Damien Goutte-Gattat',
+    author_email='dgouttegattat@incenp.org',
+    
+    packages=find_packages(),
+    include_package_data=True
+    )