B ]t\;@szdZddlZddlZddlZddlmZddlZddlZddlZddl Z ddl m Z ddl Z ddl Z ddl Z ddl mZmZddl mZmZddlmZmZmZmZmZddl mZdd lmZdd lmZmZdd lmZdd l m Z m!Z!m"Z"m#Z#dd l m$Z$ddl%m&Z'ddl(m)Z)m*Z*m+Z+ddl(m,Z-ddl.m/Z/m0Z0ddl1m2Z2ddl m3Z3ddl4m5Z5m6Z6ddl m7Z7ddl8m9Z9m:Z:ddl;ZZ>ddl?m@Z@ddlAmBZBmCZCddlDmEZEmFZFmGZGmHZHddlImJZJmKZKmLZLmMZMddlNmOZOmPZPmQZQmRZRmSZSmTZTmUZUmVZVmWZWmXZXmYZYmZZZm[Z[m\Z\m]Z]m^Z^m_Z_m`Z`maZaddlbmcZcmdZdeeefZgdahdaid d!Zjd"d#ZkejljmZmene+jodd%d&Zod'd(Zpd)d*Zqd+d,Zrd-d.Zsd/d0Ztd1d2Zuene jvd3d4Zvene j$dd5d6Z$ene jwd7d8Zwd9d:Zxeye-d;d<Z,dd?d@Zzdddddd$edfdAdBZ{dCdDZ|dEdFZ}dGdHZ~dIdJZdKdLZdMdNZenejdOdPZenejdQdRZddSdTZdUdVZdWdXZenejdYdZZenejd[d\Zenejd]d^Zenejd_d`ZenejdadbZenejdcddZdedfZejddgdhZddidjZdkdlZdmdnZejdodpZddqdrZddsdtZddudvZddwdxZddydzZdd{d|Zdd~dZdddZddZddZdddZdddZddZddZddZddZddZddZdddZe jje_dddZddZene jjdddZene jjddZdddZddZdddZddZe dree de'kredree dejepe dejene/jddddZene/jdddddZene/jddZene/jddZene/jdddZene/jdddZene/jdddZene/jddZene/jdddZene/jdddÄZene/j¡ddddƜddȄZene/jáddddʄZene/jġddd˜dd̈́Zene/jšdddddτZene/jơddddфZene/jǡddӄZene/jȡddՄZene/jɡdd>de3je3jddddf ddddلZee/j̃ddddۄZee/j̓dddd݄Zene/jΡddddd߄Zene/jϡdddddZene/jСdddddZene/jѡddddZene/jҡdddddZene/jӡdddddZene/jԡdddZee/jՃdd=dddddddddddeZee/jddddAdBZene/jdCdDZene/jddEdFZene/jddGdHZene/jdddddddIdJdKZee/jdLdMZee/jdNdOZee/jdPdQZene/jdRdSZene/jddddUdVZene/jddddWdXZene/jd$e3jd$dfdddYdZZene/jd[d\Zene/jdÐd]d^Zene/jdĐd_d`Zene/jdŐdadbZene/jdcddZene/jdedfZ dgdhZ didjZ dkdlZ dmdnZ dodpZdqdrZdsdtZdudvZdwdxZdydzZd{d|Zd}d~ZddZddZddZddZddZddZddZedS(a `matplotlib.pyplot` is a state-based interface to matplotlib. It provides a MATLAB-like way of plotting. pyplot is mainly intended for interactive plots and simple cases of programmatic plot generation:: import numpy as np import matplotlib.pyplot as plt x = np.arange(0, 5, 0.1) y = np.sin(x) plt.plot(x, y) The object-oriented API is recommended for more complex plots. N)Number)cycler)rcsetupstyle)_pylab_helpers interactive)dedent deprecated silent_listwarn_deprecated_string_to_bool) docstring)FigureCanvasBase)Figure figaspect)GridSpec)rcParamsrcParamsDefault get_backend rcParamsOrig) rc_context)interactive_bk)getpgetArtist)setp)AxesSubplot) PolarAxes)mlab)get_scale_docsget_scale_names)cm)get_cmap register_cmap) Normalize)Line2D)Text Annotation)Polygon RectangleCircleArrow) SubplotToolButtonSliderWidget) TickHelper FormatterFixedFormatter NullFormatter FuncFormatterFormatStrFormatterScalarFormatter LogFormatterLogFormatterExponentLogFormatterMathtextLocator IndexLocator FixedLocator NullLocator LinearLocator LogLocator AutoLocatorMultipleLocator MaxNLocator) pylab_setup"_get_running_interactive_frameworkFc CsGdddt}ydtjkrddlm}|}|dkr<|trDdSdd}y|jd |Wntk r|| |YnX|ad a dd l m }| t}|r||nd a Wnt|fk rd a YnXdS) z Install a repl display hook so that any stale figure are automatically redrawn when control is returned to the repl. This works both with IPython and with vanilla python shells. c@s eZdZdS)z-install_repl_displayhook.._NotIPythonN)__name__ __module__ __qualname__rJrJ0lib/python3.7/site-packages/matplotlib/pyplot.py _NotIPython\srLIPythonr) get_ipythonNcSstrtdS)N) matplotlibis_interactivedraw_allrJrJrJrK post_executeksz.install_repl_displayhook..post_executerRF) backend2guiT) ExceptionsysmodulesrMrN_IP_REGISTEREDeventsregisterAttributeErrorZregister_post_execute_INSTALL_FIG_OBSERVERZIPython.core.pylabtoolsrSrrZ enable_gui ImportError)rLrNiprRrSZipython_gui_namerJrJrKinstall_repl_displayhookRs.      r^cCsVtrJddlm}|}y|jdtWntk rDtdYnXdatrRdadS)a Uninstall the matplotlib display hook. .. warning Need IPython >= 2 for this to work. For IPython < 2 will raise a ``NotImplementedError`` .. warning If you are using vanilla python and have installed another display hook this will reset ``sys.displayhook`` to what ever function was there when matplotlib installed it's displayhook, possibly discarding your changes. r)rNrRz*Can not unregister events in IPython < 2.0NF)rWrMrNrXZ unregisterrZNotImplementedErrorr[)rNr]rJrJrKuninstall_repl_displayhooks r`TcCs|dkrt}|j||dS)N) include_self)gcffindobj)omatchrarJrJrKrcsrcc Cstd|tjkrPxwYqX|td<dSqW|drf|ddn d|}t |}t dt j jft|}td ||j|j}|dk rt j }|r|r||krtd ||||td<td<|a|ja|ja|ja|t j _dS) a Close all open figures and set the Matplotlib backend. The argument is case-insensitive. Switching to an interactive backend is possible only if no event loop for another interactive backend has started. Switching to and from non-interactive backends is always possible. Parameters ---------- newbackend : str The name of the backend to use. all) ZmacosxZqt5aggZqt4aggZgtk3aggZ gtk3cairoZtkaggZwxaggZaggZcairobackendNz module:// zmatplotlib.backends.backend_{}BackendzLoaded backend %s version %s.zdCannot load backend {!r} which requires the {!r} interactive framework, as {!r} is currently running)closer_auto_backend_sentinelswitch_backendr\r startswithformatlower importlib import_moduletyperOZbackendsZ_Backendvars_logdebugZbackend_versionZrequired_interactive_frameworkrFrrZ _backend_modnew_figure_managerdraw_if_interactiveshow_showrg)Z newbackend candidateZ backend_nameZ backend_modriZrequired_frameworkZcurrent_frameworkrJrJrKrls<        rlcOs t||S)aL Display a figure. When running in ipython with its pylab mode, display all figures and return to the ipython prompt. In non-interactive mode, display all figures and block until the figures have been closed; in interactive mode it has no effect unless figures were created prior to a change from non-interactive to interactive mode (not recommended). In that case it displays the figures but does not block. A single experimental keyword argument, *block*, may be set to True or False to override the blocking behavior described above. )ry)argskwrJrJrKrxsrxcCstS)z&Return the status of interactive mode.)rOrPrJrJrJrK isinteractivesr}cCstdtdS)zTurn the interactive mode off.FN)rOrr`rJrJrJrKioffs r~cCstdtdS)zTurn the interactive mode on.TN)rOrr^rJrJrJrKion s rcCsLtj}|dk r>|j}|jjr(|tdd||n t |dS)a Pause for *interval* seconds. If there is an active figure, it will be updated and displayed before the pause, and the GUI event loop (if any) will run during the pause. This can be used for crude animation. For more complex animation, see :mod:`matplotlib.animation`. Notes ----- This function is experimental; its behavior may be changed or extended in a future release. NF)block) rGcf get_activecanvasfigureZstale draw_idlerxZstart_event_looptimeZsleep)ZintervalmanagerrrJrJrKpauses   rcKstj|f|dS)N)rOrc)groupkwargsrJrJrKr,srcCs t||S)N)rOr)rfnamerJrJrKr1srcCsttrtdS)N)rO rcdefaultsrPrQrJrJrJrKr6srcCs tS)a Get the current colorable artist. Specifically, returns the current :class:`~matplotlib.cm.ScalarMappable` instance (image or patch collection), or *None* if no images or patch collections have been defined. The commands :func:`~matplotlib.pyplot.imshow` and :func:`~matplotlib.pyplot.figimage` create :class:`~matplotlib.image.Image` instances, and the commands :func:`~matplotlib.pyplot.pcolor` and :func:`~matplotlib.pyplot.scatter` create :class:`~matplotlib.collections.Collection` instances. The current image is an attribute of the current axes, or the nearest earlier axes in the current figure that contains an image. )rbZ_gcirJrJrJrKgci@srcOst|f||S)N)_setp)objr{rrJrJrKrUsrdcCs^tdrtdddlm}tddddgd |||f|jd d d gd dddddddddddS)a Turn on `xkcd `_ sketch-style drawing mode. This will only have effect on things drawn after this function is called. For best results, the "Humor Sans" font should be installed: it is not included with matplotlib. Parameters ---------- scale : float, optional The amplitude of the wiggle perpendicular to the source line. length : float, optional The length of the wiggle along the line. randomness : float, optional The scale factor by which the length is shrunken or expanded. Notes ----- This function works by a number of rcParams, so it will probably override others you have set before. If you want the effects of this function to be temporary, it can be used as a context manager, for example:: with plt.xkcd(): # This figure will be in XKCD-style fig1 = plt.figure() # ... # This figure will be in regular style fig2 = plt.figure() z text.usetexz3xkcd mode is not compatible with text.usetex = Truer) patheffectsxkcdz xkcd Scriptz Humor Sansz Comic Sans MSg,@w) linewidthZ foregroundg?g@ZwhitegFZblack)z font.familyz font.sizez path.sketchz path.effectszaxes.linewidthzlines.linewidthzfigure.facecolorzgrid.linewidthz axes.gridzaxes.unicode_minuszaxes.edgecolorzxtick.major.sizezxtick.major.widthzytick.major.sizezytick.major.width)r RuntimeErrorrOrrZ withStroke)scalelengthZ randomnessrrJrJrKrZs("  rc s|dkrtd}|dkr td}|dkr0td}|dkr@td}t} | rVt| dnd} d} |dkrl| }nPt|tr|} t} | | kr| dkrtd | }q| | } | | }nt |}t j |dkrtd }t | |krdkrnntd |ttd krd }t|f||||||d|| rX| jj| fdd}jd|}|_t j jj}||_trttrt|_|rjj jjS)a# Create a new figure. Parameters ---------- num : integer or string, optional, default: None If not provided, a new figure will be created, and the figure number will be incremented. The figure objects holds this number in a `number` attribute. If num is provided, and a figure with this id already exists, make it active, and returns a reference to it. If this figure does not exists, create it and returns it. If num is a string, the window title will be set to this figure's `num`. figsize : (float, float), optional, default: None width, height in inches. If not provided, defaults to :rc:`figure.figsize` = ``[6.4, 4.8]``. dpi : integer, optional, default: None resolution of the figure. If not provided, defaults to :rc:`figure.dpi` = ``100``. facecolor : the background color. If not provided, defaults to :rc:`figure.facecolor` = ``'w'``. edgecolor : the border color. If not provided, defaults to :rc:`figure.edgecolor` = ``'w'``. frameon : bool, optional, default: True If False, suppress drawing the figure frame. FigureClass : subclass of `~matplotlib.figure.Figure` Optionally use a custom `.Figure` instance. clear : bool, optional, default: False If True and the figure already exists, then it is cleared. Returns ------- figure : `~matplotlib.figure.Figure` The `.Figure` instance returned will also be passed to new_figure_manager in the backends, which allows to hook custom `.Figure` classes into the pyplot interface. Additional kwargs will be passed to the `.Figure` init function. Notes ----- If you are creating many figures, make sure you explicitly call :func:`.pyplot.close` on the figures you are not using, because this will enable pyplot to properly clean up the memory. `~matplotlib.rcParams` defines the default values, which can be modified in the matplotlibrc file. Nzfigure.figsizez figure.dpizfigure.facecolorzfigure.edgecolorr1rfz(close('all') closes all existing figureszfigure.max_open_warningzMore than %d figures have been opened. Figures created through the pyplot interface (`matplotlib.pyplot.figure`) are retained until explicitly closed and may consume too much memory. (To control this warning, see the rcParam `figure.max_open_warning`).ZpsH)figsizedpi facecolor edgecolorframeon FigureClasscstjdS)N)rr set_active)Zevent) figManagerrJrK make_activeszfigure..make_activeZbutton_press_event)!r get_fignumsmax isinstancestr get_figlabelswarningswarnindexintrrZget_fig_managerlenRuntimeWarningrrorvZset_window_titlerrZ set_label mpl_connectZ_cidgcfrZnumberrOrPrwr[_auto_draw_if_interactiveZstale_callbackclear)numrrrrrrrrZallnumsZnext_numZfigLabel allLabelsZinumZmax_open_warningrcidfigrJ)rrKrsnE            rcCs$|r tr |js |jdS)z This is an internal helper function for making sure that auto-redrawing works as intended in the plain python repl. Parameters ---------- fig : Figure A figure object which is assumed to be associated with a canvas N)rOrPrZ is_savingr)rvalrJrJrKr.s rcCs$tj}|dk r|jjStSdS)z&Get a reference to the current figure.N)rrrrr)rrJrJrKrb<s rbcCstj|p|tkS)z3Return whether the figure with the given id exists.)rrZ has_fignumr)rrJrJrK fignum_existsEsrcCs ttjjS)z)Return a list of existing figure numbers.)sortedrrZfigsrJrJrJrKrJsrcCs(tj}|jddddd|DS)z(Return a list of existing figure labels.cSs|jS)N)r)mrJrJrKRszget_figlabels..)keycSsg|]}|jjqSrJ)rrZ get_label).0rrJrJrK Ssz!get_figlabels..)rrget_all_fig_managerssort)Z figManagersrJrJrKrOs rcCs&tj}|dkr"ttj}|S)z| Return the figure manager of the active figure. If there is currently no active figure, a new one is created. N)rrrrb)rrJrJrKget_current_fig_managerVs   rcCstj||S)N)rrr)sfuncrJrJrKconnectcsrcCstj|S)N)rrmpl_disconnect)rrJrJrK disconnecthsrcCs|dkr.tj}|dkrdStj|jn|dkrBtjnt|trZtj|npt|drttj|jnVt|t rt }||krt | |}tj|n t|t rtj|ntddS)al Close a figure window. Parameters ---------- fig : None or int or str or `.Figure` The figure to close. There are a number of ways to specify this: - *None*: the current figure - `.Figure`: the given `.Figure` instance - ``int``: a figure number - ``str``: a figure name - 'all': all figures NrfrzFclose() argument must be a Figure, an int, a string, or None, not '%s')rrrZdestroyrZ destroy_allrrhasattrrrrrrZ destroy_fig TypeError)rrrrrJrJrKrjms&      rjcCstdS)zClear the current figure.N)rbclfrJrJrJrKrsrcCstjdS)a+Redraw the current figure. This is used to update a figure that has been altered, but not automatically re-drawn. If interactive mode is on (:func:`.ion()`), this should be only rarely needed, but there may be ways to modify the state of a figure without marking it as `stale`. Please report these cases as bugs. A more object-oriented alternative, given any :class:`~matplotlib.figure.Figure` instance, :attr:`fig`, that was created using a :mod:`~matplotlib.pyplot` function, is:: fig.canvas.draw_idle() N)rrrrJrJrJrKdrawsrcOs t}|j||}|j|S)N)rbsavefigrr)r{rrZresrJrJrKrs  rcOstj||S)z Blocking call to interact with the figure. This will wait for *n* clicks from the user and return a list of the coordinates of each click. If *timeout* is negative, does not timeout. )rbginput)r{rrJrJrKrs rcOstj||S)a Blocking call to interact with the figure. This will wait for *n* key or mouse clicks from the user and return a list containing True's for keyboard clicks and False's for mouse clicks. If *timeout* is negative, does not timeout. )rbwaitforbuttonpress)r{rrJrJrKrs rcOstj|||f||S)N)rbtext)xyrr{rrJrJrKfigtextsrcKstj|f|S)N)rbsuptitle)trrJrJrKrsrcOstj||S)N)rbfigimage)r{rrJrJrKrsrcOstj||S)a Place a legend in the figure. *labels* a sequence of strings *handles* a sequence of :class:`~matplotlib.lines.Line2D` or :class:`~matplotlib.patches.Patch` instances *loc* can be a string or an integer specifying the legend location A :class:`matplotlib.legend.Legend` instance is returned. Examples -------- To make a legend from existing artists on every axes:: figlegend() To make a legend for a list of lines and labels:: figlegend( (line1, line2, line3), ('label1', 'label2', 'label3'), 'upper right' ) .. seealso:: :func:`~matplotlib.pyplot.legend` )rblegend)r{rrJrJrK figlegends#rcKsP|dkrtd|St|tr8tddd|}t||S|}tj|f|SdS)a Add an axes to the current figure and make it the current axes. Call signatures:: plt.axes() plt.axes(rect, projection=None, polar=False, **kwargs) plt.axes(ax) Parameters ---------- arg : { None, 4-tuple, Axes } The exact behavior of this function depends on the type: - *None*: A new full window axes is added using ``subplot(111, **kwargs)`` - 4-tuple of floats *rect* = ``[left, bottom, width, height]``. A new axes is added with dimensions *rect* in normalized (0, 1) units using `~.Figure.add_axes` on the current figure. - `~.axes.Axes`: This is equivalent to `.pyplot.sca`. It sets the current axes to *arg*. Note: This implicitly changes the current figure to the parent of *arg*. .. note:: The use of an `.axes.Axes` as an argument is deprecated and will be removed in v3.0. Please use `.pyplot.sca` instead. projection : {None, 'aitoff', 'hammer', 'lambert', 'mollweide', 'polar', 'rectilinear', str}, optional The projection type of the `~.axes.Axes`. *str* is the name of a costum projection, see `~matplotlib.projections`. The default None results in a 'rectilinear' projection. polar : boolean, optional If True, equivalent to projection='polar'. sharex, sharey : `~.axes.Axes`, optional Share the x or y `~matplotlib.axis` with sharex and/or sharey. The axis will have the same limits, ticks, and scale as the axis of the shared axes. label : str A label for the returned axes. Other Parameters ---------------- **kwargs This method also takes the keyword arguments for the returned axes class. The keyword arguments for the rectilinear axes class `~.axes.Axes` can be found in the following table but there might also be other keyword arguments if another projection is used, see the actual axes class. %(Axes)s Returns ------- axes : `~.axes.Axes` (or a subclass of `~.axes.Axes`) The returned axes class depends on the projection used. It is `~.axes.Axes` if rectilinear projection are used and `.projections.polar.PolarAxes` if polar projection are used. Notes ----- If the figure already has a axes with key (*args*, *kwargs*) then it will simply make that axes current and return it. This behavior is deprecated. Meanwhile, if you do not want this behavior (i.e., you want to force the creation of a new axes), you must use a unique set of args and kwargs. The axes *label* attribute has been exposed for this purpose: if you want two axes that are otherwise identical to be added to the figure, make sure you give them unique labels. See Also -------- .Figure.add_axes .pyplot.subplot .Figure.add_subplot .Figure.subplots .pyplot.subplots Examples -------- :: #Creating a new full window axes plt.axes() #Creating a new axes with specified dimensions and some kwargs plt.axes((left, bottom, width, height), facecolor='w') Noz2.2z`Using pyplot.axes(ax) with ax an Axes argument is deprecated. Please use pyplot.sca(ax) instead.)message)r)subplotrrr scarbadd_axes)argraxrectrJrJrKaxes s`  rcCs|dkrt}|j|dS)z Remove the `Axes` *ax* (defaulting to the current axes) from its figure. A KeyError is raised if the axes doesn't exist. N)gcardelaxes)rrJrJrKr{srcCsPtj}x8|D]0}||jjjkrtj||jj|dSqWtddS)zj Set the current Axes instance to *ax*. The current Figure is updated to the parent of *ax*. Nz0Axes instance argument was not found in a figure) rrrrrrrr ValueError)rZmanagersrrJrJrKrs   rcKstjf|S)a Get the current :class:`~matplotlib.axes.Axes` instance on the current figure matching the given keyword args, or create one. Examples -------- To get the current polar axes on the current figure:: plt.gca(projection='polar') If the current axes doesn't exist, or isn't a polar one, the appropriate axes will be created and then returned. See Also -------- matplotlib.figure.Figure.gca : The figure's gca method. )rbr)rrJrJrKrsrcOst|dkrd}t|dkr4t|dtr4tdt}|j||}|j}g}x.|jD]$}||krfqX| |jrX| |qXWx|D] }t |qW|S)ak Add a subplot to the current figure. Wrapper of `.Figure.add_subplot` with a difference in behavior explained in the notes section. Call signatures:: subplot(nrows, ncols, index, **kwargs) subplot(pos, **kwargs) subplot(ax) Parameters ---------- *args Either a 3-digit integer or three separate integers describing the position of the subplot. If the three integers are *nrows*, *ncols*, and *index* in order, the subplot will take the *index* position on a grid with *nrows* rows and *ncols* columns. *index* starts at 1 in the upper left corner and increases to the right. *pos* is a three digit integer, where the first digit is the number of rows, the second the number of columns, and the third the index of the subplot. i.e. fig.add_subplot(235) is the same as fig.add_subplot(2, 3, 5). Note that all integers must be less than 10 for this form to work. projection : {None, 'aitoff', 'hammer', 'lambert', 'mollweide', 'polar', 'rectilinear', str}, optional The projection type of the subplot (`~.axes.Axes`). *str* is the name of a costum projection, see `~matplotlib.projections`. The default None results in a 'rectilinear' projection. polar : boolean, optional If True, equivalent to projection='polar'. sharex, sharey : `~.axes.Axes`, optional Share the x or y `~matplotlib.axis` with sharex and/or sharey. The axis will have the same limits, ticks, and scale as the axis of the shared axes. label : str A label for the returned axes. Other Parameters ---------------- **kwargs This method also takes the keyword arguments for the returned axes base class. The keyword arguments for the rectilinear base class `~.axes.Axes` can be found in the following table but there might also be other keyword arguments if another projection is used. %(Axes)s Returns ------- axes : an `.axes.SubplotBase` subclass of `~.axes.Axes` (or a subclass of `~.axes.Axes`) The axes of the subplot. The returned axes base class depends on the projection used. It is `~.axes.Axes` if rectilinear projection are used and `.projections.polar.PolarAxes` if polar projection are used. The returned axes is then a subplot subclass of the base class. Notes ----- Creating a subplot will delete any pre-existing subplot that overlaps with it beyond sharing a boundary:: import matplotlib.pyplot as plt # plot a line, implicitly creating a subplot(111) plt.plot([1,2,3]) # now create a subplot which represents the top plot of a grid # with 2 rows and 1 column. Since this subplot will overlap the # first, the plot (and its axes) previously created, will be removed plt.subplot(211) If you do not want this behavior, use the `.Figure.add_subplot` method or the `.pyplot.axes` function instead. If the figure already has a subplot with key (*args*, *kwargs*) then it will simply make that subplot current and return it. This behavior is deprecated. Meanwhile, if you do not want this behavior (i.e., you want to force the creation of a new suplot), you must use a unique set of args and kwargs. The axes *label* attribute has been exposed for this purpose: if you want two subplots that are otherwise identical to be added to the figure, make sure you give them unique labels. In rare circumstances, `.add_subplot` may be called with a single argument, a subplot axes instance already created in the present figure but not in the figure's list of axes. See Also -------- .Figure.add_subplot .pyplot.subplots .pyplot.axes .Figure.subplots Examples -------- :: plt.subplot(221) # equivalent but more general ax1=plt.subplot(2, 2, 1) # add a subplot with no frame ax2=plt.subplot(222, frameon=False) # add a polar subplot plt.subplot(223, projection='polar') # add a red subplot that shares the x-axis with ax1 plt.subplot(224, sharex=ax1, facecolor='red') #delete ax2 from the figure plt.delaxes(ax2) #add ax2 to the figure again plt.subplot(ax2) r)r1r1r1rrzbThe subplot index argument to subplot() appears to be a boolean. Did you intend to use subplots()?) rrboolrrrb add_subplotbboxrfully_overlapsappendr)r{rrarbyebyeotherrrJrJrKrs"       rc Ks*tf|}|j|||||||d} || fS)a. Create a figure and a set of subplots. This utility wrapper makes it convenient to create common layouts of subplots, including the enclosing figure object, in a single call. Parameters ---------- nrows, ncols : int, optional, default: 1 Number of rows/columns of the subplot grid. sharex, sharey : bool or {'none', 'all', 'row', 'col'}, default: False Controls sharing of properties among x (`sharex`) or y (`sharey`) axes: - True or 'all': x- or y-axis will be shared among all subplots. - False or 'none': each subplot x- or y-axis will be independent. - 'row': each subplot row will share an x- or y-axis. - 'col': each subplot column will share an x- or y-axis. When subplots have a shared x-axis along a column, only the x tick labels of the bottom subplot are created. Similarly, when subplots have a shared y-axis along a row, only the y tick labels of the first column subplot are created. To later turn other subplots' ticklabels on, use `~matplotlib.axes.Axes.tick_params`. squeeze : bool, optional, default: True - If True, extra dimensions are squeezed out from the returned array of `~matplotlib.axes.Axes`: - if only one subplot is constructed (nrows=ncols=1), the resulting single Axes object is returned as a scalar. - for Nx1 or 1xM subplots, the returned object is a 1D numpy object array of Axes objects. - for NxM, subplots with N>1 and M>1 are returned as a 2D array. - If False, no squeezing at all is done: the returned Axes object is always a 2D array containing Axes instances, even if it ends up being 1x1. num : integer or string, optional, default: None A `.pyplot.figure` keyword that sets the figure number or label. subplot_kw : dict, optional Dict with keywords passed to the `~matplotlib.figure.Figure.add_subplot` call used to create each subplot. gridspec_kw : dict, optional Dict with keywords passed to the `~matplotlib.gridspec.GridSpec` constructor used to create the grid the subplots are placed on. **fig_kw : All additional keyword arguments are passed to the `.pyplot.figure` call. Returns ------- fig : `~.figure.Figure` ax : `.axes.Axes` object or array of Axes objects. *ax* can be either a single `~matplotlib.axes.Axes` object or an array of Axes objects if more than one subplot was created. The dimensions of the resulting array can be controlled with the squeeze keyword, see above. Examples -------- :: #First create some toy data: x = np.linspace(0, 2*np.pi, 400) y = np.sin(x**2) #Creates just a figure and only one subplot fig, ax = plt.subplots() ax.plot(x, y) ax.set_title('Simple plot') #Creates two subplots and unpacks the output array immediately f, (ax1, ax2) = plt.subplots(1, 2, sharey=True) ax1.plot(x, y) ax1.set_title('Sharing Y axis') ax2.scatter(x, y) #Creates four polar axes, and accesses them through the returned array fig, axes = plt.subplots(2, 2, subplot_kw=dict(polar=True)) axes[0, 0].plot(x, y) axes[1, 1].scatter(x, y) #Share a X axis with each column of subplots plt.subplots(2, 2, sharex='col') #Share a Y axis with each row of subplots plt.subplots(2, 2, sharey='row') #Share both X and Y axes with all subplots plt.subplots(2, 2, sharex='all', sharey='all') #Note that this is the same as plt.subplots(2, 2, sharex=True, sharey=True) #Creates figure number 10 with a single subplot #and clears it if it already exists. fig, ax=plt.subplots(num=10, clear=True) See Also -------- .pyplot.figure .pyplot.subplot .pyplot.axes .Figure.subplots .Figure.add_subplot )nrowsncolssharexshareysqueeze subplot_kw gridspec_kw)rsubplots) rrrrrrrZfig_kwrZaxsrJrJrKrJs w  rcKs|dkrt}|\}}t||j|||d}|j|f|} | j} g} x.|jD]$} | | krZqL| | jrL| | qLWx| D] } t| qzW| S)a Create an axis at specific location inside a regular grid. Parameters ---------- shape : sequence of 2 ints Shape of grid in which to place axis. First entry is number of rows, second entry is number of columns. loc : sequence of 2 ints Location to place axis within grid. First entry is row number, second entry is column number. rowspan : int Number of rows for the axis to span to the right. colspan : int Number of columns for the axis to span downwards. fig : `Figure`, optional Figure to place axis in. Defaults to current figure. **kwargs Additional keyword arguments are handed to `add_subplot`. Notes ----- The following call :: subplot2grid(shape, loc, rowspan=1, colspan=1) is identical to :: gridspec=GridSpec(shape[0], shape[1]) subplotspec=gridspec.new_subplotspec(loc, rowspan, colspan) subplot(subplotspec) N)rowspancolspan) rbrZnew_subplotspecrrrrrr)shapelocrrrrs1s2Z subplotspecrrrrrrJrJrK subplot2grids"(     rcCs|dkrt}|}|S)a1 Make a second axes that shares the *x*-axis. The new axes will overlay *ax* (or the current axes if *ax* is *None*). The ticks for *ax2* will be placed on the right, and the *ax2* instance is returned. .. seealso:: :doc:`/gallery/subplots_axes_and_figures/two_scales` N)rtwinx)rax1rJrJrKrs rcCs|dkrt}|}|S)z Make a second axes that shares the *y*-axis. The new axis will overlay *ax* (or the current axes if *ax* is *None*). The ticks for *ax2* will be placed on the top, and the *ax2* instance is returned. N)rtwiny)rrrJrJrKrsrcCst}|||||||dS)a Tune the subplot layout. The parameter meanings (and suggested defaults) are:: left = 0.125 # the left side of the subplots of the figure right = 0.9 # the right side of the subplots of the figure bottom = 0.1 # the bottom of the subplots of the figure top = 0.9 # the top of the subplots of the figure wspace = 0.2 # the amount of width reserved for space between subplots, # expressed as a fraction of the average axis width hspace = 0.2 # the amount of height reserved for space between subplots, # expressed as a fraction of the average axis height The actual defaults are controlled by the rc file N)rbsubplots_adjust)leftbottomrighttopZwspaceZhspacerrJrJrKr$srcCstd}dtd<|dkr(t}|jj}n(x&tjjD]}|jj|kr2Pq2Wtdtdd}|jddt ||}|td<tj ||S) zy Launch a subplot tool window for a figure. A :class:`matplotlib.widgets.SubplotTool` instance is returned. ZtoolbarNoneNz$Could not find manager for targetfig)r)rg?)r) rrrrrrZ _activeQuerrr-r)Z targetfigZtbarrZtoolfigretrJrJrK subplot_tool:s      rHzG?cCstj||||ddS)ay Automatically adjust subplot parameters to give specified padding. Parameters ---------- pad : float Padding between the figure edge and the edges of subplots, as a fraction of the font size. h_pad, w_pad : float, optional Padding (height/width) between edges of adjacent subplots, as a fraction of the font size. Defaults to *pad*. rect : tuple (left, bottom, right, top), optional A rectangle (left, bottom, right, top) in the normalized figure coordinate that the whole subplots area (including labels) will fit into. Default is (0, 0, 1, 1). )padh_padw_padrN)rb tight_layout)r r r rrJrJrKr Usr cCs.t}|dkr| }t|}||dS)a@ Turn the axes box on or off on the current axes. Parameters ---------- on : bool or None The new `~matplotlib.axes.Axes` box state. If ``None``, toggle the state. See Also -------- :meth:`matplotlib.axes.Axes.set_frame_on` :meth:`matplotlib.axes.Axes.get_frame_on` N)rZ get_frame_onr Z set_frame_on)ZonrrJrJrKboxis  rcOs&t}|s|s|S|j||}|S)a Get or set the x limits of the current axes. Call signatures:: left, right = xlim() # return the current xlim xlim((left, right)) # set the xlim to left, right xlim(left, right) # set the xlim to left, right If you do not specify args, you can pass *left* or *right* as kwargs, i.e.:: xlim(right=3) # adjust the right leaving left unchanged xlim(left=1) # adjust the left leaving right unchanged Setting limits turns autoscaling off for the x-axis. Returns ------- left, right A tuple of the new x-axis limits. Notes ----- Calling this function with no arguments (e.g. ``xlim()``) is the pyplot equivalent of calling `~.Axes.get_xlim` on the current axes. Calling this function with arguments is the pyplot equivalent of calling `~.Axes.set_xlim` on the current axes. All arguments are passed though. )rZget_xlimZset_xlim)r{rrrrJrJrKxlims  rcOs&t}|s|s|S|j||}|S)a Get or set the y-limits of the current axes. Call signatures:: bottom, top = ylim() # return the current ylim ylim((bottom, top)) # set the ylim to bottom, top ylim(bottom, top) # set the ylim to bottom, top If you do not specify args, you can alternatively pass *bottom* or *top* as kwargs, i.e.:: ylim(top=3) # adjust the top leaving bottom unchanged ylim(bottom=1) # adjust the bottom leaving top unchanged Setting limits turns autoscaling off for the y-axis. Returns ------- bottom, top A tuple of the new y-axis limits. Notes ----- Calling this function with no arguments (e.g. ``ylim()``) is the pyplot equivalent of calling `~.Axes.get_ylim` on the current axes. Calling this function with arguments is the pyplot equivalent of calling `~.Axes.set_ylim` on the current axes. All arguments are passed though. )rZget_ylimZset_ylim)r{rrrrJrJrKylims  rcKst}|dkr(|dkr(|}|}n4|dkrD||}|}n||}|j|f|}x|D]}||qbW|td|fS)a Get or set the current tick locations and labels of the x-axis. Call signatures:: locs, labels = xticks() # Get locations and labels xticks(ticks, [labels], **kwargs) # Set locations and labels Parameters ---------- ticks : array_like A list of positions at which ticks should be placed. You can pass an empty list to disable xticks. labels : array_like, optional A list of explicit labels to place at the given *locs*. **kwargs :class:`.Text` properties can be used to control the appearance of the labels. Returns ------- locs An array of label locations. labels A list of `.Text` objects. Notes ----- Calling this function with no arguments (e.g. ``xticks()``) is the pyplot equivalent of calling `~.Axes.get_xticks` and `~.Axes.get_xticklabels` on the current axes. Calling this function with arguments is the pyplot equivalent of calling `~.Axes.set_xticks` and `~.Axes.set_xticklabels` on the current axes. Examples -------- Get the current locations and labels: >>> locs, labels = xticks() Set label locations: >>> xticks(np.arange(0, 1, step=0.2)) Set text labels: >>> xticks(np.arange(5), ('Tom', 'Dick', 'Harry', 'Sally', 'Sue')) Set text labels and properties: >>> xticks(np.arange(12), calendar.month_name[1:13], rotation=20) Disable xticks: >>> xticks([]) NzText xticklabel)rZ get_xticksZget_xticklabelsZ set_xticksZset_xticklabelsupdater )tickslabelsrrlocslrJrJrKxtickss<     rcKst}|dkr(|dkr(|}|}n4|dkrD||}|}n||}|j|f|}x|D]}||qbW|td|fS)a Get or set the current tick locations and labels of the y-axis. Call signatures:: locs, labels = yticks() # Get locations and labels yticks(ticks, [labels], **kwargs) # Set locations and labels Parameters ---------- ticks : array_like A list of positions at which ticks should be placed. You can pass an empty list to disable yticks. labels : array_like, optional A list of explicit labels to place at the given *locs*. **kwargs :class:`.Text` properties can be used to control the appearance of the labels. Returns ------- locs An array of label locations. labels A list of `.Text` objects. Notes ----- Calling this function with no arguments (e.g. ``yticks()``) is the pyplot equivalent of calling `~.Axes.get_yticks` and `~.Axes.get_yticklabels` on the current axes. Calling this function with arguments is the pyplot equivalent of calling `~.Axes.set_yticks` and `~.Axes.set_yticklabels` on the current axes. Examples -------- Get the current locations and labels: >>> locs, labels = yticks() Set label locations: >>> yticks(np.arange(0, 1, step=0.2)) Set text labels: >>> yticks(np.arange(5), ('Tom', 'Dick', 'Harry', 'Sally', 'Sue')) Set text labels and properties: >>> yticks(np.arange(12), calendar.month_name[1:13], rotation=45) Disable yticks: >>> yticks([]) NzText yticklabel)rZ get_yticksZget_yticklabelsZ set_yticksZset_yticklabelsrr )rrrrrrrJrJrKytickss<     rcOs^t}t|tstdt|dkr:|j}|j}n|j||\}}t d|t d|fS)a  Get or set the radial gridlines on the current polar plot. Call signatures:: lines, labels = rgrids() lines, labels = rgrids(radii, labels=None, angle=22.5, fmt=None, **kwargs) When called with no arguments, `.rgrids` simply returns the tuple (*lines*, *labels*). When called with arguments, the labels will appear at the specified radial distances and angle. Parameters ---------- radii : tuple with floats The radii for the radial gridlines labels : tuple with strings or None The labels to use at each radial gridline. The `matplotlib.ticker.ScalarFormatter` will be used if None. angle : float The angular position of the radius labels in degrees. fmt : str or None Format string used in `matplotlib.ticker.FormatStrFormatter`. For example '%f'. Returns ------- lines, labels : list of `.lines.Line2D`, list of `.text.Text` *lines* are the radial gridlines and *labels* are the tick labels. Other Parameters ---------------- **kwargs *kwargs* are optional `~.Text` properties for the labels. Examples -------- :: # set the locations of the radial gridlines lines, labels = rgrids( (0.25, 0.5, 1.0) ) # set the locations and labels of the radial gridlines lines, labels = rgrids( (0.25, 0.5, 1.0), ('Tom', 'Dick', 'Harry' )) See Also -------- .pyplot.thetagrids .projections.polar.PolarAxes.set_rgrids .Axis.get_gridlines .Axis.get_ticklabels z"rgrids only defined for polar axesrzLine2D rgridlinezText rgridlabel) rrrrrZyaxisZ get_gridlinesget_ticklabelsZ set_rgridsr )r{rrlinesrrJrJrKrgridses:    rcOs^t}t|tstdt|dkr:|j}|j}n|j||\}}t d|t d|fS)a Get or set the theta gridlines on the current polar plot. Call signatures:: lines, labels = thetagrids() lines, labels = thetagrids(angles, labels=None, fmt=None, **kwargs) When called with no arguments, `.thetagrids` simply returns the tuple (*lines*, *labels*). When called with arguments, the labels will appear at the specified angles. Parameters ---------- angles : tuple with floats, degrees The angles of the theta gridlines. labels : tuple with strings or None The labels to use at each radial gridline. The `.projections.polar.ThetaFormatter` will be used if None. fmt : str or None Format string used in `matplotlib.ticker.FormatStrFormatter`. For example '%f'. Note that the angle in radians will be used. Returns ------- lines, labels : list of `.lines.Line2D`, list of `.text.Text` *lines* are the theta gridlines and *labels* are the tick labels. Other Parameters ---------------- **kwargs *kwargs* are optional `~.Text` properties for the labels. Examples -------- :: # set the locations of the angular gridlines lines, labels = thetagrids( range(45,360,90) ) # set the locations and labels of the angular gridlines lines, labels = thetagrids( range(45,360,90), ('NE', 'NW', 'SW','SE') ) See Also -------- .pyplot.rgrids .projections.polar.PolarAxes.set_thetagrids .Axis.get_gridlines .Axis.get_ticklabels z&thetagrids only defined for polar axesrzLine2D thetagridlinezText thetagridlabel) rrrrrZxaxisZ get_ticklinesrZset_thetagridsr )r{rrrrrJrJrK thetagridss5    rcCsdS)NrJrJrJrJrKplottingsrc sLddddddddd h ttOtttfd d tDS) z< Get a sorted list of all of the plotting commands. colormapscolorsrrget_plot_commandsrrrrc3s@|]8\}}|ds|krt|rt|kr|VqdS)_N)rminspectZ isfunction getmodule)rnamer)exclude this_modulerJrK s z$get_plot_commands..)setrr!r"rrglobalsitemsrJrJ)r$r%rKrs rcCs ttjS)a|+ Matplotlib provides a number of colormaps, and others can be added using :func:`~matplotlib.cm.register_cmap`. This function documents the built-in colormaps, and will also return a list of all registered colormaps if called. You can set the colormap for an image, pcolor, scatter, etc, using a keyword argument:: imshow(X, cmap=cm.hot) or using the :func:`set_cmap` function:: imshow(X) pyplot.set_cmap('hot') pyplot.set_cmap('jet') In interactive mode, :func:`set_cmap` will update the colormap post-hoc, allowing you to see which one works best for your data. All built-in colormaps can be reversed by appending ``_r``: For instance, ``gray_r`` is the reverse of ``gray``. There are several common color schemes used in visualization: Sequential schemes for unipolar data that progresses from low to high Diverging schemes for bipolar data that emphasizes positive or negative deviations from a central value Cyclic schemes for plotting values that wrap around at the endpoints, such as phase angle, wind direction, or time of day Qualitative schemes for nominal data that has no inherent ordering, where color is used only to distinguish categories Matplotlib ships with 4 perceptually uniform color maps which are the recommended color maps for sequential data: ========= =================================================== Colormap Description ========= =================================================== inferno perceptually uniform shades of black-red-yellow magma perceptually uniform shades of black-red-white plasma perceptually uniform shades of blue-red-yellow viridis perceptually uniform shades of blue-green-yellow ========= =================================================== The following colormaps are based on the `ColorBrewer `_ color specifications and designs developed by Cynthia Brewer: ColorBrewer Diverging (luminance is highest at the midpoint, and decreases towards differently-colored endpoints): ======== =================================== Colormap Description ======== =================================== BrBG brown, white, blue-green PiYG pink, white, yellow-green PRGn purple, white, green PuOr orange, white, purple RdBu red, white, blue RdGy red, white, gray RdYlBu red, yellow, blue RdYlGn red, yellow, green Spectral red, orange, yellow, green, blue ======== =================================== ColorBrewer Sequential (luminance decreases monotonically): ======== ==================================== Colormap Description ======== ==================================== Blues white to dark blue BuGn white, light blue, dark green BuPu white, light blue, dark purple GnBu white, light green, dark blue Greens white to dark green Greys white to black (not linear) Oranges white, orange, dark brown OrRd white, orange, dark red PuBu white, light purple, dark blue PuBuGn white, light purple, dark green PuRd white, light purple, dark red Purples white to dark purple RdPu white, pink, dark purple Reds white to dark red YlGn light yellow, dark green YlGnBu light yellow, light green, dark blue YlOrBr light yellow, orange, dark brown YlOrRd light yellow, orange, dark red ======== ==================================== ColorBrewer Qualitative: (For plotting nominal data, :class:`ListedColormap` is used, not :class:`LinearSegmentedColormap`. Different sets of colors are recommended for different numbers of categories.) * Accent * Dark2 * Paired * Pastel1 * Pastel2 * Set1 * Set2 * Set3 A set of colormaps derived from those of the same name provided with Matlab are also included: ========= ======================================================= Colormap Description ========= ======================================================= autumn sequential linearly-increasing shades of red-orange-yellow bone sequential increasing black-white color map with a tinge of blue, to emulate X-ray film cool linearly-decreasing shades of cyan-magenta copper sequential increasing shades of black-copper flag repetitive red-white-blue-black pattern (not cyclic at endpoints) gray sequential linearly-increasing black-to-white grayscale hot sequential black-red-yellow-white, to emulate blackbody radiation from an object at increasing temperatures jet a spectral map with dark endpoints, blue-cyan-yellow-red; based on a fluid-jet simulation by NCSA [#]_ pink sequential increasing pastel black-pink-white, meant for sepia tone colorization of photographs prism repetitive red-yellow-green-blue-purple-...-green pattern (not cyclic at endpoints) spring linearly-increasing shades of magenta-yellow summer sequential linearly-increasing shades of green-yellow winter linearly-increasing shades of blue-green ========= ======================================================= A set of palettes from the `Yorick scientific visualisation package `_, an evolution of the GIST package, both by David H. Munro are included: ============ ======================================================= Colormap Description ============ ======================================================= gist_earth mapmaker's colors from dark blue deep ocean to green lowlands to brown highlands to white mountains gist_heat sequential increasing black-red-orange-white, to emulate blackbody radiation from an iron bar as it grows hotter gist_ncar pseudo-spectral black-blue-green-yellow-red-purple-white colormap from National Center for Atmospheric Research [#]_ gist_rainbow runs through the colors in spectral order from red to violet at full saturation (like *hsv* but not cyclic) gist_stern "Stern special" color table from Interactive Data Language software ============ ======================================================= A set of cyclic color maps: ================ ========================================================= Colormap Description ================ ========================================================= hsv red-yellow-green-cyan-blue-magenta-red, formed by changing the hue component in the HSV color space twilight perceptually uniform shades of white-blue-black-red-white twilight_shifted perceptually uniform shades of black-blue-white-red-black ================ ========================================================= Other miscellaneous schemes: ============= ======================================================= Colormap Description ============= ======================================================= afmhot sequential black-orange-yellow-white blackbody spectrum, commonly used in atomic force microscopy brg blue-red-green bwr diverging blue-white-red coolwarm diverging blue-gray-red, meant to avoid issues with 3D shading, color blindness, and ordering of colors [#]_ CMRmap "Default colormaps on color images often reproduce to confusing grayscale images. The proposed colormap maintains an aesthetically pleasing color image that automatically reproduces to a monotonic grayscale with discrete, quantifiable saturation levels." [#]_ cubehelix Unlike most other color schemes cubehelix was designed by D.A. Green to be monotonically increasing in terms of perceived brightness. Also, when printed on a black and white postscript printer, the scheme results in a greyscale with monotonically increasing brightness. This color scheme is named cubehelix because the r,g,b values produced can be visualised as a squashed helix around the diagonal in the r,g,b color cube. gnuplot gnuplot's traditional pm3d scheme (black-blue-red-yellow) gnuplot2 sequential color printable as gray (black-blue-violet-yellow-white) ocean green-blue-white rainbow spectral purple-blue-green-yellow-orange-red colormap with diverging luminance seismic diverging blue-white-red nipy_spectral black-purple-blue-green-yellow-red-white spectrum, originally from the Neuroimaging in Python project terrain mapmaker's colors, blue-green-yellow-brown-white, originally from IGOR Pro ============= ======================================================= The following colormaps are redundant and may be removed in future versions. It's recommended to use the names in the descriptions instead, which produce identical output: ========= ======================================================= Colormap Description ========= ======================================================= gist_gray identical to *gray* gist_yarg identical to *gray_r* binary identical to *gray_r* ========= ======================================================= .. rubric:: Footnotes .. [#] Rainbow colormaps, ``jet`` in particular, are considered a poor choice for scientific visualization by many researchers: `Rainbow Color Map (Still) Considered Harmful `_ .. [#] Resembles "BkBlAqGrYeOrReViWh200" from NCAR Command Language. See `Color Table Gallery `_ .. [#] See `Diverging Color Maps for Scientific Visualization `_ by Kenneth Moreland. .. [#] See `A Color Map for Effective Black-and-White Rendering of Color-Scale Images `_ by Carey Rappaport )rr"Zcmap_drJrJrJrKr sprc st}tjdtjd}g}tdtdx|D]x}t|j}d}|dk rv||}|dk rvt | d dd }d |}| ||gt t|t t|q0Wd d d }|d dd|gfd d|D|g}d|t_dS)z Generates the plotting docstring. These must be done after the entire module is imported, so it is called from the end of this module, which is generated by boilerplate.py. z(?:\s*).+?\.(?:\s+|$))flagsZFunctionZ DescriptionrNr  z`%s`=z {:{}} {:{}}cs g|]\}}d||qS)z {:{}} {:{}})rn)rr#summary)max_name max_summaryrJrKrsz1_setup_pyplot_info_docstrings..)rrecompileDOTALLrr(__doc__rer!Zcleandocrreplacerrrnjoinr) ZcommandsZfirst_sentenceZrowsr#docr.reZ separatorrrJ)r/r0rK_setup_pyplot_info_docstringss(   (r8cKsJ|dkrt}|dkrtd|dkr,t}tj|f||d|}|S)NzNo mappable was found to use for colorbar creation. First define a mappable such as an image (with imshow) or a contour set (with contourf).)caxr)rrrrbcolorbar)Zmappabler9rr|rrJrJrKr:)sr:cCs&t}|dkrtd|||dS)ap Set the color limits of the current image. To apply clim to all axes images do:: clim(0, 0.5) If either *vmin* or *vmax* is None, the image min/max respectively will be used for color scaling. If you want to set the clim of multiple images, use, for example:: for im in gca().get_images(): im.set_clim(0, 0.05) Nz1You must first define an image, e.g., with imshow)rrZset_clim)vminvmaximrJrJrKclim9sr>cCs4t|}td|jdt}|dk r0||dS)a; Set the default colormap. Applies to the current image if any. See help(colormaps) for more information. *cmap* must be a :class:`~matplotlib.colors.Colormap` instance, or the name of a registered colormap. See :func:`matplotlib.cm.register_cmap` and :func:`matplotlib.cm.get_cmap`. image)cmapN)r"r#rr#rset_cmap)r@r=rJrJrKrARs rAcCstj||S)N)rOr?imread)rrnrJrJrKrBfsrBcKstjj||f|S)N)rOr?imsave)rZarrrrJrJrKrCksrCcKsVt|}|dkrt}n"t|t|d}|ddddg}|j|f|}t||S)al Display an array as a matrix in a new figure window. The origin is set at the upper left hand corner and rows (first dimension of the array) are displayed horizontally. The aspect ratio of the figure window is that of the array, unless this would make an excessively short or narrow figure. Tick labels for the xaxis are placed on top. Parameters ---------- A : array-like(M, N) The matrix to be displayed. fignum : None or int or False If *None*, create a new figure window with automatic numbering. If a nonzero integer, draw into the figure with the given number (create it if it does not exist). If 0, use the current axes (or create one if it does not exist). .. note:: Because of how `.Axes.matshow` tries to set the figure aspect ratio to be the one of the array, strange things may happen if you reuse an existing figure. Returns ------- image : `~matplotlib.image.AxesImage` Other Parameters ---------------- **kwargs : `~matplotlib.axes.Axes.imshow` arguments r)rg333333?g ףp= ?g?)npZ asanyarrayrrrrmatshowsci)AZfignumrrrr=rJrJrKrEps' rEcOs:tr ttts tdtdd}|j||}|S)z Make a polar plot. call signature:: polar(theta, r, **kwargs) Multiple *theta*, *r* arguments are supported, with format strings, as in :func:`~matplotlib.pyplot.plot`. zMTrying to create polar plot on an axis that does not have a polar projection.T)polar)rbZget_axesrrrrrplot)r{rrrrJrJrKrHs     rHr#,c  s| r t} nt} t|dkr&td|dkr4t}ddlm} t(t d| t j ||||||dWdQRXfdd } | |d\}}g}t|dkr| ddd}| |dd }t||}||f| ||nt|}xtd|D]}|r:|dkr"| |dd|}}n| j |dd||d }n|dkrR| ddd}| ||\}}||| ||d }t||}|||f| |r|||r||q|d qW|s|||d kr| dS)a Plot the data in a file. *cols* is a sequence of column identifiers to plot. An identifier is either an int or a string. If it is an int, it indicates the column number. If it is a string, it indicates the column header. matplotlib will make column headers lower case, replace spaces with underscores, and remove all illegal characters; so ``'Adj Close*'`` will have name ``'adj_close'``. - If len(*cols*) == 1, only that column will be plotted on the *y* axis. - If len(*cols*) > 1, the first element will be an identifier for data for the *x* axis and the remaining elements will be the column indexes for multiple subplots if *subplots* is *True* (the default), or for lines in a single subplot if *subplots* is *False*. *plotfuncs*, if not *None*, is a dictionary mapping identifier to an :class:`~matplotlib.axes.Axes` plotting function as a string. Default is 'plot', other choices are 'semilogy', 'fill', 'bar', etc. You must use the same type of identifier in the *cols* vector as you use in the *plotfuncs* dictionary, e.g., integer column numbers in both or column names in both. If *subplots* is *False*, then including any function such as 'semilogy' that changes the axis scaling will set the scaling for all columns. *comments*, *skiprows*, *checkrows*, *delimiter*, and *names* are all passed on to :func:`matplotlib.mlab.csv2rec` to load the data into a record array. If *newfig* is *True*, the plot always will be made in a new figure; if *False*, it will be made in the current figure if one exists, else in a new figure. kwargs are passed on to plotting functions. Example usage:: # plot the 2nd and 4th column against the 1st in two subplots plotfile(fname, (0,1,3)) # plot using column names; specify an alternate plot type for volume plotfile(fname, ('date', 'volume', 'adj_close'), plotfuncs={'volume': 'semilogy'}) Note: plotfile is intended as a convenience for quickly plotting data from flat files; it is not intended as an alternative interface to general plotting with pyplot or matplotlib. r1z%must have at least one column of dataNr)MatplotlibDeprecationWarningignore)commentsskiprows checkrows delimiternamescsHt|tr||fSt|tr<jjt|}||fStddS)z.return the name and column data for identifierz&identifier must be a string or integerN)rrrZdtyperTrr)Z identifierr#)rrJrK getname_val s     zplotfile..getname_valrI)rrZdate)rrbrrdictmatplotlib.cbookrNrcatch_warnings simplefilterrZcsv2recrrgetattr set_ylabelrangerZ is_last_row set_xlabelrZ autofmt_xdate)rZcolsZ plotfuncsrPrQrRrSrTrZnewfigrrrNrVZxnamerZ ynamelistrfuncnamerNirZynamerrJ)rUrKplotfilesV7                   rbcsd}t|dfddS)zZAutogenerated wrappers will get their docstring from a base function with an addendum.rz cst|S)N)r copy_dedent)r)addendumbaserJrKr9 sz$_autogen_docstring..)r ZAppender)remsgrJ)rdrerK_autogen_docstring4 s rgZbackend_fallbackrg)datacKs$tj|f|dk rd|ini|S)Nrh)racorr)rrhrrJrJrKriR sricKs2tj|f|||||d|dk r(d|ini|S)N)FsFcwindowpad_tosidesrh)rangle_spectrum)rrjrkrlrmrnrhrrJrJrKroY srocOstj||f||S)N)rannotate)rZxyr{rrJrJrKrpc srpcKstj||||f|S)N)rarrow)rrZdxZdyrrJrJrKrqi srqbothcCstj|||dS)N)enableaxistight)r autoscale)rsrtrurJrJrKrvo srvcKstjf|||d|S)N)rxminxmax)raxhline)rrwrxrrJrJrKryu srycKstj||f||d|S)N)rwrx)raxhspan)yminymaxrwrxrrJrJrKrz{ srzcOstj||S)N)rrt)vrrJrJrKrt srtcKstjf|||d|S)N)rr{r|)raxvline)rr{r|rrJrJrKr~ sr~cKstj||f||d|S)N)r{r|)raxvspan)rwrxr{r|rrJrJrKr sr皙?center)alignrhcKs0tj||f|||d|dk r&d|ini|S)N)widthrrrh)rbar)rheightrrrrhrrJrJrKr srcOs"tj||dk rd|ini|S)Nrh)rbarbs)rhr{r|rJrJrKr sr)rcKstj||f|||d|S)N)rrr)rbarh)rrrrrrrJrJrKr srcCsXtj|f||||||||| | | | | ||||||||||||d|dk rPd|iniS)N)notchsymvertwhis positionswidths patch_artist bootstrap usermediansconf_intervalsmeanline showmeansshowcapsshowbox showfliersboxpropsr flierprops medianprops meanpropscapprops whiskerprops manage_xticks autorangezorderrh)rboxplot)rrrrrrrrrrrrrrrrrrrrrrrrrrrhrJrJrKr s   rcKs&tj||f|dk rd|ini|S)Nrh)r broken_barh)ZxrangesZyrangerhrrJrJrKr srcCs tS)N)rclarJrJrJrKr srcOstj|f||S)N)rclabel)ZCSr{rrJrJrKr srdefaultc  Ks<tj||f|||||||| | d | dk r2d| ini| S)N) NFFTrjrkdetrendrlnoverlaprmrn scale_by_freqrh)rcohere) rrrrjrkrrlrrmrnrrhrrJrJrKr s rcOs8tj||dk rd|ini|}|jdk r4t||S)Nrh)rcontour_ArF)rhr{r__retrJrJrKr s  rcOs8tj||dk rd|ini|}|jdk r4t||S)Nrh)rcontourfrrF)rhr{rrrJrJrKr s  rc  Ks>tj||f|||||||| | | d | dk r4d| ini| S)N) rrjrkrrlrrmrnr return_linerh)rcsd)rrrrjrkrrlrrmrnrrrhrrJrJrKr s rrcKsDtj||f|||||||| | | | | |d |dk r:d|ini|S)N) yerrxerrfmtecolor elinewidthcapsize barsabovelolimsuplimsxlolimsxuplims erroreverycapthickrh)rerrorbar)rrrrrrrrrrrrrrrrhrrJrJrKr s  r horizontalsolidc Ks4tj|f||||||d|dk r*d|ini|S)N) orientation lineoffsets linelengths linewidthsr linestylesrh)r eventplot) rrrrrrrrhrrJrJrKr s rcOs"tj||dk rd|ini|S)Nrh)rfill)rhr{rrJrJrKr" srcKs2tj||f||||d|dk r(d|ini|S)N)y2where interpolatesteprh)r fill_between)rZy1rrrrrhrrJrJrKr* srcKs2tj||f||||d|dk r(d|ini|S)N)x2rrrrh)r fill_betweenx)rZx1rrrrrhrrJrJrKr4 srmajorcKstjf|||d|S)N)bwhichrt)rgrid)rrrtrrJrJrKr> srZlinearZfacecKsVtj||f|||||||| | | | | ||||d|dk r@d|ini|}t||S)N)Cgridsizebinsxscaleyscaleextentr@normr;r<alphar edgecolorsreduce_C_functionmincnt marginalsrh)rhexbinrF)rrrrrrrrr@rr;r<rrrrrrrhrrrJrJrKrD s  rmidverticalcKsFtj|f||||||||| | | | | ||d|dk r皙?cCsTtj||||f|||||| | | | | |||d |dk r>d|ini}t|j|S)N) rrrr@r arrowsize arrowstyle minlength transformr start_points maxlengthintegration_directionrh)r streamplotrFr)rrur}rrrr@rr<r=r>r?rr@rArBrhrrJrJrKrCx s rCcKstjf|S)N)rtable)rrJrJrKrE srEcKstj|||f||d|S)N)fontdictwithdash)rr)rrrrFrGrrJrJrKr srcKstjfd|i|S)Nrt)r tick_params)rtrrJrJrKrH srH)rtr scilimits useOffset useLocale useMathTextcCstj||||||dS)N)rtrrIrJrKrL)rticklabel_format)rtrrIrJrKrLrJrJrKrM srMcOs$tj||}|jdk r t||S)N)r tricontourrrF)r{rrrJrJrKrN s rNcOs$tj||}|jdk r t||S)N)r tricontourfrrF)r{rrrJrJrKrO s rOcOstj||}t||S)N)r tripcolorrF)r{rrrJrJrKrP srPcOstj||S)N)rtriplot)r{rrJrJrKrQ srQ?c  Cs6tj|f||||||||d| dk r.d| iniS)N)rrrr showextrema showmedianspoints bw_methodrh)r violinplot) ZdatasetrrrrrSrTrUrVrhrJrJrKrW s   rWcKs2tj|||f|||d|dk r(d|ini|S)N)rrrrh)rvlines)rr{r|rrrrhrrJrJrKrX s rXcKs2tj||f||||d|dk r(d|ini|S)N)rr usevlinesmaxlagsrh)rxcorr)rrrrrYrZrhrrJrJrKr[ s r[cCs t|S)N)r_sci)r=rJrJrKrF srFcKstj|f|||d|S)N)rFrr )r set_title)rrFrr rrJrJrKtitle sr^cKstj|f||d|S)N)rFlabelpad)rr^)xlabelrFr_rrJrJrKr` sr`cKstj|f||d|S)N)rFr_)rr\)ylabelrFr_rrJrJrKra sracKstj|f|S)N)r set_xscale)valuerrJrJrKr srcKstj|f|S)N)r set_yscale)rcrrJrJrKr srcCs tddS)z Set the colormap to "autumn". This changes the default colormap as well as the colormap of the current image if there is one. See ``help(colormaps)`` for more information. autumnN)rArJrJrJrKre srecCs tddS)z Set the colormap to "bone". This changes the default colormap as well as the colormap of the current image if there is one. See ``help(colormaps)`` for more information. boneN)rArJrJrJrKrf srfcCs tddS)z Set the colormap to "cool". This changes the default colormap as well as the colormap of the current image if there is one. See ``help(colormaps)`` for more information. coolN)rArJrJrJrKrg& srgcCs tddS)z Set the colormap to "copper". This changes the default colormap as well as the colormap of the current image if there is one. See ``help(colormaps)`` for more information. copperN)rArJrJrJrKrh1 srhcCs tddS)z Set the colormap to "flag". This changes the default colormap as well as the colormap of the current image if there is one. See ``help(colormaps)`` for more information. flagN)rArJrJrJrKri< sricCs tddS)z Set the colormap to "gray". This changes the default colormap as well as the colormap of the current image if there is one. See ``help(colormaps)`` for more information. grayN)rArJrJrJrKrjG srjcCs tddS)z Set the colormap to "hot". This changes the default colormap as well as the colormap of the current image if there is one. See ``help(colormaps)`` for more information. hotN)rArJrJrJrKrkR srkcCs tddS)z Set the colormap to "hsv". This changes the default colormap as well as the colormap of the current image if there is one. See ``help(colormaps)`` for more information. hsvN)rArJrJrJrKrl] srlcCs tddS)z Set the colormap to "jet". This changes the default colormap as well as the colormap of the current image if there is one. See ``help(colormaps)`` for more information. jetN)rArJrJrJrKrmh srmcCs tddS)z Set the colormap to "pink". This changes the default colormap as well as the colormap of the current image if there is one. See ``help(colormaps)`` for more information. pinkN)rArJrJrJrKrns srncCs tddS)z Set the colormap to "prism". This changes the default colormap as well as the colormap of the current image if there is one. See ``help(colormaps)`` for more information. prismN)rArJrJrJrKro~ srocCs tddS)z Set the colormap to "spring". This changes the default colormap as well as the colormap of the current image if there is one. See ``help(colormaps)`` for more information. springN)rArJrJrJrKrp srpcCs tddS)z Set the colormap to "summer". This changes the default colormap as well as the colormap of the current image if there is one. See ``help(colormaps)`` for more information. summerN)rArJrJrJrKrq srqcCs tddS)z Set the colormap to "winter". This changes the default colormap as well as the colormap of the current image if there is one. See ``help(colormaps)`` for more information. winterN)rArJrJrJrKrr srrcCs tddS)z Set the colormap to "magma". This changes the default colormap as well as the colormap of the current image if there is one. See ``help(colormaps)`` for more information. magmaN)rArJrJrJrKrs srscCs tddS)z Set the colormap to "inferno". This changes the default colormap as well as the colormap of the current image if there is one. See ``help(colormaps)`` for more information. infernoN)rArJrJrJrKrt srtcCs tddS)z Set the colormap to "plasma". This changes the default colormap as well as the colormap of the current image if there is one. See ``help(colormaps)`` for more information. plasmaN)rArJrJrJrKru srucCs tddS)z Set the colormap to "viridis". This changes the default colormap as well as the colormap of the current image if there is one. See ``help(colormaps)`` for more information. viridisN)rArJrJrJrKrv srvcCs tddS)z Set the colormap to "nipy_spectral". This changes the default colormap as well as the colormap of the current image if there is one. See ``help(colormaps)`` for more information. nipy_spectralN)rArJrJrJrKrw srw)NNT)NN)r1rr)N)N)N)r1r1FFTNN)r1r1N)N)N)NNNNNN)N)r NNN)N)NN)NN)NNN)NN)N)N) rJNrKrrLrMNTT)NNNNN)TrrN)rrr1)rr1)rrr1)rr1)rN)rN)NNNNNNNNNNNNNNNNNNNNNNTFN) NNNNNNNNNN) NNrNNNFFFFFr1N)rr1r1NNr)rNFN)rNNF)Nrrr)NNNNFNrrrNFNNFN)rNFNNN)rrr)NNNNNNNNNNr1rNNN)rrN)NNNNNN)NNNNN)NNNNr Fr NNTNNr FF)rdNTF) NNNNNNNNNN) NNNNNNNNNNN)NNNNNNNNNNNNNNN)rNNr/r0) r1NNNNr1r:r;NNNrrr)NF)rr)NTrRFTFrN)rrr)NrN)NN)NN(r4rpr!ZloggingZnumbersrr1rUrrrrOZmatplotlib.colorbarZmatplotlib.imagerrrrrXrr r r r r Zmatplotlib.backend_basesrZmatplotlib.figurerrZmatplotlib.gridspecrrrrrrZmatplotlib.rcsetuprZ_interactive_bkZmatplotlib.artistrrrrrZmatplotlib.axesrrZmatplotlib.projectionsrrZmatplotlib.scaler r!r"Z matplotlib.cmr#r$ZnumpyrDZmatplotlib.colorsr%Zmatplotlib.linesr&Zmatplotlib.textr'r(Zmatplotlib.patchesr)r*r+r,Zmatplotlib.widgetsr-r.r/r0Ztickerr2r3r4r5r6r7r8r9r:r;r<r=r>r?r@rArBrCrDZmatplotlib.backendsrErFZ getLoggerrGrtrWr[r^r`rrQrcrcrlrxr}r~rrrrrcopyrrrrbrrrrrrrrrjrrrrrrrrrrZdedent_interpdrrrrrrrrrrrr rrrrrrrrrrr8r:Z colorbar_docr>rAr?rBrCrErHrbrgrW __getitem__ __setitem__rkrirorprqrvryrzrtr~rrrrrrrrrZ detrend_noneZwindow_hanningrrrrrrrrrrZmeanrrrrrrrrrrrrrr rrIrr r!r"r)r*r+r.r3r5r9rrCrErHrMrNrOrPrQrWrXr[r\rFr]r^r^r`r\rarbrrdrrerfrgrhrirjrkrlrmrnrorprqrrrsrtrurvrwrJrJrJrKs             T 5  <   <    * (o  } =      %% M MGFs.    4 x