thinkplot
index
/home/downey/ThinkDSP/code/thinkplot.py

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

 
Modules
       
math
matplotlib
numpy
pandas
matplotlib.pyplot
warnings

 
Functions
       
Bar(xs, ys, **options)
Plots a line.
 
Args:
  xs: sequence of x values
  ys: sequence of y values
  options: keyword args passed to pyplot.bar
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
  Config, Show or Save.
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
the corresponding pyplot functions.
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.
FillBetween(xs, y1, y2=None, where=None, **options)
Fills the space between two lines.
 
Args:
  xs: sequence of x values
  y1: sequence of y values
  y2: sequence of y values
  where: sequence of boolean
  options: keyword args passed to pyplot.fill_between
HexBin(xs, ys, **options)
Makes a scatter plot.
 
xs: x values
ys: y values
options: options passed to pyplot.scatter
Hist(hist, **options)
Plots a Pmf or Hist with a bar plot.
 
The default width of the bars is based on the minimum difference
between values in the Hist.  If that's too small, you can override
it by providing a width keyword argument, in the same units
as the values.
 
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
Hlines(ys, x1, x2, **options)
Plots a set of horizontal lines.
 
Args:
  ys: sequence of y values
  x1: sequence of x values
  x2: sequence of x values
  options: keyword args passed to pyplot.vlines
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
Pdf(pdf, **options)
Plots a Pdf, Pmf, or Hist as a line.
 
Args:
  pdf: Pdf, Pmf, or Hist object
  options: keyword args passed to pyplot.plot
Pdfs(pdfs, **options)
Plots a sequence of PDFs.
 
Options are passed along for all PDFs.  If you want different
options for each pdf, make multiple calls to Pdf.
 
Args:
  pdfs: sequence of PDF objects
  options: keyword args passed to pyplot.plot
Plot(obj, ys=None, style='', **options)
Plots a line.
 
Args:
  obj: sequence of x values, or Series, or anything with Render()
  ys: sequence of y values
  style: style string passed along to pyplot.plot
  options: keyword args passed to pyplot.plot
Plotly(**options)
Shows the plot.
 
For options, see Config.
 
options: keyword args used to invoke various pyplot functions
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=None, cols=None)
Takes hints about what's coming.
 
num: number of lines that will be plotted
rows: number of rows of subplots
cols: number of columns of subplots
Save(root=None, formats=None, **options)
Saves the plot in the given formats and clears the figure.
 
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=None, **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, rows=None, cols=None, **options)
Configures the number of subplots and changes the current plot.
 
rows: int
cols: int
plot_number: int
options: passed to subplot
Text(x, y, s, **options)
Puts text in a figure.
 
x: number
y: number
s: string
options: keyword args passed to pyplot.text
Vlines(xs, y1, y2, **options)
Plots a set of vertical lines.
 
Args:
  xs: sequence of x values
  y1: sequence of y values
  y2: sequence of y values
  options: keyword args passed to pyplot.vlines
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
  Config, Show or Save.
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
the corresponding pyplot functions.
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.
fill_between = FillBetween(xs, y1, y2=None, where=None, **options)
Fills the space between two lines.
 
Args:
  xs: sequence of x values
  y1: sequence of y values
  y2: sequence of y values
  where: sequence of boolean
  options: keyword args passed to pyplot.fill_between
hist = Hist(hist, **options)
Plots a Pmf or Hist with a bar plot.
 
The default width of the bars is based on the minimum difference
between values in the Hist.  If that's too small, you can override
it by providing a width keyword argument, in the same units
as the values.
 
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
hlines = Hlines(ys, x1, x2, **options)
Plots a set of horizontal lines.
 
Args:
  ys: sequence of y values
  x1: sequence of x values
  x2: sequence of x values
  options: keyword args passed to pyplot.vlines
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(obj, ys=None, style='', **options)
Plots a line.
 
Args:
  obj: sequence of x values, or Series, or anything with Render()
  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=None, cols=None)
Takes hints about what's coming.
 
num: number of lines that will be plotted
rows: number of rows of subplots
cols: number of columns of subplots
save = Save(root=None, formats=None, **options)
Saves the plot in the given formats and clears the figure.
 
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=None, **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, rows=None, cols=None, **options)
Configures the number of subplots and changes the current plot.
 
rows: int
cols: int
plot_number: int
options: passed to subplot
text = Text(x, y, s, **options)
Puts text in a figure.
 
x: number
y: number
s: string
options: keyword args passed to pyplot.text
vlines = Vlines(xs, y1, y2, **options)
Plots a set of vertical lines.
 
Args:
  xs: sequence of x values
  y1: sequence of y values
  y2: sequence of y values
  options: keyword args passed to pyplot.vlines

 
Data
        LEGEND = True
LOC = None
print_function = _Feature((2, 6, 0, 'alpha', 2), (3, 0, 0, 'alpha', 0), 65536)