Skip to main content

A thin wrapper around matplotlib, which lets you save figure and axes settings.

Project description

justplot
========

A thin wrapper around matplotlib, which lets you save figure and axes
settings for plot reuse.

How to install the module
-------------------------

``pip`` with ``--pre`` as option cause this is still an alpha build.

::

python3.4 -m pip install justplot --pre

-- or --

::

git clone https://github.com/OnionNinja/justplot
cd justplot
python3.4 setup.py install

How to use the module [STRIKEOUT:(as intended)]
-----------------------------------------------

TL;DR
^^^^^

.. code:: python

from justplot import justplot as jp
import numpy

lat = np.asarray([-90,-60,-30,0,30,60,90])
lon = np.asarray([-180,-150,-120,-90,-60,-30,0,30,60,90,120,150,180])
lat,lon = np.meshgrid(lat, lon)
temp = np.random.randint(10,35,(13,7))
wind = np.random.randint(100,300,(13,7))

awesome = jp.JustPlot(
title = "Where is it hot?",
xlabel = "Latitude",
ylabel = "Longitude",
# add more options here
)
awesome.add_contourf(lat,lon,temp,ylabel="Temperature (K)", colorbar=True)
awesome.save("Sun.pdf","pdf")

windy = awesome.copy()
windy(title="A little bit windy don't ya think?")
windy.add_contour(lat,lon,wind,ylabel="Wind")
windy.save("Wind.pdf","pdf")

Usage
^^^^^

Let's say we have some data we want to plot (suprise!). As an example I
will use a small data set from
`ucar <https: www.unidata.ucar.edu="" software="" netcdf="" examples="" sresa1b_ncar_ccsm3-example.nc="">`__.

First off we have to load the data.

.. code:: python

import xarray
import numpy as np

data = xarray.open_dataset('sresa1b_ncar_ccsm3-example.nc', decode_times=False) #pretty cryptic, I know
lat = data.lat.data
lon = data.lon.data
temp = data.tas.data[0,:,:]
wind = data.ua.data[0,4,:,:]
xlat,xlon = np.meshgrid(lon,lat)

Let's build the figure and axes settings first.

.. code:: python

from justplot import justplot as jp

blueprint = jp.JustPlot(
title = "Where is it hot?",
ylabel = "Latitude",
xlabel = "Longitude",
# add more options here
)

Now we can add the actual plot.

.. code:: python

blueprint.add_contourf(xlat,xlon,temp,label="Temperature")
blueprint.save('blueprint-1.png','png')

This plot looks like the following: |First plot|

The title is strange and too small. Additionally the both Axes could use
some more ticks. So let's change that.

.. code:: python

from matplotlib.ticker import AutoMinorLocator
blueprint(title={"label":"World Temperature","fontsize":19},
xticks=np.arange(0,361,30),
yticks=np.arange(-90,91,30),
xminor=AutoMinorLocator(3),
yminor=AutoMinorLocator(3),
)
blueprint.save("blueprint-2.png",'png')

Basemap support will be added further down the road. It is on
[STRIKEOUT:my] **our** to do list. You can quote me on that.

Our plot looks now like this: |Second plot|

Good enough. Now pretend that you like this plot sooo much that you want
all your future plots to look like it. For example the one for Wind.
Let's do that.

.. code:: python

windy = blueprint.copy()
windy(title={"label":"Wind","fontsize":19})
windy.add_contour(xlat,xlon,wind,label="Wind")
windy.save("windy.png","png")

This looks like... |Wind plot|

Let's put all this together.

.. code:: python

both = windy.copy()
both(title={"label":"World Temperature + Wind","fontsize":19})
both.add_contourf(xlat,xlon,temp,label="Temperature")
both.add_contour(xlat,xlon,wind,label="Wind",colors=('k'))
blueprint.save("both.png","png")

.. figure:: https://raw.githubusercontent.com/OnionNinja/justplot/master/example/both.png
:alt: Both plots

Both plots
You can find the example code and data in the `example <example>`__
folder.

How to improve the module (aka **our** todo list)
-------------------------------------------------

- [ ] Save settings from y axes
- [ ] Add Basemap support
- [ ] Restructure
- [ ] Write import function from preconfigured or used figures/axes
- [x] Write README.md

.. |First plot| image:: https://raw.githubusercontent.com/OnionNinja/justplot/master/example/blueprint-1.png
.. |Second plot| image:: https://raw.githubusercontent.com/OnionNinja/justplot/master/example/blueprint-2.png
.. |Wind plot| image:: https://raw.githubusercontent.com/OnionNinja/justplot/master/example/windy.png

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

justplot-0.1.3a2.tar.gz (5.6 kB view hashes)

Uploaded Source

Supported by

AWS AWS Cloud computing and Security Sponsor Datadog Datadog Monitoring Fastly Fastly CDN Google Google Download Analytics Microsoft Microsoft PSF Sponsor Pingdom Pingdom Monitoring Sentry Sentry Error logging StatusPage StatusPage Status page