myplot
index
/home/downey/thinkstats/trunk/workspace.thinkstats/ThinkStats/myplot.py

This file contains code for use with "Think Stats",
by Allen B. Downey, available from greenteapress.com
 
Copyright 2010 Allen B. Downey
License: GNU GPLv3 http://www.gnu.org/licenses/gpl.html

 
Modules
       
math
matplotlib
numpy
matplotlib.pyplot

 
Classes
       
__builtin__.list(__builtin__.object)
InfiniteList
__builtin__.object
Brewer

 
class Brewer(__builtin__.object)
    Encapsulates a nice sequence of colors.
 
Shades of blue that look good in color and can be distinguished
in grayscale (up to a point).
 
Borrowed from http://colorbrewer2.org/
 
  Class methods defined here:
ClearIter(cls) from __builtin__.type
Sets the color iterator to None.
ColorGenerator(cls, n) from __builtin__.type
Returns an iterator of color strings.
 
n: how many colors will be used
Colors(cls) from __builtin__.type
Returns the list of colors.
GetIter(cls) from __builtin__.type
Gets the color iterator.
InitializeIter(cls, num) from __builtin__.type
Initializes the color iterator with the given number of colors.

Data descriptors defined here:
__dict__
dictionary for instance variables (if defined)
__weakref__
list of weak references to the object (if defined)

Data and other attributes defined here:
color_iter = None
colors = ['#081D58', '#253494', '#225EA8', '#1D91C0', '#41B6C4', '#7FCDBB', '#C7E9B4', '#EDF8B1', '#FFFFD9']
which_colors = [[], [1], [1, 3], [0, 2, 4], [0, 2, 4, 6], [0, 2, 3, 5, 6], [0, 2, 3, 4, 5, 6], [0, 1, 2, 3, 4, 5, 6]]

 
class InfiniteList(__builtin__.list)
    list that returns the same value for all indices.
 
 
Method resolution order:
InfiniteList
__builtin__.list
__builtin__.object

Methods defined here:
__getitem__(self, index)
Gets the item with the given index.
 
index: int
 
returns: the stored value
__init__(self, val)
Initializes the list.
 
val: value to be stored

Data descriptors defined here:
__dict__
dictionary for instance variables (if defined)
__weakref__
list of weak references to the object (if defined)

Methods inherited from __builtin__.list:
__add__(...)
x.__add__(y) <==> x+y
__contains__(...)
x.__contains__(y) <==> y in x
__delitem__(...)
x.__delitem__(y) <==> del x[y]
__delslice__(...)
x.__delslice__(i, j) <==> del x[i:j]
 
Use of negative indices is not supported.
__eq__(...)
x.__eq__(y) <==> x==y
__ge__(...)
x.__ge__(y) <==> x>=y
__getattribute__(...)
x.__getattribute__('name') <==> x.name
__getslice__(...)
x.__getslice__(i, j) <==> x[i:j]
 
Use of negative indices is not supported.
__gt__(...)
x.__gt__(y) <==> x>y
__iadd__(...)
x.__iadd__(y) <==> x+=y
__imul__(...)
x.__imul__(y) <==> x*=y
__iter__(...)
x.__iter__() <==> iter(x)
__le__(...)
x.__le__(y) <==> x<=y
__len__(...)
x.__len__() <==> len(x)
__lt__(...)
x.__lt__(y) <==> x<y
__mul__(...)
x.__mul__(n) <==> x*n
__ne__(...)
x.__ne__(y) <==> x!=y
__repr__(...)
x.__repr__() <==> repr(x)
__reversed__(...)
L.__reversed__() -- return a reverse iterator over the list
__rmul__(...)
x.__rmul__(n) <==> n*x
__setitem__(...)
x.__setitem__(i, y) <==> x[i]=y
__setslice__(...)
x.__setslice__(i, j, y) <==> x[i:j]=y
 
Use  of negative indices is not supported.
__sizeof__(...)
L.__sizeof__() -- size of L in memory, in bytes
append(...)
L.append(object) -- append object to end
count(...)
L.count(value) -> integer -- return number of occurrences of value
extend(...)
L.extend(iterable) -- extend list by appending elements from the iterable
index(...)
L.index(value, [start, [stop]]) -> integer -- return first index of value.
Raises ValueError if the value is not present.
insert(...)
L.insert(index, object) -- insert object before index
pop(...)
L.pop([index]) -> item -- remove and return item at index (default last).
Raises IndexError if list is empty or index is out of range.
remove(...)
L.remove(value) -- remove first occurrence of value.
Raises ValueError if the value is not present.
reverse(...)
L.reverse() -- reverse *IN PLACE*
sort(...)
L.sort(cmp=None, key=None, reverse=False) -- stable sort *IN PLACE*;
cmp(x, y) -> -1, 0, 1

Data and other attributes inherited from __builtin__.list:
__hash__ = None
__new__ = <built-in method __new__ of type object>
T.__new__(S, ...) -> a new object with type S, a subtype of T

 
Functions
       
Cdf(cdf, complement=False, transform=None, **options)
Plots a CDF as a line.
 
Args:
  cdf: Cdf object
  complement: boolean, whether to plot the complementary CDF
  transform: string, one of 'exponential', 'pareto', 'weibull', 'gumbel'
  options: keyword args passed to pyplot.plot
 
Returns:
  dictionary with the scale options that should be passed to
  myplot.Save or myplot.Show
Cdfs(cdfs, complement=False, transform=None, **options)
Plots a sequence of CDFs.
 
cdfs: sequence of CDF objects
complement: boolean, whether to plot the complementary CDF
transform: string, one of 'exponential', 'pareto', 'weibull', 'gumbel'
options: keyword args passed to pyplot.plot
Clf()
Clears the figure and any hints that have been set.
Config(**options)
Configures the plot.
 
Pulls options out of the option dictionary and passes them to
title, xlabel, ylabel, xscale, yscale, xticks, yticks, axis, legend,
and loc.
Contour(obj, pcolor=False, contour=True, imshow=False, **options)
Makes a contour plot.
 
d: map from (x, y) to z, or object that provides GetDict
pcolor: boolean, whether to make a pseudocolor plot
contour: boolean, whether to make a contour plot
imshow: boolean, whether to use pyplot.imshow
options: keyword args passed to pyplot.pcolor and/or pyplot.contour
Diff(t)
Compute the differences between adjacent elements in a sequence.
 
Args:
    t: sequence of number
 
Returns:
    sequence of differences (length one less than t)
Figure(**options)
Sets options for the current figure.
Hist(hist, **options)
Plots a Pmf or Hist with a bar plot.
 
Args:
  hist: Hist or Pmf object
  options: keyword args passed to pyplot.bar
Hists(hists, **options)
Plots two histograms as interleaved bar plots.
 
Options are passed along for all PMFs.  If you want different
options for each pmf, make multiple calls to Pmf.
 
Args:
  hists: list of two Hist or Pmf objects
  options: keyword args passed to pyplot.plot
Pcolor(xs, ys, zs, pcolor=True, contour=False, **options)
Makes a pseudocolor plot.
 
xs:
ys:
zs:
pcolor: boolean, whether to make a pseudocolor plot
contour: boolean, whether to make a contour plot
options: keyword args passed to pyplot.pcolor and/or pyplot.contour
Plot(xs, ys, style='', **options)
Plots a line.
 
Args:
  xs: sequence of x values
  ys: sequence of y values
  style: style string passed along to pyplot.plot
  options: keyword args passed to pyplot.plot
Pmf(pmf, **options)
Plots a Pmf or Hist as a line.
 
Args:
  pmf: Hist or Pmf object
  options: keyword args passed to pyplot.plot
Pmfs(pmfs, **options)
Plots a sequence of PMFs.
 
Options are passed along for all PMFs.  If you want different
options for each pmf, make multiple calls to Pmf.
 
Args:
  pmfs: sequence of PMF objects
  options: keyword args passed to pyplot.plot
PrePlot(num=None, rows=1, cols=1)
Takes hints about what's coming.
 
num: number of lines that will be plotted
Save(root=None, formats=None, **options)
Saves the plot in the given formats.
 
For options, see Config.
 
Args:
  root: string filename root
  formats: list of string formats
  options: keyword args used to invoke various pyplot functions
SaveFormat(root, fmt='eps')
Writes the current figure to a file in the given format.
 
Args:
  root: string filename root
  fmt: string format
Scatter(xs, ys, **options)
Makes a scatter plot.
 
xs: x values
ys: y values
options: options passed to pyplot.scatter
Show(**options)
Shows the plot.
 
For options, see Config.
 
options: keyword args used to invoke various pyplot functions
SubPlot(plot_number)
Underride(d, **options)
Add key-value pairs to d only if key is not in d.
 
If d is None, create a new dictionary.
 
d: dictionary
options: keyword args to add to d
cdf = Cdf(cdf, complement=False, transform=None, **options)
Plots a CDF as a line.
 
Args:
  cdf: Cdf object
  complement: boolean, whether to plot the complementary CDF
  transform: string, one of 'exponential', 'pareto', 'weibull', 'gumbel'
  options: keyword args passed to pyplot.plot
 
Returns:
  dictionary with the scale options that should be passed to
  myplot.Save or myplot.Show
cdfs = Cdfs(cdfs, complement=False, transform=None, **options)
Plots a sequence of CDFs.
 
cdfs: sequence of CDF objects
complement: boolean, whether to plot the complementary CDF
transform: string, one of 'exponential', 'pareto', 'weibull', 'gumbel'
options: keyword args passed to pyplot.plot
clf = Clf()
Clears the figure and any hints that have been set.
config = Config(**options)
Configures the plot.
 
Pulls options out of the option dictionary and passes them to
title, xlabel, ylabel, xscale, yscale, xticks, yticks, axis, legend,
and loc.
contour = Contour(obj, pcolor=False, contour=True, imshow=False, **options)
Makes a contour plot.
 
d: map from (x, y) to z, or object that provides GetDict
pcolor: boolean, whether to make a pseudocolor plot
contour: boolean, whether to make a contour plot
imshow: boolean, whether to use pyplot.imshow
options: keyword args passed to pyplot.pcolor and/or pyplot.contour
diff = Diff(t)
Compute the differences between adjacent elements in a sequence.
 
Args:
    t: sequence of number
 
Returns:
    sequence of differences (length one less than t)
figure = Figure(**options)
Sets options for the current figure.
hist = Hist(hist, **options)
Plots a Pmf or Hist with a bar plot.
 
Args:
  hist: Hist or Pmf object
  options: keyword args passed to pyplot.bar
hists = Hists(hists, **options)
Plots two histograms as interleaved bar plots.
 
Options are passed along for all PMFs.  If you want different
options for each pmf, make multiple calls to Pmf.
 
Args:
  hists: list of two Hist or Pmf objects
  options: keyword args passed to pyplot.plot
main()
pcolor = Pcolor(xs, ys, zs, pcolor=True, contour=False, **options)
Makes a pseudocolor plot.
 
xs:
ys:
zs:
pcolor: boolean, whether to make a pseudocolor plot
contour: boolean, whether to make a contour plot
options: keyword args passed to pyplot.pcolor and/or pyplot.contour
plot = Plot(xs, ys, style='', **options)
Plots a line.
 
Args:
  xs: sequence of x values
  ys: sequence of y values
  style: style string passed along to pyplot.plot
  options: keyword args passed to pyplot.plot
pmf = Pmf(pmf, **options)
Plots a Pmf or Hist as a line.
 
Args:
  pmf: Hist or Pmf object
  options: keyword args passed to pyplot.plot
pmfs = Pmfs(pmfs, **options)
Plots a sequence of PMFs.
 
Options are passed along for all PMFs.  If you want different
options for each pmf, make multiple calls to Pmf.
 
Args:
  pmfs: sequence of PMF objects
  options: keyword args passed to pyplot.plot
preplot = PrePlot(num=None, rows=1, cols=1)
Takes hints about what's coming.
 
num: number of lines that will be plotted
save = Save(root=None, formats=None, **options)
Saves the plot in the given formats.
 
For options, see Config.
 
Args:
  root: string filename root
  formats: list of string formats
  options: keyword args used to invoke various pyplot functions
scatter = Scatter(xs, ys, **options)
Makes a scatter plot.
 
xs: x values
ys: y values
options: options passed to pyplot.scatter
show = Show(**options)
Shows the plot.
 
For options, see Config.
 
options: keyword args used to invoke various pyplot functions
subplot = SubPlot(plot_number)