freealg 0.0.1__py3-none-any.whl → 0.0.2__py3-none-any.whl

freealg/__version__.py

@@ -1 +1 @@
-__version__ = "0.0.1"
+__version__ = "0.0.2"

freealg/_plot_util.py

@@ -54,7 +54,7 @@ def plot_coeff(psi, latex=False, save=False):
                 save_filename = 'energy.pdf'
 
         texplot.show_or_save_plot(plt, default_filename=save_filename,
-                                  transparent_background=True, dpi=200,
+                                  transparent_background=True, dpi=400,
                                   show_and_save=save_status, verbose=True)
 
 
@@ -76,6 +76,8 @@ def plot_density(x, rho, eig=None, support=None, label='',
             bins = numpy.linspace(lam_m, lam_p, 250)
             _ = ax.hist(eig, bins, density=True, color='silver',
                         edgecolor='none', label='Histogram')
+        else:
+            plt.fill_between(x, y1=rho, y2=0, color='silver', zorder=-1)
 
         ax.plot(x, rho, color='black', label=label, zorder=3)
         ax.set_xlabel(r'$\lambda$')
@@ -99,7 +101,7 @@ def plot_density(x, rho, eig=None, support=None, label='',
                 save_filename = 'density.pdf'
 
         texplot.show_or_save_plot(plt, default_filename=save_filename,
-                                  transparent_background=True, dpi=200,
+                                  transparent_background=True, dpi=400,
                                   show_and_save=save_status, verbose=True)
 
 
@@ -149,7 +151,7 @@ def plot_hilbert(x, hilb, support=None, latex=False, save=False):
                 save_filename = 'hilbert.pdf'
 
         texplot.show_or_save_plot(plt, default_filename=save_filename,
-                                  transparent_background=True, dpi=200,
+                                  transparent_background=True, dpi=400,
                                   show_and_save=save_status, verbose=True)
 
 
@@ -327,7 +329,7 @@ def plot_stieltjes(x, y, m1, m2, support, latex=False, save=False):
                 save_filename = 'stieltjes.pdf'
 
         texplot.show_or_save_plot(plt, default_filename=save_filename,
-                                  transparent_background=True, dpi=200,
+                                  transparent_background=True, dpi=400,
                                   show_and_save=save_status, verbose=True)
 
 
@@ -456,7 +458,7 @@ def plot_stieltjes_on_disk(r, t, m1_D, m2_D, support, latex=False, save=False):
                 save_filename = 'stieltjes_disk.pdf'
 
         texplot.show_or_save_plot(plt, default_filename=save_filename,
-                                  transparent_background=True, dpi=200,
+                                  transparent_background=True, dpi=400,
                                   show_and_save=save_status, verbose=True)
 
 
@@ -495,5 +497,44 @@ def plot_glue_fit(x_supp, g_supp, g_supp_approx, support, latex=False,
                 save_filename = 'glue_fit.pdf'
 
         texplot.show_or_save_plot(plt, default_filename=save_filename,
-                                  transparent_background=True, dpi=200,
+                                  transparent_background=True, dpi=400,
+                                  show_and_save=save_status, verbose=True)
+
+
+# ============
+# plot samples
+# ============
+
+def plot_samples(x, rho, x_min, x_max, samples, latex=False, save=False):
+    """
+    """
+
+    with texplot.theme(use_latex=latex):
+
+        fig, ax = plt.subplots(figsize=(6, 3))
+
+        bins = numpy.linspace(x_min, x_max, samples.size // 10)
+        _ = ax.hist(samples, bins, density=True, color='silver',
+                    label='Samples histogram')
+        ax.plot(x, rho, color='black', label='Exact density')
+        ax.legend(fontsize='small')
+        ax.set_ylim(bottom=0)
+        ax.set_xlim([x[0], x[-1]])
+        ax.set_xlabel(r'$\lambda$')
+        ax.set_ylabel(r'$\rho(\lambda)$''')
+        ax.set_title('Histogram of Samples from Distribution')
+
+        # Save
+        if save is False:
+            save_status = False
+            save_filename = ''
+        else:
+            save_status = True
+            if isinstance(save, str):
+                save_filename = save
+            else:
+                save_filename = 'samples.pdf'
+
+        texplot.show_or_save_plot(plt, default_filename=save_filename,
+                                  transparent_background=True, dpi=400,
                                   show_and_save=save_status, verbose=True)

freealg/distributions/marchenko_pastur.py

@@ -14,7 +14,7 @@
 import numpy
 from scipy.interpolate import interp1d
 from .._plot_util import plot_density, plot_hilbert, plot_stieltjes, \
-    plot_stieltjes_on_disk
+    plot_stieltjes_on_disk, plot_samples
 
 try:
     from scipy.integrate import cumtrapz
@@ -77,9 +77,7 @@ class MarchenkoPastur(object):
 
     .. [1] Marcenko, V. A., Pastur, L. A. (1967). Distribution of eigenvalues
            for some sets of random matrices. Mathematics of the USSR-Sbornik,
-           1(4), 457 (`DOI<
-           https://iopscience.iop.org/article/10.1070/SM1967v001n04ABEH001994
-           >`__)
+           1(4), 457
 
     Examples
     --------
@@ -149,6 +147,10 @@ class MarchenkoPastur(object):
             >>> from freealg.distributions import MarchenkoPastur
             >>> mp = MarchenkoPastur(1/50)
             >>> rho = mp.density(plot=True)
+
+        .. image:: ../_static/images/plots/mp_density.png
+            :align: center
+            :class: custom-dark
         """
 
         # Create x if not given
@@ -167,8 +169,7 @@ class MarchenkoPastur(object):
             numpy.sqrt((self.lam_p - x[mask]) * (x[mask] - self.lam_m))
 
         if plot:
-            plot_density(x, rho, support=self.support, label='', latex=latex,
-                         save=save)
+            plot_density(x, rho, label='', latex=latex, save=save)
 
         return rho
 
@@ -189,7 +190,7 @@ class MarchenkoPastur(object):
             spectral density is used.
 
         plot : bool, default=False
-            If `True`, density is plotted.
+            If `True`, Hilbert transform is plotted.
 
         latex : bool, default=False
             If `True`, the plot is rendered using LaTeX. This option is
@@ -214,6 +215,10 @@ class MarchenkoPastur(object):
             >>> from freealg.distributions import MarchenkoPastur
             >>> mp = MarchenkoPastur(1/50)
             >>> hilb = mp.hilbert(plot=True)
+
+        .. image:: ../_static/images/plots/mp_hilbert.png
+            :align: center
+            :class: custom-dark
         """
 
         # Create x if not given
@@ -225,9 +230,18 @@ class MarchenkoPastur(object):
             x_max = numpy.ceil(center + radius * scale)
             x = numpy.linspace(x_min, x_max, 500)
 
-        hilb = numpy.zeros_like(x)
-        mask = numpy.logical_and(x >= self.lam_m, x <= self.lam_p)
-        hilb[mask] = (x[mask] - 1 - self.lam) / (self.lam * x[mask])
+        def _P(x):
+            return x - 1 + self.lam
+
+        def _Q(x):
+            return self.lam * x
+
+        P = _P(x)
+        Q = _Q(x)
+        Delta2 = P**2 - 4.0 * Q
+        Delta = numpy.sqrt(numpy.maximum(Delta2, 0))
+        sign = numpy.sign(P)
+        hilb = (P - sign * Delta) / (2.0 * Q)
 
         # using negative sign convention
         hilb = -hilb
@@ -319,7 +333,7 @@ class MarchenkoPastur(object):
             If `None`, a grid on the interval ``[-1, 1]`` is used.
 
         plot : bool, default=False
-            If `True`, density is plotted.
+            If `True`, Stieltjes transform is plotted.
 
         on_disk : bool, default=False
             If `True`, the Stieltjes transform is mapped on unit disk. This
@@ -346,12 +360,25 @@ class MarchenkoPastur(object):
         Examples
         --------
 
-        .. code-block::python
+        .. code-block:: python
 
             >>> from freealg.distributions import MarchenkoPastur
             >>> mp = MarchenkoPastur(1/50)
             >>> m1, m2 = mp.stieltjes(plot=True)
 
+        .. image:: ../_static/images/plots/mp_stieltjes.png
+            :align: center
+            :class: custom-dark
+
+        Plot on unit disk using Cayley transform:
+
+        .. code-block:: python
+
+            >>> m1, m2 = mp.stieltjes(plot=True, on_disk=True)
+
+        .. image:: ../_static/images/plots/mp_stieltjes_disk.png
+            :align: center
+            :class: custom-dark
         """
 
         if (plot is True) and (on_disk is True):
@@ -407,7 +434,8 @@ class MarchenkoPastur(object):
     # sample
     # ======
 
-    def sample(self, size, x_min=None, x_max=None):
+    def sample(self, size, x_min=None, x_max=None, plot=False, latex=False,
+               save=False):
         """
         Sample from distribution.
 
@@ -425,6 +453,18 @@ class MarchenkoPastur(object):
             Maximum of sample values. If `None`, the right edge of the support
             is used.
 
+        plot : bool, default=False
+            If `True`, samples histogram is plotted.
+
+        latex : bool, default=False
+            If `True`, the plot is rendered using LaTeX. This option is
+            relevant only if ``plot=True``.
+
+        save : bool, default=False
+            If not `False`, the plot is saved. If a string is given, it is
+            assumed to the save filename (with the file extension). This option
+            is relevant only if ``plot=True``.
+
         Returns
         -------
 
@@ -443,7 +483,11 @@ class MarchenkoPastur(object):
 
             >>> from freealg.distributions import MarchenkoPastur
             >>> mp = MarchenkoPastur(1/50)
-            >>> s = mp.sample(2000)
+
+         >>> s = mp.sample(2000)
+        .. image:: ../_static/images/plots/mp_samples.png
+            :align: center
+            :class: custom-dark
         """
 
         if x_min is None:
@@ -466,7 +510,19 @@ class MarchenkoPastur(object):
 
         # Sample and map
         u = numpy.random.rand(size)
-        return inv_cdf(u)
+        samples = inv_cdf(u)
+
+        if plot:
+            radius = 0.5 * (self.lam_p - self.lam_m)
+            center = 0.5 * (self.lam_p + self.lam_m)
+            scale = 1.25
+            x_min = numpy.floor(center - radius * scale)
+            x_max = numpy.ceil(center + radius * scale)
+            x = numpy.linspace(x_min, x_max, 500)
+            rho = self.density(x)
+            plot_samples(x, rho, x_min, x_max, samples, latex=latex, save=save)
+
+        return samples
 
     # ======
     # matrix

freealg/freeform.py

@@ -625,7 +625,7 @@ class FreeForm(object):
         --------
 
         density
-        stiletjes
+        stieltjes
 
         Notes
         -----

{freealg-0.0.1.dist-info → freealg-0.0.2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: freealg
-Version: 0.0.1
+Version: 0.0.2
 Summary: Free probability for large matrices
 Keywords: leaderboard bot chat
 Platform: Linux
@@ -30,27 +30,7 @@ Requires-Dist: texplot
 Requires-Dist: matplotlib
 Requires-Dist: colorcet
 Provides-Extra: test
-Requires-Dist: tox; extra == "test"
-Requires-Dist: pytest-cov; extra == "test"
-Requires-Dist: codecov; extra == "test"
 Provides-Extra: docs
-Requires-Dist: sphinx; extra == "docs"
-Requires-Dist: sphinx-math-dollar; extra == "docs"
-Requires-Dist: sphinx-toggleprompt<0.4; extra == "docs"
-Requires-Dist: pydata_sphinx_theme==0.9.0; extra == "docs"
-Requires-Dist: graphviz; extra == "docs"
-Requires-Dist: sphinx-automodapi; extra == "docs"
-Requires-Dist: numpydoc; extra == "docs"
-Requires-Dist: nbsphinx; extra == "docs"
-Requires-Dist: sphinx_design; extra == "docs"
-Requires-Dist: sphinx-multitoc-numbering; extra == "docs"
-Requires-Dist: pandoc; extra == "docs"
-Requires-Dist: sphinx-prompt; extra == "docs"
-Requires-Dist: sphinx-copybutton; extra == "docs"
-Requires-Dist: sphinx-gallery; extra == "docs"
-Requires-Dist: sphinxcontrib-youtube; extra == "docs"
-Requires-Dist: sphinxext-opengraph; extra == "docs"
-Requires-Dist: sphinx-argparse; extra == "docs"
 Dynamic: classifier
 Dynamic: description
 Dynamic: description-content-type

freealg-0.0.2.dist-info/RECORD

@@ -0,0 +1,16 @@
+freealg/__init__.py,sha256=K92neXJZ9VE1U_j_pj28Qyq1MzlMXhOuYK2ZihgwCaU,463
+freealg/__version__.py,sha256=QvlVh4JTl3JL7jQAja76yKtT-IvF4631ASjWY1wS6AQ,22
+freealg/_chebyshev.py,sha256=Cw48gXF6kd3IAQuLTEWadySeKGnY9TynzX9MNmycMUU,4697
+freealg/_damp.py,sha256=k2vtBtWOxQBf4qXaWu_En81lQBXbEO4QbxxWpvuVhdE,1802
+freealg/_jacobi.py,sha256=F9K0IzbExpxGQY2OJQb-XT4Ee1zWTZvMXRUdU8cbwSE,4916
+freealg/_pade.py,sha256=rWILLpEL910YdixC2c5Cw77HqogAYoxi2YZgRVXEELM,4022
+freealg/_plot_util.py,sha256=OILqOqKdLGEabk41KnmxNftItzMMFHOL-LZ35yNvdyk,18653
+freealg/_util.py,sha256=wJ-t8LMZZFEr2PsZEVqTJP_jQTQ3rHUf0dO27F6L7YQ,2310
+freealg/freeform.py,sha256=LLkGoPLxvn8VMHuVQ1LGNkcAYZUi3GduFB8qlsr69qg,18595
+freealg/distributions/__init__.py,sha256=7t4HbP_EofiFDYLH6jbD94AIumOdcHn1y_Qo54mpLFM,614
+freealg/distributions/marchenko_pastur.py,sha256=k8SoEgB2DFXLqGF-Hyqm8Sfh1DlF3khMssuxak5Uaqw,15827
+freealg-0.0.2.dist-info/licenses/LICENSE.txt,sha256=J-EEYEtxb3VVf_Bn1TYfWnpY5lMFIM15iLDDcnaDTPA,1443
+freealg-0.0.2.dist-info/METADATA,sha256=lMjLvkUpPUAr_QPthjnrIfg9KTs8DLEpTG68w8Ak0rU,2888
+freealg-0.0.2.dist-info/WHEEL,sha256=DnLRTWE75wApRYVsjgc6wsVswC54sMSJhAEd4xhDpBk,91
+freealg-0.0.2.dist-info/top_level.txt,sha256=eR2wrgYwDdnnJ9Zf5PruPqe4kQav0GMvRsqct6y00Q8,8
+freealg-0.0.2.dist-info/RECORD,,

freealg-0.0.1.dist-info/RECORD

@@ -1,16 +0,0 @@
-freealg/__init__.py,sha256=K92neXJZ9VE1U_j_pj28Qyq1MzlMXhOuYK2ZihgwCaU,463
-freealg/__version__.py,sha256=sXLh7g3KC4QCFxcZGBTpG2scR7hmmBsMjq6LqRptkRg,22
-freealg/_chebyshev.py,sha256=Cw48gXF6kd3IAQuLTEWadySeKGnY9TynzX9MNmycMUU,4697
-freealg/_damp.py,sha256=k2vtBtWOxQBf4qXaWu_En81lQBXbEO4QbxxWpvuVhdE,1802
-freealg/_jacobi.py,sha256=F9K0IzbExpxGQY2OJQb-XT4Ee1zWTZvMXRUdU8cbwSE,4916
-freealg/_pade.py,sha256=rWILLpEL910YdixC2c5Cw77HqogAYoxi2YZgRVXEELM,4022
-freealg/_plot_util.py,sha256=RfGBV25-tEkdoo-RzgSllQH80WV8tOeXmDr_0B8au0w,17347
-freealg/_util.py,sha256=wJ-t8LMZZFEr2PsZEVqTJP_jQTQ3rHUf0dO27F6L7YQ,2310
-freealg/freeform.py,sha256=4njw6OZw5pH9omm18ioLx3WHP0L0-Y663A7bleZWC0o,18595
-freealg/distributions/__init__.py,sha256=7t4HbP_EofiFDYLH6jbD94AIumOdcHn1y_Qo54mpLFM,614
-freealg/distributions/marchenko_pastur.py,sha256=eAH1qlHgxCPPqOh8WucJgYnrHDqV4x4mif1rTKIBFDE,14079
-freealg-0.0.1.dist-info/licenses/LICENSE.txt,sha256=J-EEYEtxb3VVf_Bn1TYfWnpY5lMFIM15iLDDcnaDTPA,1443
-freealg-0.0.1.dist-info/METADATA,sha256=nm9YVJz4d-3BBPowFGbfSXkGFctl9sEvqEUI-sOf1JE,3825
-freealg-0.0.1.dist-info/WHEEL,sha256=DnLRTWE75wApRYVsjgc6wsVswC54sMSJhAEd4xhDpBk,91
-freealg-0.0.1.dist-info/top_level.txt,sha256=eR2wrgYwDdnnJ9Zf5PruPqe4kQav0GMvRsqct6y00Q8,8
-freealg-0.0.1.dist-info/RECORD,,