.. _pngsuite-pngsuite:

pngsuite example code: pngsuite.py
==================================

[`source code <pngsuite.py>`_]

::

    """
    This test loads a subset of the files in Willem van Schaik's PNG test
    suite available here:
    
       http://libpng.org/pub/png/pngsuite.html
    
    The result should look like truth.png.
    """
    
    from matplotlib import pyplot as plt
    import matplotlib.cm as cm
    import glob
    
    files = glob.glob("basn*.png")
    files.sort()
    
    plt.figure(figsize=(len(files), 2))
    
    for i, fname in enumerate(files):
        data = plt.imread(fname)
        cmap = None # use default colormap
        if data.ndim==2:
            # keep grayscale images gray
            cmap = cm.gray
        plt.imshow(data, extent=[i,i+1,0,1], cmap=cmap)
    
    plt.gca().get_frame().set_facecolor("#ddffff")
    plt.gca().set_xlim(0, len(files))
    plt.show()
    

Keywords: python, matplotlib, pylab, example, codex (see :ref:`how-to-search-examples`)