U Ev f‡ëã@sXdZddlmZddlZddlZddlZddlZddlZddl Z ddl m Z m Z m Z mZddlmZddlmZddlmZddlmZddlmZddlmZddl m!Z"ddl#m$Z%ddl&m'Z(ddlm)Z)ddl*m+Z+m,Z,ddl&m-Z-ddl.m/Z/d d l0m1Z1d d l0m2Z2d d l0m3Z3e j4e  5d gdgdgdœ¡Gdd„de)ƒƒƒZ6ddd„Z7dS)aR axes3d.py, original mplot3d version by John Porter Created: 23 Sep 2005 Parts fixed by Reinier Heeres Minor additions by Ben Axelrod Significant updates and revisions by Ben Root Module containing Axes3D, an object which can plot 3D objects on a 2D matplotlib figure. é)Ú defaultdictN)Ú_apiÚcbookÚ _docstringÚ_preprocess_data)ÚAxes)Ú_axis_method_wrapperÚ_process_plot_format)ÚBbox)Ú Triangulationé)Úart3d)Úproj3d)Úaxis3dZxlim3dZylim3dZzlim3d©ÚxlimÚylimÚzlimcsxeZdZdZdZdZe ¡ej d<e   d¡Z e   d¡Z e   d¡Ze   d¡Ze   d¡Zddd d dd dd dd œ‡fdd„ Zdd„Zdd„Zdd„Zdd„Zdd„Zdd„ZeddƒZeddƒZe jdd d!d"ed#d$„ƒƒZe jdd%d!d"ed&d$„ƒƒZe jddd!d"ed'd$„ƒƒZe  d¡dd(d)„ƒZ dd*d+„Z!e  d¡dd,d-„ƒZ"dd.d/„Z#e  d¡dd0d1„ƒZ$d d2d3„Z%d ‡fd5d6„ Z&d7d8„Z'd9d:œd;d<„Z(d d=d>„Z)e*j+‡fd?d@„ƒZ,dAdB„Z-dCdD„Z.eddEƒZ/eddFƒZ0dGdH„Z1dddd dIœdJdK„Z2d dMdN„Z3d dOdP„Z4ddQdR„Z5dSdT„Z6e  7ddU¡ddddVœdWdX„ƒZ8e9jj:Z;e9jjdYdZ„Z?d[d\„Z@d]d^„ZAedd_ƒZBed d`ƒZCed%d`ƒZDedd`ƒZEeFdajGdbdcdgƒ\eC_eD_eE_edddƒZHeddeƒZIeddfƒZJeddgƒZKeddhƒZLeddidjdkidlZMeddmƒZNeNjr"eNjeO Pdn¡7_dodp„ZQddqdr„ZRddsdt„ZSdudv„ZTdwdx„ZUdd{d|„ZVd}d~„ZWdd€„ZXdd‚„ZYdƒd„„ZZ‡fd…d†„Z[d‡dˆ„Z\d‰dŠ„Z]d‹dŒ„Z^ddŽ„Z_dd„Z`d‘d’„Zad“d”„Zbd•d–„Zcd—d˜„Zdddšd›„Zedœd„ZfdždŸ„Zgdd d¡„Zhd¢d£„Zid¤d¥„Zjd¦d§„Zkdd¨d©„Zld‡fdªd«„ Zmd¬d­„Znedd®ƒZod¯d°„Zpdd±d²„Zqd‡fd³d´„ ZrerZsejrZtddµœ‡fd¶d·„ ZueuZvddddd¸œd¹dº„Zwd»d¼„Zxdddddd½œd¾d¿„ZyddÁd„ZzddÃdÄ„Z{ddÅdÆ„Z|ddÇdÈ„Z}e~ƒd4dÀdddÉœ‡fdÊdË„ ƒZeZ€e~ƒd4dÀdddÉœ‡fdÌdÍ„ ƒZdÎdÏ„Z‚e~ƒdddÐœ‡fdÑdÒ„ ƒZƒeƒZ„e~ƒdddÐœ‡fdÓdÔ„ ƒZ…d‡fdÕdÖ„ Z†e~d×dØdÙdÚdÛdÜdÝdÞdßg dàd‡fdâdã„ ƒZ‡e‡Zˆe~ƒd‡fdädå„ ƒZ‰e~ƒd dçd脃ZŠd!‡fdêdë„ Z‹e~ƒd9dìdíd4dîœdïdð„ƒZŒeŒZddd ddñœdòdó„ZŽe~dbdcddôdõdögdàd"dødù„ƒZd#d4dúœ‡fdûdü„ Ze~ƒdýdþdÿd dddœdd„ƒZ‘e‘Z’‡Z“S($ÚAxes3Da 3D Axes object. .. note:: As a user, you do not instantiate Axes directly, but use Axes creation methods instead; e.g. from `.pyplot` or `.Figure`: `~.pyplot.subplots`, `~.pyplot.subplot_mosaic` or `.Figure.add_axes`. Z3d©ÚxÚyÚzrz3.6z3.7NéiÄÿÿÿrÚperspT)ÚelevÚazimÚrollÚsharezÚ proj_typeÚ box_aspectÚcomputed_zorderÚ focal_lengthcsˆ|dkrddddg}||_||_||_| || ¡| |_t ¡|_t ¡|_t ¡|_ t ¡|_ |  |j|j|j¡||_ |dk rš|j d ||¡d|_|  dd¡r®tdƒ‚tƒj||f| žd |d œ| —Žtƒ ¡| ¡d|_d|_| ¡|jjj d |j¡|jjj d |j¡|jjj d |j¡|  ¡|j! "d¡|j# $¡ %ddg¡} | d| d\|_&|_'|j(dd… )d¡dS)a2 Parameters ---------- fig : Figure The parent figure. rect : tuple (left, bottom, width, height), default: None. The ``(left, bottom, width, height)`` axes position. elev : float, default: 30 The elevation angle in degrees rotates the camera above and below the x-y plane, with a positive angle corresponding to a location above the plane. azim : float, default: -60 The azimuthal angle in degrees rotates the camera about the z axis, with a positive angle corresponding to a right-handed rotation. In other words, a positive azimuth rotates the camera about the origin from its location along the +x axis towards the +y axis. roll : float, default: 0 The roll angle in degrees rotates the camera about the viewing axis. A positive angle spins the camera clockwise, causing the scene to rotate counter-clockwise. sharez : Axes3D, optional Other Axes to share z-limits with. proj_type : {'persp', 'ortho'} The projection type, default 'persp'. box_aspect : 3-tuple of floats, default: None Changes the physical dimensions of the Axes3D, such that the ratio of the axis lengths in display units is x:y:z. If None, defaults to 4:4:3 computed_zorder : bool, default: True If True, the draw order is computed based on the average position of the `.Artist`\s along the view direction. Set to False if you want to manually control the order in which Artists are drawn on top of each other using their *zorder* attribute. This can be used for fine-tuning if the automatic order does not produce the desired result. Note however, that a manual zorder will only be correct for a limited view angle. If the figure is rotated by the user, it will look wrong from certain angles. focal_length : float, default: None For a projection type of 'persp', the focal length of the virtual camera. Must be > 0. If None, defaults to 1. For a projection type of 'ortho', must be set to either None or infinity (numpy.inf). If None, defaults to infinity. The focal length can be computed from a desired Field Of View via the equation: focal_length = 1/tan(FOV/2) **kwargs Other optional keyword arguments: %(Axes3D:kwdoc)s Nçgð?rÚdatalimZauto_add_to_figureFzSauto_add_to_figure is no longer supported for Axes3D. Use fig.add_axes(ax) instead.T)Zframeonr Zmotion_notify_eventZbutton_press_eventZbutton_release_eventr©rr)r r r )*Ú initial_azimÚ initial_elevÚ initial_rollÚ set_proj_typer!r ÚunitÚ xy_viewLimÚ zz_viewLimÚ xy_dataLimÚ zz_dataLimÚ view_initÚ_sharezÚ _shared_axesÚjoinÚ _adjustableÚpopÚAttributeErrorÚsuperÚ__init__Ú set_axis_offÚ set_axis_onÚMÚ fmt_zdataÚ mouse_initÚfigureÚcanvasÚ callbacksZ_connect_picklableÚ_on_moveÚ _button_pressÚ_button_releaseÚ set_top_viewÚpatchZ set_linewidthZ transLimitsÚinvertedÚ transformÚ _pseudo_wÚ _pseudo_hZspinesZ set_visible)ÚselfZfigZrectrrrrrr r!r"ÚargsÚkwargsZ pseudo_bbox©Ú __class__©ú?/tmp/pip-unpacked-wheel-7vhvci0g/mpl_toolkits/mplot3d/axes3d.pyr7Bsj8       ÿÿÿÿÿ  ÿ ÿ ÿ zAxes3D.__init__cCsd|_d|_dS)NFT©Ú _axis3donÚstale©rIrNrNrOr8¶szAxes3D.set_axis_offcCsd|_d|_dS)NTrPrSrNrNrOr9ºszAxes3D.set_axis_oncCs |j |¡S)zs For artists in an Axes, if the zaxis has units support, convert *z* using zaxis unit type )ÚzaxisZ convert_units)rIrrNrNrOÚconvert_zunits¾szAxes3D.convert_zunitscCsNd|j}d|j}d|j}d|j}| |f|j_| |f|j_d|_dS)Ngffffffî?gÍÌÌÌÌÌì?T)Ú_distZviewLimÚ intervalxÚ intervalyrR)rIZxdwlZxdwZydwlZydwrNrNrOrCÅs    zAxes3D.set_top_viewcCs(t |¡|_t |¡|_t |¡|_dS)z5Init 3D axes; overrides creation of regular X/Y axes.N)rZXAxisÚxaxisZYAxisÚyaxisZZAxisrTrSrNrNrOÚ _init_axisÑs  zAxes3D._init_axiscCs|jS)z0Return the ``ZAxis`` (`~.axis3d.Axis`) instance.©rTrSrNrNrOÚ get_zaxis×szAxes3D.get_zaxisrTZ get_gridlinesZ get_ticklinesz3.1rYz3.8)Ú alternativeZremovalcCs|jS©N)rYrSrNrNrOÚßózAxes3D.rZcCs|jSr_)rZrSrNrNrOr`áracCs|jSr_r\rSrNrNrOr`ãracCs | |¡Sr_)Ú _unit_cube)rIÚvalsrNrNrOÚ unit_cubeåszAxes3D.unit_cubec Cs\|p | ¡\}}}}}}|||f|||f|||f|||f|||f|||f|||f|||fgSr_)Ú get_w_lims)rIrcÚminxÚmaxxÚminyÚmaxyÚminzÚmaxzrNrNrOrbésùzAxes3D._unit_cubecCs | ||¡Sr_©Ú _tunit_cube©rIrcr:rNrNrOÚ tunit_cubeôszAxes3D.tunit_cubecCs(|dkr|j}| |¡}t ||¡}|Sr_)r:rbrZ proj_points)rIrcr:ZxyzsZtcuberNrNrOrmøs   zAxes3D._tunit_cubecCs | ||¡Sr_)Ú _tunit_edgesrnrNrNrOÚ tunit_edgesÿszAxes3D.tunit_edgescCs¼| ||¡}|d|df|d|df|d|df|d|df|d|df|d|df|d|df|d|df|d|df|d|df|d|df|d|dfg }|S) Nrr éééééérl)rIrcr:ÚtcÚedgesrNrNrOrps ózAxes3D._tunit_edgesFcsptjd|dtƒjd|||d||_|dkrl| |¡}t |j  ¡|j   ¡|j   ¡g¡}tj |dd}|j dkrötj|dd}t|||j|ƒ} | |j} t|j|j|jfƒD]:\} } | |kr¸| || | | d || | | d ƒq¸nvt |j¡} ||| |<d dd h |¡}|rb| ¡}tj |j|¡}tj | |¡}| |||9<| | ¡d S) aq Set the aspect ratios. Parameters ---------- aspect : {'auto', 'equal', 'equalxy', 'equalxz', 'equalyz'} Possible values: ========= ================================================== value description ========= ================================================== 'auto' automatic; fill the position rectangle with data. 'equal' adapt all the axes to have equal aspect ratios. 'equalxy' adapt the x and y axes to have equal aspect ratios. 'equalxz' adapt the x and z axes to have equal aspect ratios. 'equalyz' adapt the y and z axes to have equal aspect ratios. ========= ================================================== adjustable : None or {'box', 'datalim'}, optional If not *None*, this defines which parameter will be adjusted to meet the required aspect. See `.set_adjustable` for further details. anchor : None or str or 2-tuple of float, optional If not *None*, this defines where the Axes will be drawn if there is extra space due to aspect constraints. The most common way to specify the anchor are abbreviations of cardinal directions: ===== ===================== value description ===== ===================== 'C' centered 'SW' lower left corner 'S' middle of bottom edge 'SE' lower right corner etc. ===== ===================== See `~.Axes.set_anchor` for further details. share : bool, default: False If ``True``, apply the settings to all shared Axes. See Also -------- mpl_toolkits.mplot3d.axes3d.Axes3D.set_box_aspect )ÚautoÚequalÚequalxyÚequalyzÚequalxz©Úaspectrz)r€Ú adjustableÚanchorÚshare©r{r|r~r}r ©Úaxisr$ç@rrrN)rÚ check_in_listr6Ú set_aspectÚ_aspectÚ_equal_aspect_axis_indicesÚnpÚarrayrYZget_view_intervalrZrTÚptpr3ÚmeanÚmaxÚ _box_aspectÚ enumerateÚ set_xlim3dÚ set_ylim3dÚ set_zlim3dÚ differencer4ÚlinalgÚnormÚset_box_aspect)rIr€rr‚rƒÚ ax_indicesZview_intervalsrŽrÚscaleZdeltasÚiZset_limr Zremaining_ax_indicesÚ remainingZold_diagZnew_diagrLrNrOr‰sH0ÿÿ   þ  þ.  zAxes3D.set_aspectcCsPg}|dkrdddg}n4|dkr*ddg}n"|dkr 0. rzArgument zoom = z must be > 0N)rtrtrs©Zdtype)rsrg‘Ir|Eý?T) Ú ValueErrorrŒÚasarrayÚfloatrZ check_shaper—r˜r‘rR)rIr€ržrNrNrOr™szAxes3D.set_box_aspectcCsp|dkr|jdd}| ¡j}tj ¡ |¡}|j|j}d}|  ¡}|  |||¡}|  |  |  ¡|¡d¡dS)NT)Úoriginalr Úactive)Ú get_positionZ get_figureZtransSubfigureÚ mtransformsr r*Z transformedÚheightÚwidthÚfrozenZshrunk_to_aspectZ _set_positionZanchoredZ get_anchor)rIÚpositionZtransZbbZ fig_aspectr ZpbZpb1rNrNrOÚ apply_aspect¥s   zAxes3D.apply_aspectc s:| ¡s dS| ¡|j |¡d|_| ¡}| |r@|||ƒnd¡| ¡|_dd„|j Dƒ}|j rÚt dd„|j   ¡Dƒƒd}|}}t|dd„dd D]:}t|tjƒr¼||_|d7}qœt|tjƒrœ||_|d7}qœn|D] }| ¡qÞ|jr*|j   ¡D]}| |¡qþ|j   ¡D]}| |¡qtƒ |¡dS) NFcss,|]$}t|tjtjfƒr| ¡r|VqdSr_)Ú isinstanceÚmcollÚ CollectionÚmpatchesÚPatchÚ get_visible)Ú.0ÚartistrNrNrOÚ ËsþzAxes3D.draw..css|]}| ¡VqdSr_)Z get_zorder)r²r†rNrNrOr´Òsÿr cSs| ¡Sr_)Údo_3d_projection)r³rNrNrOr`×razAxes3D.draw..T)ÚkeyÚreverse)r±Z_unstale_viewLimrDÚdrawÚ_frameonZget_axes_locatorr«Úget_projr:Ú _childrenr!rÚ _axis_mapÚvaluesÚsortedr¬r­r®Úzorderr¯r°rµrQZ draw_paner6) rIÚrendererÚlocatorZcollections_and_patchesZ zorder_offsetZcollection_zorderZ patch_zorderr³r†rLrNrOr¸¶sH  ÿÿÿþ       z Axes3D.drawcCsh| ¡}| ||j¡}|dd|ddk}|dd|ddk}|dd|ddk}|||fS)Nr rrrsr)rermr:)rIrcrxZxhighZyhighZzhighrNrNrOÚget_axis_positionîs zAxes3D.get_axis_positioncKsdS)zK Not implemented in `~mpl_toolkits.mplot3d.axes3d.Axes3D`. NrN)rIZxysrKrNrNrOÚupdate_datalimöszAxes3D.update_datalimZ_get_autoscale_onZ_set_autoscale_oncCs*|dkrtdƒ‚||_| d¡d|_dS)a Set padding of Z data limits prior to autoscaling. *m* times the data interval will be added to each end of that interval before it is used in autoscaling. If *m* is negative, this will clip the data range instead of expanding it. For example, if your data is in the range [0, 2], a margin of 0.1 will result in a range [-0.2, 2.2]; a margin of -0.1 will result in a range of [0.2, 1.8]. Parameters ---------- m : float greater than -0.5 gà¿z margin must be greater than -0.5rTN)r Ú_zmarginÚ_request_autoscale_viewrR)rIÚmrNrNrOÚ set_zmarginÿs  zAxes3D.set_zmargin)rrrÚtightcGs|r&|dk s|dk s|dk r&tdƒ‚nBt|ƒdkrD|d}}}n$t|ƒdkr\|\}}}n |rhtdƒ‚|dkrª|dkrª|dkrª|dk ršt d|›d ¡|j|j|jfS|dk r¼| |¡|dk rÎ| |¡|dk rà|  |¡|j ||dk |dk |dk d dS) zñ Set or retrieve autoscaling margins. See `.Axes.margins` for full documentation. Because this function applies to 3D Axes, it also takes a *z* argument, and returns ``(xmargin, ymargin, zmargin)``. NzECannot pass both positional and keyword arguments for x, y, and/or z.r rrszYMust pass a single positional argument for all margins, or one for each margin (x, y, z).Tzignoring tight=z in get mode)rÈÚscalexÚscaleyÚscalez) Ú TypeErrorÚlenrZ warn_externalÚ_xmarginÚ_ymarginrÄZ set_xmarginZ set_ymarginrÇÚautoscale_view)rIrrrrÈÚmarginsrNrNrOrÑs0       þzAxes3D.marginsÚbothcCs¼|dkrd}d}d}nl|dkr6| t|ƒ¡| ¡}nd}|dkrZ| t|ƒ¡| ¡}nd}|dkr~| t|ƒ¡| ¡}nd}|r”|jd|d|r¦|jd |d|r¸|jd |ddS) a Convenience method for simple axis view autoscaling. See `.Axes.autoscale` for full documentation. Because this function applies to 3D Axes, *axis* can also be set to 'z', and setting *axis* to 'both' autoscales all three axes. NT)rrÒF)rrÒ©rrÒr)rÈrr)Zset_autoscalex_onÚboolÚget_autoscalex_onZset_autoscaley_onÚget_autoscaley_onÚset_autoscalez_onÚget_autoscalez_onrÅ)rIÚenabler†rÈrÉrÊrËrNrNrOÚ autoscale9s,   zAxes3D.autoscalecCs€t |¡t |¡kr<|j t t |¡t |¡g¡| ¡n |j || ¡|j || ¡|dk rt|j || ¡|  ¡dSr_) rŒÚshaper-Zupdate_from_data_xyÚ column_stackÚravelZupdate_from_data_xZupdate_from_data_yr.rÐ)rIÚXÚYÚZÚhad_datarNrNrOÚauto_scale_xyz\sÿzAxes3D.auto_scale_xyzcCs|dkrL|j}|sZ|jD]0}t|tjƒr.d}qt|tjtjfƒrd}qZqnt |ƒ}|_|râ|  ¡râ|j d  ¡|j j\}}|j ¡} |  ||¡\}}|jdkrÂ|||j} || 8}|| 7}|sÖ|  ||¡\}}| ||¡|rr| ¡rr|j d  ¡|j j\} } |j ¡} |  | | ¡\} } |jdkrP| | |j} | | 8} | | 7} |sf|  | | ¡\} } | | | ¡|r| ¡r|j d  ¡|jj\}}|j ¡}| ||¡\}}|jdkrà|||j} || 8}|| 7}|sö| ||¡\}}| ||¡dS)zÔ Autoscale the view limits using the data limits. See `.Axes.autoscale_view` for full documentation. Because this function applies to 3D Axes, it also takes a *scalez* argument. NTFrrrr)Ú_tightr»r¬ÚmimageZ AxesImageÚmlinesÚLine2Dr¯r°rÔrÕr1Úcleanr-rWrYZget_major_locatorZ nonsingularrÎZ view_limitsZ set_xboundrÖrXrZrÏZ set_yboundrØr.rTrÄÚ set_zbound)rIrÈrÉrÊrËrãr³Úx0Úx1ZxlocatorÚdeltaÚy0Úy1ZylocatorÚz0Úz1ZzlocatorrNrNrOrÐjs\               zAxes3D.autoscale_viewcCs4| ¡\}}| ¡\}}| ¡\}}||||||fS)zGet 3D world limits.)Ú get_xlim3dÚ get_ylim3dÚ get_zlim3d)rIrfrgrhrirjrkrNrNrOre¨s   zAxes3D.get_w_limsÚemit)ÚzminÚzmaxcCsf|dkrt |¡r|\}}|dk r6|dk r2tdƒ‚|}|dk rR|dk rNtdƒ‚|}|jj||||dS)zW Set 3D z limits. See `.Axes.set_ylim` for full documentation Nz$Cannot pass both 'bottom' and 'zmin'z!Cannot pass both 'top' and 'zmax'©rórz)rŒÚiterablerÌrTZ_set_lim)rIÚbottomÚtoprórzrôrõrNrNrOÚset_zlim°szAxes3D.set_zlimcCs t|jjƒSr_)Útupler+rWrSrNrNrOÚget_xlimÈszAxes3D.get_xlimcCs t|jjƒSr_)rûr+rXrSrNrNrOÚget_ylimÌszAxes3D.get_ylimcCs t|jjƒS)zGet 3D z limits.)rûr,rWrSrNrNrOÚget_zlimÐszAxes3D.get_zlimZ get_scaleZ_set_axes_scalea“ Set the {}-axis scale. Parameters ---------- value : {{"linear"}} The axis scale type to apply. 3D axes currently only support linear scales; other scales yield nonsensical results. **kwargs Keyword arguments are nominally forwarded to the scale class, but none of them is applicable for linear scales. rrZ get_ticklocsZ set_ticksZget_majorticklabelsZget_minorticklabelsZget_ticklabelsZset_ticklabelszAxis.set_tickszAxes3D.set_zticks)Zdoc_subZ axis_datez» Notes ----- This function is merely provided for completeness, but 3D axes do not support dates for ticks, and so this may not work as expected. cOsdS)z:Currently not implemented for 3D axes, and returns *None*.NrN)rIrJrKrNrNrOÚclabelýsz Axes3D.clabelcCsld|_|dkr|j|_n||_|dkr0|j|_n||_|dkrH|j|_n||_tjt dddd|d|_ dS)aŠ Set the elevation and azimuth of the axes in degrees (not radians). This can be used to rotate the axes programmatically. To look normal to the primary planes, the following elevation and azimuth angles can be used. A roll angle of 0, 90, 180, or 270 deg will rotate these views while keeping the axes at right angles. ========== ==== ==== view plane elev azim ========== ==== ==== XY 90 -90 XZ 0 -90 YZ 0 0 -XY -90 90 -XZ 0 90 -YZ 0 180 ========== ==== ==== Parameters ---------- elev : float, default: None The elevation angle in degrees rotates the camera above the plane pierced by the vertical axis, with a positive angle corresponding to a location above that plane. For example, with the default vertical axis of 'z', the elevation defines the angle of the camera location above the x-y plane. If None, then the initial value as specified in the `Axes3D` constructor is used. azim : float, default: None The azimuthal angle in degrees rotates the camera about the vertical axis, with a positive angle corresponding to a right-handed rotation. For example, with the default vertical axis of 'z', a positive azimuth rotates the camera about the origin from its location along the +x axis towards the +y axis. If None, then the initial value as specified in the `Axes3D` constructor is used. roll : float, default: None The roll angle in degrees rotates the camera about the viewing axis. A positive angle spins the camera clockwise, causing the scene to rotate counter-clockwise. If None, then the initial value as specified in the `Axes3D` constructor is used. vertical_axis : {"z", "x", "y"}, default: "z" The axis to align vertically. *azim* rotates about this axis. é Nrr rrr)Ú vertical_axis) rVr'rr&rr(rrZ check_getitemÚdictÚ_vertical_axis)rIrrrrrNrNrOr/s1    ÿzAxes3D.view_initcCsvtjddg|d|dkrH|dkr(d}n|dkr@td|›dƒ‚||_n*|dtjfkrjtd|›d |›ƒ‚tj|_dS) aÐ Set the projection type. Parameters ---------- proj_type : {'persp', 'ortho'} The projection type. focal_length : float, default: None For a projection type of 'persp', the focal length of the virtual camera. Must be > 0. If None, defaults to 1. The focal length can be computed from a desired Field Of View via the equation: focal_length = 1/tan(FOV/2) rZortho)rNr rzfocal_length = z must be greater than 0z must be None for proj_type = )rrˆr Ú _focal_lengthrŒÚinf)rIrr"rNrNrOr)GszAxes3D.set_proj_typecCst ||jd¡S)z1Roll arrays to match the different vertical axis.rr)rŒrr)rIZarrrNrNrOÚ_roll_to_verticalcszAxes3D._roll_to_verticalcCst| |j¡}tj| ¡| ¡| ¡žd|iŽ}d|}t t   |j ¡¡}t t   |j ¡¡}t  |¡t  |¡}t  |¡t |¡}t |¡}| |||g¡} ||j| } || } | |_| tj | ¡|_| | ¡\} } }| |_| |_||_|jtjkrt | | || ¡}t |j |j¡}n:||j| |j}t | | ||¡}t |j |j|j¡}t ||¡}t ||¡}|S)z?Create the projection matrix from the current viewing position.Z pb_aspectçà?)rr‘rZworld_transformationrðrñròrŒÚdeg2radr Ú _norm_anglerrÚcosÚsinrVZ_eyer—r˜Z_vvecÚ_calc_view_axesÚ_view_uÚ_view_vÚ_view_wrrZ_view_transformation_uvwZortho_transformationZpersp_transformationÚdot)rIr ZworldMÚRÚelev_radZazim_radÚp0Úp1Úp2ÚpsÚeyeÚvvecÚuÚvÚwZviewMZprojMZ eye_focalZM0r:rNrNrOrºgsF ýü  þ  zAxes3D.get_projrrrscCs:d|_t |¡ ¡|_t |¡ ¡|_t |¡ ¡|_dS)aÕ Set the mouse buttons for 3D rotation and zooming. Parameters ---------- rotate_btn : int or list of int, default: 1 The mouse button or buttons to use for 3D rotation of the axes. pan_btn : int or list of int, default: 2 The mouse button or buttons to use to pan the 3D axes. zoom_btn : int or list of int, default: 3 The mouse button or buttons to use to zoom the 3D axes. N)Úbutton_pressedrŒÚ atleast_1dÚtolistÚ _rotate_btnÚ_pan_btnÚ _zoom_btn)rIÚ rotate_btnÚpan_btnÚzoom_btnrNrNrOr<ªs zAxes3D.mouse_initcCs|jgggddS)ztƒ ¡|jtjkr$tjd|_nd|_| tjd¡dS)Nz axes.zmarginr#z axes3d.grid) r6ÚclearrrŒrÚmplÚrcParamsrÄÚgridrSrLrNrOr-âs   z Axes3D.clearcCsT|j|krP|j|_|j|j|_|_t|jj dƒ}|rP|  ¡dkrP|jj j   ¡dS©NÚtoolbar) ÚinaxesÚbuttonrÚxdataÚydataÚ_sxÚ_syÚgetattrr=r>Z _nav_stackr2Ú push_current©rIÚeventr2rNrNrOrAës  zAxes3D._button_presscCs6d|_t|jjdƒ}|r2| ¡dkr2|jjj ¡dSr1)rr9r=r>Úget_navigate_moder2r:r;rNrNrOrBószAxes3D._button_releasecCs"| ¡| ¡| ¡|j|j|jfSr_)rürýrþrrrrSrNrNrOÚ _get_viewûs ÿzAxes3D._get_viewcCs6|\}}}}}}|j|||d||_||_||_dS)Nr)Úsetrrr)rIÚviewrrrrrrrNrNrOÚ _set_views zAxes3D._set_viewc CsBz | |¡WSttfk r<|j ¡j}||ƒ}|YSXdS)zº Return *z* string formatted. This function will use the :attr:`fmt_zdata` attribute if it is callable, else will fall back on the zaxis major formatter N)r;r5rÌrTZget_major_formatterZformat_data_short)rIrÚfuncÚvalrNrNrOÚ format_zdatas   zAxes3D.format_zdatacs"|jdkrdS|j|jkrdt |j¡}t |j¡}t |j¡}d|d›d|d›d|d›d dd ¡St |  ¡‡‡fd d „d \}}|\}} } |\} } } t   |ˆ| ˆ¡}t   | ˆ| ˆ¡}||}||| ||| }t  ˆˆ||j¡\}}}| |¡}| |¡}| |¡}d |||fS)zå Given the 2D view coordinates attempt to guess a 3D coordinate. Looks for the nearest edge to the point and then assumes that the point is at the same z location as the nearest point on the edge. NÚz elevation=z.0fu °, azimuth=u °, roll=õ°ú-u−cs*t ˆˆf|ddd…|ddd…¡S)Nrrrr )rZ_line2d_seg_dist)Zedge©ÚxdÚydrNrOr`+sÿz%Axes3D.format_coord..)r¶zx=%s, y=%s, z=%s)r:rrr r rrrÚreplaceÚminrprŒÚhypotrZ inv_transformZ format_xdataZ format_ydatarD)rIrIrJZ norm_elevZ norm_azimZ norm_rollrrrérìrîrêrírïZd0Úd1ÚdtrrrÚxsÚysÚzsrNrHrOÚ format_coords0     ý ÿ      zAxes3D.format_coordcCs°|js dS| ¡dk rdS|jdkr(dS|j|j}}|dksH|j|krLdS||j||j}}|j}|j }|j|j kr|dkr|dkrdSt   |j ¡}|| dt  |¡||dt  |¡} || dt  |¡||dt  |¡} |j| |_|j| |_d|_nz|j|jkrj|j |j|jg¡\} } | | | d¡| dd|j|j¡| ¡n(|j|jkr’|||} | | | | ¡|||_|_|jj ¡dS)z¡ Mouse moving. By default, button-1 rotates, button-2 pans, and button-3 zooms; these buttons can be modified via `mouse_init`. Nré´Trr) rr=r:r5r6r3r7r8rGrHrrŒrrr r rrrRr Z transDatarFÚ start_panÚdrag_panrrÚend_panr!Ú_scale_axis_limitsr=r>Z draw_idle)rIr<rrÚdxÚdyrÚhrZdelevZdazimZpxÚpyr›rNrNrOr@=s>   ..    zAxes3D._on_movecCsP|j}|j ||f|j|jfg¡\\}}\}} |||_|_| |||¡|||| } } d} |dkrnd} n |dkrzd} | dkrŽ| dkrŽdSt  |j |j |j g¡} | |j |j} | jt  | | | g¡}| ¡\}}}}}}|||d}|||d}|||d}| ||||¡| ||||¡| ||||¡dS)Nrrrr rr)Z _pan_startZ trans_inverserFrrr7r8rUrŒrr rrr‘rVÚTrer“r”r•)rIr4r¶rrÚpr5r6Z xdata_startZ ydata_startZduZdvZdwrZduvw_projectedrfrgrhrirjrkrYrZÚdzrNrNrOrVzs0ÿzAxes3D.drag_panc Cs~t t |j¡¡}t t |j¡¡}d| |j¡}t d¡}t |ƒtj dkrTdnd||j <t   ||||¡\}}}|||fS)zÍ Get the unit vectors for the viewing axes in data coordinates. `u` is towards the right of the screen `v` is towards the top of the screen `w` is out of the screen rrsrréÿÿÿÿr )rŒrr r rrrr‘ÚzerosÚabsÚpirrZ _view_axes) rIrrZroll_radrÚVrrrrNrNrOr žs  zAxes3D._calc_view_axesÚincCs–|\}}}} |dkr.|jjd}|jjd} n |dkrN|jjd}|jjd}t t||gƒ|jjd|jjd¡\}}t t|| gƒ|jjd|jjd¡\}} ||d} || d} |jjd|jjdd} |jjd|jjdd} | | | d¡| dd| | ¡| ¡t ||ƒ}t || ƒ}||jjd|jjd}||jjd|jjd}t||ƒ}|dkr„d|}|  |||¡dS)zÏ Zoom in or out of the bounding box. Will center the view in the center of the bounding box, and zoom by the ratio of the size of the bounding box to the size of the Axes3D. rr rrrrNÚout) ÚbboxrLrrŒZclipr¾rUrVrWrbÚ_zoom_data_limits)rIrgÚ directionÚmodeZtwinxZtwinyZstart_xZstart_yZstop_xZstop_yZ zoom_center_xZ zoom_center_yZ ax_center_xZ ax_center_yrYrZÚscale_uÚscale_vr›rNrNrOÚ_set_view_from_bbox´s<     ÿ ÿ      zAxes3D._set_view_from_bboxc Cs²t |||g¡}t ||¡s”t |j|j|jg¡}|t d¡}tjj|j |dd}|j dkr”|  |j ¡}t  t  ||d¡¡}|||||<| |d|d|d¡dS)aò Zoom in or out of a 3D plot. Will scale the data limits by the scale factors. These will be transformed to the x, y, z data axes based on the current view angles. A scale factor > 1 zooms out and a scale factor < 1 zooms in. For an axes that has had its aspect ratio set to 'equal', 'equalxy', 'equalyz', or 'equalxz', the relevant axes are constrained to zoom equally. Parameters ---------- scale_u : float Scale factor for the u view axis (view screen horizontal). scale_v : float Scale factor for the v view axis (view screen vertical). scale_w : float Scale factor for the w view axis (view screen depth). rsr r…r„rrrN)rŒrZallcloser rrrr—r˜r]rŠr‹ZargminrbrX) rIrkrlZscale_wr›rÚSZax_idxsZ min_ax_idxsrNrNrOrhäs   zAxes3D._zoom_data_limitscCs´| ¡\}}}}}} ||d} ||d} | |d} |||} |||}| ||}| | | d| | d¡| | |d| |d¡| | |d| |d¡dS)a´ Keeping the center of the x, y, and z data axes fixed, scale their limits by scale factors. A scale factor > 1 zooms out and a scale factor < 1 zooms in. Parameters ---------- scale_x : float Scale factor for the x data axis. scale_y : float Scale factor for the y data axis. scale_z : float Scale factor for the z data axis. rrN)rer“r”r•)rIZscale_xZscale_yZscale_zrfrgrhrirjrkZcxÚcyÚczrYrZr_rNrNrOrX s      zAxes3D._scale_axis_limitscKs"|dk r||j_|jj||f|ŽS)zI Set zlabel. See doc for `.set_ylabel` for description. N)rTÚlabelpadZset_label_text)rIZzlabelÚfontdictrqrKrNrNrOÚ set_zlabel)szAxes3D.set_zlabelcCs|j ¡}| ¡S)z. Get the z-label text string. )rTZ get_labelZget_text)rIÚlabelrNrNrOÚ get_zlabel1s zAxes3D.get_zlabelcCs|jS)z)Get whether the 3D axes panels are drawn.)r¹rSrNrNrOÚ get_frame_on:szAxes3D.get_frame_oncCst|ƒ|_d|_dS)zs Set whether the 3D axes panels are drawn. Parameters ---------- b : bool TN)rÔr¹rR)rIÚbrNrNrOÚ set_frame_on>s zAxes3D.set_frame_oncKst|ƒr d}||_d|_dS)zÞ Set / unset 3D grid. .. note:: Currently, this function does not behave the same as `.axes.Axes.grid`, but it is intended to eventually support that behavior. TN)rÍZ _draw_gridrR)rIZvisiblerKrNrNrOr0Is z Axes3D.gridc s€tjddddg|d|dkr.tƒj|f|Ž|dkr|t|ƒ}| dd ¡| d d ¡| d d ¡| d d ¡|jjf|Žd S) ax Convenience method for changing the appearance of ticks and tick labels. See `.Axes.tick_params` for full documentation. Because this function applies to 3D Axes, *axis* can also be set to 'z', and setting *axis* to 'both' autoscales all three axes. Also, because of how Axes3D objects are drawn very differently from regular 2D axes, some of these settings may have ambiguous meaning. For simplicity, the 'z' axis will accept settings as if it was like the 'y' axis. .. note:: Axes3D currently ignores some of these settings. rrrrÒr…)rrrÒrÓrùNrøZlabeltopZ labelbottom)rrˆr6Ú tick_paramsrr4rTZset_tick_params)rIr†rKZzkwrLrNrOryYs    zAxes3D.tick_paramscCs | ¡\}}|j||dddS)z$ Invert the z-axis. N©rz)rþrú©rIrørùrNrNrOÚ invert_zaxisws zAxes3D.invert_zaxisZ get_invertedcCs(| ¡\}}||kr||fS||fSdS)zP Return the lower and upper z-axis bounds, in increasing order. N)rþr{rNrNrOÚ get_zbound€s zAxes3D.get_zboundcCsd|dkrt |¡r|\}}| ¡\}}|dkr2|}|dkr>|}|jt||ft| ¡ƒddddS)zä Set the lower and upper numerical bounds of the z-axis. This method will honor axes inversion regardless of parameter order. It will not change the autoscaling setting (`.get_autoscalez_on()`). N)r·rz)rŒr÷r}rúr¾rÔÚzaxis_inverted)rIÚlowerÚupperZ old_lowerZ old_upperrNrNrOrèŠs   ÿþzAxes3D.set_zboundc s&tƒj|||f|Ž}t |||¡|S)zÌ Add text to the plot. Keyword arguments will be passed on to `.Axes.text`, except for the *zdir* keyword, which sets the direction to be used as the z direction. )r6Útextr Z text_2d_to_3d)rIrrrÚsÚzdirrKrrLrNrOržsz Axes3D.text©rƒc s¬| ¡}|r4t|dtƒs4|^}}d|kr@tdƒ‚n | dd¡}t |t |¡¡}tƒj ||f|ž|Ž}|D]} t j | ||dqlt   ||||¡\}}}|  ||||¡|S)a Plot 2D or 3D data. Parameters ---------- xs : 1D array-like x coordinates of vertices. ys : 1D array-like y coordinates of vertices. zs : float or 1D array-like z coordinates of vertices; either one for all points or one for each point. zdir : {'x', 'y', 'z'}, default: 'z' When plotting 2D data, the direction to use as z. **kwargs Other arguments are forwarded to `matplotlib.axes.Axes.plot`. rrRz+plot() for multiple values for argument 'z'©rRrƒ)Úhas_datar¬ÚstrrÌr4rŒÚ broadcast_torÛr6Úplotr Ú line_2d_to_3dÚ juggle_axesrâ) rIrPrQrƒrJrKrárRÚlinesÚlinerLrNrOr‰­s  z Axes3D.plot)r˜ÚvminÚvmaxÚ lightsourcec! s| ¡} |jdkrtdƒ‚t |¡}t |||¡\}}}|j\} } d|kpPd|k} d|kp`d|k} | rr| rrtdƒ‚| dd¡‰| dd¡‰| dd ¡}| dd ¡}t j d r²| }n| }|rìt t t  | |¡d ƒƒ‰t t t  | |¡d ƒƒ‰| d d ¡}|d kr2| dd ¡}|d kr"|j ¡}t t |¡¡}| dd ¡}| d|d k¡}|d kr`tdƒ‚g}| d ˆdkr¸| d ˆdkr¸|d kr¸tj‡‡fdd„|||fDƒdd}nÜttd| d ˆƒƒ| d g}ttd| d ˆƒƒ| d g}g}t|d d…|d d …ƒD]€\‰‰t|d d…|d d …ƒD]Z\‰‰‡‡‡‡fdd„|||fDƒ}t |¡j}| |¡|d k r4| |ˆˆ¡q4qt|tjƒr²t |¡ ¡r g}g}t ||¡D]F\}}t |¡t |¡jd d}t |ƒrÆ| |¡| |¡qÆ|}|d k r |}|d k rHt!j"|f||||dœ|—Ž}n¢|rÐt!j"|f|Ž}t|tjƒr||dj#dd} nt dd„|Dƒ¡} | $| ¡|d k s®|d k rº| %||¡|d k rê| &|¡nt!j"|f|||dœ|—Ž}| '|¡| (|||| ¡|S)a7 Create a surface plot. By default, it will be colored in shades of a solid color, but it also supports colormapping by supplying the *cmap* argument. .. note:: The *rcount* and *ccount* kwargs, which both default to 50, determine the maximum number of samples used in each direction. If the input data is larger, it will be downsampled (by slicing) to these numbers of points. .. note:: To maximize rendering speed consider setting *rstride* and *cstride* to divisors of the number of rows minus 1 and columns minus 1 respectively. For example, given 51 rows rstride can be any of the divisors of 50. Similarly, a setting of *rstride* and *cstride* equal to 1 (or *rcount* and *ccount* equal the number of rows and columns) can use the optimized path. Parameters ---------- X, Y, Z : 2D arrays Data values. rcount, ccount : int Maximum number of samples used in each direction. If the input data is larger, it will be downsampled (by slicing) to these numbers of points. Defaults to 50. rstride, cstride : int Downsampling stride in each direction. These arguments are mutually exclusive with *rcount* and *ccount*. If only one of *rstride* or *cstride* is set, the other defaults to 10. 'classic' mode uses a default of ``rstride = cstride = 10`` instead of the new default of ``rcount = ccount = 50``. color : color-like Color of the surface patches. cmap : Colormap Colormap of the surface patches. facecolors : array-like of colors. Colors of each individual patch. norm : Normalize Normalization for the colormap. vmin, vmax : float Bounds for the normalization. shade : bool, default: True Whether to shade the facecolors. Shading is always disabled when *cmap* is specified. lightsource : `~matplotlib.colors.LightSource` The lightsource to use when *shade* is True. **kwargs Other keyword arguments are forwarded to `.Poly3DCollection`. rrú!Argument Z must be 2-dimensional.ÚrstrideÚcstrideÚrcountÚccountú.Cannot specify both stride and count argumentsré2ú_internal.classic_moder Ú facecolorsNÚcolorÚcmapÚshadezshade cannot be None.rcsg|]}t |ˆˆ¡‘qSrN)rZ_array_patch_perimeters©r²Úa)r“r’rNrOÚ Rsÿz'Axes3D.plot_surface..r`r…c s.g|]&}t |ˆˆd…ˆˆd…f¡‘qS)r )rZ_array_perimeterr)ÚcsÚcs_nextÚrsÚrs_nextrNrOrŸ]sÿ)Ú edgecolorsr™rœr).rrcSs g|]}|dd…df ¡‘qS©Nrr)r)r²rrNrNrOrŸ‹s)r™rœr))r†Úndimr rZ_to_unmasked_float_arrayrŒÚbroadcast_arraysrÛr4r.r/ÚintrÚceilÚ _get_linesÚget_next_colorrÚmcolorsÚto_rgbaÚgetÚstackÚlistÚrangeÚzipr]Úappendr¬ZndarrayÚisnanÚanyÚ itertoolsÚ zip_longestrÍr ÚPoly3DCollectionrÚ set_arrayÚset_climÚset_normÚadd_collectionrâ)!rIrÞrßràr˜rŽrrrKráÚrowsÚcolsÚ has_strideÚ has_countr”r•Zcompute_stridesZfcolorsršr›rœZcolsetÚpolysZrow_indsZcol_indsrZ new_polysZ new_colsetr^ÚcolZnew_polyÚpolycÚavg_zrN)r r¡r“r¢r£r’rOÚ plot_surfaceØsÄF               ÿþ ÿý""ý       ÿþþ    ÿþþ zAxes3D.plot_surfacec sÔ| ¡}ˆjdkrtdƒ‚t ˆˆˆ¡\‰‰‰ˆj\}}d|kpFd|k}d|kpVd|k} |rh| rhtdƒ‚| dd¡} | dd¡} | dd ¡} | dd ¡} tjd rè| ræ| rÂt t t  || ¡dƒƒnd } | rât t t  || ¡dƒƒnd } nJ|s2| r t t t  || ¡dƒƒnd } | r.t t t  || ¡dƒƒnd } t  ˆ¡t  ˆ¡t  ˆ¡‰‰‰| r–t td || ƒƒ}|d krš|d |dkrš||dg7}ng}| rÜt td || ƒƒ}|d krà|d |dkrà||dg7}ng}| d krü| d krütd ƒ‚ˆjd krg}g}‡fdd„|Dƒ}‡fdd„|Dƒ}‡fdd„|Dƒ}‡fdd„|Dƒ}‡fdd„|Dƒ}‡fdd„|Dƒ}dd„t|||ƒDƒdd„t|||ƒDƒ}tj|f|Ž}| |¡| ˆˆˆ|¡|S)a¿ Plot a 3D wireframe. .. note:: The *rcount* and *ccount* kwargs, which both default to 50, determine the maximum number of samples used in each direction. If the input data is larger, it will be downsampled (by slicing) to these numbers of points. Parameters ---------- X, Y, Z : 2D arrays Data values. rcount, ccount : int Maximum number of samples used in each direction. If the input data is larger, it will be downsampled (by slicing) to these numbers of points. Setting a count to zero causes the data to be not sampled in the corresponding direction, producing a 3D line plot rather than a wireframe plot. Defaults to 50. rstride, cstride : int Downsampling stride in each direction. These arguments are mutually exclusive with *rcount* and *ccount*. If only one of *rstride* or *cstride* is set, the other defaults to 1. Setting a stride to zero causes the data to be not sampled in the corresponding direction, producing a 3D line plot rather than a wireframe plot. 'classic' mode uses a default of ``rstride = cstride = 1`` instead of the new default of ``rcount = ccount = 50``. **kwargs Other keyword arguments are forwarded to `.Line3DCollection`. rrr‘r’r“r”r•r–r r—r˜rr`z*Either rstride or cstride must be non zerocsg|] }ˆ|‘qSrNrN©r²rœ)rÞrNrOrŸþsz)Axes3D.plot_wireframe..csg|] }ˆ|‘qSrNrNrÆ)rßrNrOrŸÿscsg|] }ˆ|‘qSrNrNrÆ)ràrNrOrŸscsg|] }ˆ|‘qSrNrNrÆ)ÚtXrNrOrŸscsg|] }ˆ|‘qSrNrNrÆ)ÚtYrNrOrŸscsg|] }ˆ|‘qSrNrNrÆ)ÚtZrNrOrŸscSs"g|]\}}}tt|||ƒƒ‘qSrN©r°r²©r²ÚxlÚylÚzlrNrNrOrŸsÿcSs"g|]\}}}tt|||ƒƒ‘qSrNrÊrËrNrNrOrŸsÿ)r†r¦r rŒr§rÛr4r.r/r¨rr©Z transposer°r±Úsizer²r ÚLine3DCollectionr¼râ)rIrÞrßràrKrár½r¾r¿rÀr’r“r”r•ZriiZciiZxlinesZylinesZzlinesZtxlinesZtylinesZtzlinesrŒÚlinecrN)rÞrßràrÇrÈrÉrOÚplot_wireframe›sj&        """"  ÿ ÿþ zAxes3D.plot_wireframe)ršr˜rŽrrcOs|| ¡}|dkr|j ¡}t t |¡¡}| dd¡} | d| dk¡} t j ||Ž\} }}z| d¡} Wnt k r‚|^} }YnXt  | ¡} |   ¡} | j| }| j| }| | }tj|||fdd}| r.r)râ)rIrÞrßràrƒrÙráZlimitsrNrùrOÚ_auto_scale_contourf%s  ÿzAxes3D._auto_scale_contourfrêcs\| ¡}t ||||¡\} } } tƒj| | | f|ž|Ž} | | ||¡} | ||||| |¡| S)a« Create a 3D filled contour plot. Parameters ---------- X, Y, Z : array-like Input data. See `.Axes.contourf` for supported data shapes. zdir : {'x', 'y', 'z'}, default: 'z' The direction to use. offset : float, optional If specified, plot a projection of the contour lines at this position in a plane normal to *zdir*. data : indexable object, optional DATA_PARAMETER_PLACEHOLDER *args, **kwargs Other arguments are forwarded to `matplotlib.axes.Axes.contourf`. Returns ------- matplotlib.contour.QuadContourSet )r†r rñr6Úcontourfrërú)rIrÞrßràrƒrèrJrKrárórôrõràrÙrLrNrOrû/s zAxes3D.contourfcs¦| ¡}tj||Ž\}}}|j}|j}d|kr:| d¡} n|^} }t ||| |¡\} } } t| | |j|j ƒ}t ƒj || f|ž|Ž} |  | ||¡}|  ||| |||¡| S)aW Create a 3D filled contour plot. .. note:: This method currently produces incorrect output due to a longstanding bug in 3D PolyCollection rendering. Parameters ---------- X, Y, Z : array-like Input data. See `.Axes.tricontourf` for supported data shapes. zdir : {'x', 'y', 'z'}, default: 'z' The direction to use. offset : float, optional If specified, plot a projection of the contour lines at this position in a plane normal to zdir. data : indexable object, optional DATA_PARAMETER_PLACEHOLDER *args, **kwargs Other arguments are forwarded to `matplotlib.axes.Axes.tricontourf`. Returns ------- matplotlib.tri._tricontour.TriContourSet rà)r†r rÓrrr4r rñrÖrör6Ú tricontourfrërú)rIrƒrèrJrKrárÕrÞrßràrórôrõràrÙrLrNrOrüQs ÿ  zAxes3D.tricontourfcsªt |¡}|jrt |¡nd}t|ƒtjkrHtj|||d|  |¡nRt|ƒtj krrtj |||d|  |¡n(t|ƒtj krštj |||d|  |¡tƒ |¡}|S)a Add a 3D collection object to the plot. 2D collection types are converted to a 3D version by modifying the object and adding z coordinate information. Supported are: - PolyCollection - LineCollection - PatchCollection rr…)rŒrrÏrLÚtyper­ZPolyCollectionr rðrÝZLineCollectionræZPatchCollectionÚpatch_collection_2d_to_3dr6r¼)rIrÂrRrƒZzvalsZzsortvalZ collectionrLrNrOrÞ‚s ÿ    zAxes3D.add_collection3drPrQrRr‚r¤ÚcÚ facecolorr™rš)Z replace_nameséc sÌ| ¡} |} tjdd„|||fDƒŽ\}}}tj |¡}t |||||¡\}}}}}t | |¡rh| ¡}t ƒj ||f|ž||dœ| —Ž} t j | |||d|j dkr¸|jdkr¸| d¡| |||| ¡| S)a Create a scatter plot. Parameters ---------- xs, ys : array-like The data positions. zs : float or array-like, default: 0 The z-positions. Either an array of the same length as *xs* and *ys* or a single value to place all points in the same plane. zdir : {'x', 'y', 'z', '-x', '-y', '-z'}, default: 'z' The axis direction for the *zs*. This is useful when plotting 2D data on a 3D Axes. The data must be passed as *xs*, *ys*. Setting *zdir* to 'y' then plots the data to the x-z-plane. See also :doc:`/gallery/mplot3d/2dcollections3d`. s : float or array-like, default: 20 The marker size in points**2. Either an array of the same length as *xs* and *ys* or a single value to make all markers the same size. c : color, sequence, or sequence of colors, optional The marker color. Possible values: - A single color format string. - A sequence of colors of length n. - A sequence of n numbers to be mapped to colors using *cmap* and *norm*. - A 2D array in which the rows are RGB or RGBA. For more details see the *c* argument of `~.axes.Axes.scatter`. depthshade : bool, default: True Whether to shade the scatter markers to give the appearance of depth. Each call to ``scatter()`` will perform its depthshading independently. data : indexable object, optional DATA_PARAMETER_PLACEHOLDER **kwargs All other keyword arguments are passed on to `~.axes.Axes.scatter`. Returns ------- paths : `~matplotlib.collections.PathCollection` cSs"g|]}t tj |tj¡¡‘qSrN)rŒrÝÚmaÚfilledÚnan)r²ÚtrNrNrOrŸÙsz"Axes3D.scatter..)r‚rÿ)rRrƒÚ depthshadeçš™™™™™©?r)r†rŒr§rrÝrZdelete_masked_pointsZmay_share_memoryÚcopyr6Úscatterr rþrÄrÏrÇrâ) rIrPrQrRrƒr‚rÿrrJrKráZzs_origÚpatchesrLrNrOr £s"2ÿ     ÿ zAxes3D.scattercsä| ¡}tƒj||f|ž|Ž}t |t|ƒ¡}g} g} t||ƒD]T\} } t | ¡} | |   ¡7} | | gt| ƒ7} t  | | |¡d|kr@|   |d¡q@t| ƒdkr°t| Ž\}}n gg}}t  ||| |¡\}}} |  ||| |¡|S)a Add 2D bar(s). Parameters ---------- left : 1D array-like The x coordinates of the left sides of the bars. height : 1D array-like The height of the bars. zs : float or 1D array-like Z coordinate of bars; if a single value is specified, it will be used for all bars. zdir : {'x', 'y', 'z'}, default: 'z' When plotting 2D data, the direction to use as z ('x', 'y' or 'z'). data : indexable object, optional DATA_PARAMETER_PLACEHOLDER **kwargs Other keyword arguments are forwarded to `matplotlib.axes.Axes.bar`. Returns ------- mpl_toolkits.mplot3d.art3d.Patch3DCollection Úalphar)r†r6ÚbarrŒrˆrÍr²r Z_get_patch_vertsrZpatch_2d_to_3dZ set_alphar‹râ)rIÚleftr§rRrƒrJrKrár r×Zverts_zsr^rÚvsrPrQrLrNrOr ïs$    z Axes3D.barÚaveragec Osæ| ¡} t t |¡|||||¡\}}}}}}t |¡}t ||¡}t |¡}t ||¡}t |¡}t ||¡}t ddddddg¡}t |j|j¡}d||fd||fd ||ffD]J\}}}|d tj tj f}|d tj tj f}|||d |f|d |f<qº|  d |jd d …¡}g}|d kr8|j   ¡g}t t |¡ƒ}t|ƒt|ƒkrv|D]}| |gd ¡q\n&|}t|ƒt|ƒkrœ|d t|ƒ9}tj|f| ž||| | dœ| —Ž}| |¡| ||f||f||f| ¡|S)aŠ Generate a 3D barplot. This method creates three-dimensional barplot where the width, depth, height, and color of the bars can all be uniquely set. Parameters ---------- x, y, z : array-like The coordinates of the anchor point of the bars. dx, dy, dz : float or array-like The width, depth, and height of the bars, respectively. color : sequence of colors, optional The color of the bars can be specified globally or individually. This parameter can be: - A single color, to color all bars the same color. - An array of colors of length N bars, to color each bar independently. - An array of colors of length 6, to color the faces of the bars similarly. - An array of colors of length 6 * N bars, to color each face independently. When coloring the faces of the boxes specifically, this is the order of the coloring: 1. -Z (bottom of box) 2. +Z (top of box) 3. -Y 4. +Y 5. -X 6. +X zsort : str, optional The z-axis sorting scheme passed onto `~.art3d.Poly3DCollection` shade : bool, default: True When true, this shades the dark sides of the bars (relative to the plot's source of light). lightsource : `~matplotlib.colors.LightSource` The lightsource to use when *shade* is True. data : indexable object, optional DATA_PARAMETER_PLACEHOLDER **kwargs Any additional keyword arguments are passed onto `~.art3d.Poly3DCollection`. Returns ------- collection : `~.art3d.Poly3DCollection` A collection of three-dimensional polygons representing the bars. )©rrr©rr r©r r r©r rr)©rrr ©r rr ©r r r ©rr r )rrrr)rrrr)rrrr)rrrrrr rr.)r`Nrv)Úzsortr™rœr)r†rŒr§rrLrrÚemptyrÛZnewaxisÚreshapeÚ_get_patches_for_fillr«r°r¬Z to_rgba_arrayrÍÚextendr r¸r¼râ)rIrrrrYrZr_ršrrœrrJrKrárfrgrhrirjrkZcuboidrÁrœr^Zdpr™rÿrÂrNrNrOÚbar3d& sh>ÿ   Û.&  ûüû z Axes3D.bar3dÚcenterc s<tƒj|f||dœ|—Ž}|j ¡\}}|j d|¡|S)N)rrÚlocgq= ×£pí?)r6Ú set_titleÚtitler¥Zset_y)rIrtrrrrKÚretrrrLrNrOr Ç szAxes3D.set_titleg333333Ó?Útail)ÚlengthÚarrow_length_ratioÚpivotÚ normalizec sNddd„} | ¡} ||||||g}dd„|Dƒ}tj||žŽ}|dd…}|dd…}|r|t tj|¡‰‡fdd„|Dƒ}nd d„|Dƒ}td d „|Dƒƒr¸tjgf| Ž}|  |¡|Stj d |gt d }||}t j dddg| d| dkrø||8}n| dkr||d8}t |dd…¡}t |dd…¡ t ¡}tjj|dd}|dk‰|ˆ}| rv|ˆ|ˆ d¡}n|ˆ}t|ƒdkrú|tj ||¡ dd¡}| |ƒ}|dd…dd…ftj ||¡}| t|ƒddf¡}| dd¡}||•}ng}tj|f| Ž}|  |¡| |dd…df|dd…df|dd…df| ¡|S)aF Plot a 3D field of arrows. The arguments can be array-like or scalars, so long as they can be broadcast together. The arguments can also be masked arrays. If an element in any of argument is masked, then that corresponding quiver element will not be plotted. Parameters ---------- X, Y, Z : array-like The x, y and z coordinates of the arrow locations (default is tail of arrow; see *pivot* kwarg). U, V, W : array-like The x, y and z components of the arrow vectors. length : float, default: 1 The length of each quiver. arrow_length_ratio : float, default: 0.3 The ratio of the arrow head with respect to the quiver. pivot : {'tail', 'middle', 'tip'}, default: 'tail' The part of the arrow that is at the grid point; the arrow rotates about this point, hence the name *pivot*. normalize : bool, default: False Whether all arrows are normalized to have the same length, or keep the lengths defined by *u*, *v*, and *w*. data : indexable object, optional DATA_PARAMETER_PLACEHOLDER **kwargs Any additional keyword arguments are delegated to :class:`.Line3DCollection` éc Ssl|dd…df}|dd…df}tjj|dd…dd…fdd}tj|||dkt |¡d}tj| ||dkt |¡d}t |¡}t |¡}t  |¡} t  ||dd|||d||| g||d|||dd|| | g| | || t  ||¡gg¡} |   ¡} | ddddgddddgfd9<t  d| |¡} t  d| |¡} tj| | gdd}|S)Nrr rrr…)Úwhererfr`zij...,...j->...i)rŒr—r˜ÚdivideZ zeros_likeZ ones_likeÚmathÚradiansr r rZ full_likerZeinsumr¯)ÚUVWZanglerrr˜Zx_pZy_pÚrarÿr‚ZRposZRnegZ Rpos_vecsZ Rneg_vecsÚ head_dirsrNrNrOÚ calc_arrowsù s(    (*þÿ$z"Axes3D.quiver..calc_arrowscSs g|]}t|tjjƒr|j‘qSrN)r¬rŒrZ MaskedArrayrö©r²ÚkrNrNrOrŸ sÿz!Axes3D.quiver..Nrvcs g|]}tjj|ˆd ¡‘qS)©rö)rŒrrÚ compressedr1r3rNrOrŸ" sÿcSsg|]}t |¡‘qSrN)rŒrÝr1rNrNrOrŸ% scss|]}t|ƒdkVqdS)rN)rÍ)r²rrNrNrOr´' sz Axes3D.quiver..r#rŸr#ZmiddleZtip)r&rrrsr r…r)r`r r`)r()r†rŒr§Ú functoolsÚreduceÚ logical_orrµr rÐr¼rr¢rrˆrÜÚastyper—r˜rrÍÚmultiplyÚouterZswapaxesrâ)rIrÞrßràÚUrdÚWr$r%r&r'rKr0ráZ input_argsÚmasksZbcastrÑZshaft_dtZarrow_dtZXYZr-r˜Zshaftsr/ZheadsrŒrNr3rOÚquiverÎ sX+    ÿ    "   4z Axes3D.quiver©r™r¤rœrc1 s¢t|ƒdkrdd„}ndd„}|||Ž\}‰}ˆjdkr@tdƒ‚tjˆjtjd} t| dƒ‰|dkrxt ˆ¡\} } } n‡fd d „|Dƒ\} } } ‡fd d „} |dkr®|j   ¡}| |d ƒ}| |dƒ}|  | | | ¡tjdddgdddgdddgdddggtjd}t t ƒ}dd„}|dƒD]Œ}|j | ¡\}}}t |¡}t |¡}t |¡}| |j¡}|ddd…}|D]4}|D]&}| ||dg¡}t|ƒ}ˆ|r°|| ||¡t|dd…|dd…ƒD]ˆ\}}| |||g¡} | |||g¡}!t| ƒ}"t|!ƒ}#ˆ|"r*ˆ|#s*||" |!|¡n&ˆ|"sʈ|#rÊ||# |!|¡qÊ| |||dg¡}$| |||g¡}%t|$ƒ}&ˆ|&rv||& |%|¡qvqlqi}'| ¡D]è\}(})|dkrÌ|)}*ng}*|)D]†}+|+dd…df|+dd…df|+dd…dff},t |+j¡}-| |,|-dd…df<| |,|-dd…df<| |,|-dd…df<|* |-¡qÔ||(}.||(}/tj|*f|.|/||dœ|—Ž}0| |0¡|0|'|(<q´|'S)a‰ ax.voxels([x, y, z,] /, filled, facecolors=None, edgecolors=None, **kwargs) Plot a set of filled voxels All voxels are plotted as 1x1x1 cubes on the axis, with ``filled[0, 0, 0]`` placed with its lower corner at the origin. Occluded faces are not plotted. Parameters ---------- filled : 3D np.array of bool A 3D array of values, with truthy values indicating which voxels to fill x, y, z : 3D np.array, optional The coordinates of the corners of the voxels. This should broadcast to a shape one larger in every dimension than the shape of *filled*. These can be used to plot non-cubic voxels. If not specified, defaults to increasing integers along each axis, like those returned by :func:`~numpy.indices`. As indicated by the ``/`` in the function signature, these arguments can only be passed positionally. facecolors, edgecolors : array-like, optional The color to draw the faces and edges of the voxels. Can only be passed as keyword arguments. These parameters can be: - A single color value, to color all voxels the same color. This can be either a string, or a 1D rgb/rgba array - ``None``, the default, to use a single color for the faces, and the style default for the edges. - A 3D `~numpy.ndarray` of color names, with each item the color for the corresponding voxel. The size must match the voxels. - A 4D `~numpy.ndarray` of rgb/rgba data, with the components along the last axis. shade : bool, default: True Whether to shade the facecolors. lightsource : `~matplotlib.colors.LightSource` The lightsource to use when *shade* is True. **kwargs Additional keyword arguments to pass onto `~mpl_toolkits.mplot3d.art3d.Poly3DCollection`. Returns ------- faces : dict A dictionary indexed by coordinate, where ``faces[i, j, k]`` is a `.Poly3DCollection` of the faces drawn for the voxel ``filled[i, j, k]``. If no faces were drawn for a given voxel, either because it was not asked to be drawn, or it is fully occluded, then ``(i, j, k) not in faces``. Examples -------- .. plot:: gallery/mplot3d/voxels.py .. plot:: gallery/mplot3d/voxels_rgb.py .. plot:: gallery/mplot3d/voxels_torus.py .. plot:: gallery/mplot3d/voxels_numpy_logo.py rsc[s|||f||fSr_rN)Z _Axes3D__xZ _Axes3D__yZ _Axes3D__zrrKrNrNrOÚvoxels¦ szAxes3D.voxels..voxelsc[s d||fSr_rN)rrKrNrNrOr@© sz%Argument filled must be 3-dimensionalrŸr Nc3s|]}t |ˆ¡VqdSr_)rŒrˆ)r²rÿ)Ú coord_shaperNrOr´¸ sz Axes3D.voxels..cspt |¡dkr&t |ˆjt |¡¡St |¡dkr^t |¡dd…ˆjkrZtd |¡ƒ‚|Std |¡ƒ‚dS)N)rr )rsrtrsz8When multidimensional, {} must match the shape of filledzInvalid {} argument)rŒr¦rˆrÛr Úformat)ršÚname)rrNrOÚ_broadcast_color_argº sÿÿz+Axes3D.voxels.._broadcast_color_argr™r¤rcss8tj|tjd}t|ƒD]}|Vtj|ddd}qdS)z%Generate cyclic permutation matrices.rŸr rr…N)rŒrÚintpr±r)ÚnÚmatrœrNrNrOÚpermutation_matricesÝ s z+Axes3D.voxels..permutation_matricesr`rrr?)rÍr¦r rŒrrÛrErûÚindicesrr«rârr°r]rÚaranger³r²Úitemsrr r¸rÞ)1rIr™r¤rœrrJrKr@ZxyzrÏrrrrDZsquareZ voxel_facesrHZpermuteZpcZqcÚrcZpindsZqindsZrindsZsquare_rot_posZsquare_rot_negr^ÚqrZi0Zr1Zr2rrrãräÚpkZpk2ZikZpolygonsÚcoordZ faces_indsZfacesZ face_indsÚindÚfacerZ edgecolorZpolyrN)rArrOr@] s¢G        üû        "  . ÿþþ  z Axes3D.voxelsÚxerrÚyerrÚzerrrEcC s"| ¡}t |tj¡}dd„| ¡Dƒ}| dd¡|jd|fd|fd|fg|dd t  |¡rd|n|g}t  |¡rx|n|g}t  |¡rŒ|n|g}t |ƒt |ƒkr²t |ƒks¼nt d ƒ‚|  || ¡‰|  d d ¡}d |d <|jj|dkrò||fn|||f|dd\\}}tj||d|r2| |dd¡n| |dd¡| ¡dkr^| |¡nd }|  d¡d|kr~d|d<| d kr|d} dD]}|  |d ¡q”|d| i–}| rÄ| |d<nd|krÚ|d|d<dD]}||krÞ||||<qÞ|ddi–}| d krtjd} | dkr2d| |d<| d k rD| |d<| |d<d d!„}‡fd"d#„‰ggg}}}g} d$d$d%d&œ}!dd'dd(œ‰| dtjd)¡d}"|"|jjd*9}"|j ¡ d+|"|"fg¡}"t tj|"dd,¡}"tj|dddd-tj  !| "¡¡}#W5QRXt #|#|"dddg¡d'}"|"d.9}"||"d'd/œ–}$|$  dd ¡t$dddg|||g|||g|||g|||gƒD]\}%}&‰‰‰t %|%¡‰ˆ|%}'ˆd kr¤qtt  ˆ¡s¾ˆgt |&ƒ‰t &ˆ¡‰t 'ˆt |&ƒ¡ (t)¡‰t 'ˆt |&ƒ¡ (t)¡‰‡‡‡‡‡fd0d1„t*|||gƒDƒ}(|(\\})}*\}+},\}-}.ˆˆB}/|/ +¡rÐ| dkrÐ||)|+|-g|/ˆ@ƒ}0||*|,|.g|/ˆ@ƒ}1tj,|0d|!|'d2œ|—Ž}2tj,|1d|!|'d2œ|—Ž}3| |2¡| |3¡| -|2¡| -|3¡ˆ +¡r ||*|,|.gˆˆ@ƒ\}4}5}6|j.|4|5|6fˆž|$Žˆ +¡rF||)|+|-gˆˆ@ƒ\}7}8}9|j.|7|8|9fˆ ž|$Žtj/t 0|(¡j1f|Ž}:| 2|:¡| -|:¡|  -|(¡qtt 0| ¡} ‡fd3d4„};|;| dƒ\}<}=|;| dƒ\}>}?|;| dƒ\}@}A| 3|<|=f|>|?f|@|Af|¡t4j5|t6|ƒt6|ƒf|d k pþ|d k |d k |d5}B|j7 -|B¡|||fS)6a¼ Plot lines and/or markers with errorbars around them. *x*/*y*/*z* define the data locations, and *xerr*/*yerr*/*zerr* define the errorbar sizes. By default, this draws the data markers/lines as well the errorbars. Use fmt='none' to draw errorbars only. Parameters ---------- x, y, z : float or array-like The data positions. xerr, yerr, zerr : float or array-like, shape (N,) or (2, N), optional The errorbar sizes: - scalar: Symmetric +/- values for all data points. - shape(N,): Symmetric +/-values for each data point. - shape(2, N): Separate - and + values for each bar. First row contains the lower errors, the second row contains the upper errors. - *None*: No errorbar. Note that all error arrays should have *positive* values. fmt : str, default: '' The format for the data points / data lines. See `.plot` for details. Use 'none' (case-insensitive) to plot errorbars without any data markers. ecolor : color, default: None The color of the errorbar lines. If None, use the color of the line connecting the markers. elinewidth : float, default: None The linewidth of the errorbar lines. If None, the linewidth of the current style is used. capsize : float, default: :rc:`errorbar.capsize` The length of the error bar caps in points. capthick : float, default: None An alias to the keyword argument *markeredgewidth* (a.k.a. *mew*). This setting is a more sensible name for the property that controls the thickness of the error bar cap in points. For backwards compatibility, if *mew* or *markeredgewidth* are given, then they will over-ride *capthick*. This may change in future releases. barsabove : bool, default: False If True, will plot the errorbars above the plot symbols. Default is below. xlolims, ylolims, zlolims : bool, default: False These arguments can be used to indicate that a value gives only lower limits. In that case a caret symbol is used to indicate this. *lims*-arguments may be scalars, or array-likes of the same length as the errors. To use limits with inverted axes, `~.Axes.set_xlim` or `~.Axes.set_ylim` must be called before `errorbar`. Note the tricky parameter names: setting e.g. *ylolims* to True means that the y-value is a *lower* limit of the True value, so, only an *upward*-pointing arrow will be drawn! xuplims, yuplims, zuplims : bool, default: False Same as above, but for controlling the upper limits. errorevery : int or (int, int), default: 1 draws error bars on a subset of the data. *errorevery* =N draws error bars on the points (x[::N], y[::N], z[::N]). *errorevery* =(start, N) draws error bars on the points (x[start::N], y[start::N], z[start::N]). e.g. errorevery=(6, 3) adds error bars to the data at (x[6], x[9], x[12], x[15], ...). Used to avoid overlapping error bars when two series share x-axis values. Returns ------- errlines : list List of `~mpl_toolkits.mplot3d.art3d.Line3DCollection` instances each containing an errorbar line. caplines : list List of `~mpl_toolkits.mplot3d.art3d.Line3D` instances each containing a capline object. limmarks : list List of `~mpl_toolkits.mplot3d.art3d.Line3D` instances each containing a marker with an upper or lower limit. Other Parameters ---------------- data : indexable object, optional DATA_PARAMETER_PLACEHOLDER **kwargs All other keyword arguments for styling errorbar lines are passed `~mpl_toolkits.mplot3d.art3d.Line3DCollection`. Examples -------- .. plot:: gallery/mplot3d/errorbar3d.py cSsi|]\}}|dk r||“qSr_rN)r²r2rrNrNrOÚ sz#Axes3D.errorbar..r¿rrrrrF)Úconvertz)'x', 'y', and 'z' must have the same sizertNÚ _nolegend_rET)Z return_kwargs)rRgš™™™™™¹?ÚnoneršZC0) ÚmarkerÚ markersizeZmarkerfacecolorÚmarkeredgewidthZmarkeredgecolorZ markeveryÚ linestyleZ fillstyleZ drawstyleZ dash_capstyleZdash_joinstyleZsolid_capstyleZsolid_joinstyleZ linewidth)rFr r¿Z rasterizedr\ÚNonezerrorbar.capsizerr‡rZr[cs‡fdd„|DƒS)Ncsg|]}t |ˆ¡•‘qSrN)r¶Úcompress)r²rr3rNrOrŸí sz8Axes3D.errorbar.._apply_mask..rN)ZarraysrörNr3rOÚ _apply_maskê sz$Axes3D.errorbar.._apply_maskcsZt|jƒdkr|\}}n ||}}t |ˆB|||¡}t |ˆB|||¡}||fSr¥)rÍrÛrŒr))ÚerrÚdataZlomaskZhimaskZlow_errZhigh_errZlowsZhighs)Ú everymaskrNrOÚ _extract_errsï s   z&Axes3D.errorbar.._extract_errsú|Ú_)rr rrr rzlines.markersizeéHr%r…)rrrgV&,t=Ûý?)r$r%cs&g|]\}}ˆˆˆ||ˆˆƒ‘qSrNrN)r²rœrO)rcÚ dir_vectorr`ÚlolimsÚuplimsrNrOrŸ8 sÿz#Axes3D.errorbar..)ZlsrYc sLt |dd…ˆ|dd…dd…f¡t |dd…ˆ|dd…dd…f¡fSr_rø)Zerr_arrZ coord_label)Úi_xyzrNrOÚ_digout_minmax_ s$$ÿz'Axes3D.errorbar.._digout_minmax)Zhas_xerrZhas_yerrrt)8r†rZnormalize_kwargsrårærKÚ setdefaultZ_process_unit_inforŒr÷rÍr Z_errorevery_to_maskr4rªZ _plot_argsr rŠZ set_zorderrZadd_liner.r/r®r=ZdpiZ transAxesrErFrrîZ _setattr_cmr—Úinvrºrr²Zget_dir_vectorrrˆr8rÔr’rµZLine3Dr³r>rÐrr]r¼râÚ mcontainerZErrorbarContainerrûZ containers)CrIrrrrTrSrRÚfmtZ barsaboveZ erroreveryZecolorZ elinewidthZcapsizeZcapthickZxlolimsZxuplimsZylolimsZyuplimsZzlolimsZzuplimsrKrártZ data_lineZ base_styler¶Zeb_lines_styleZ eb_cap_styler_ZerrlinesZcaplinesZlimmarksZ coorderrsZ capmarkerZ quiversizeZinvMZeb_quiver_stylerƒraZi_zdirZcoorderrrÌÚxhrÍZyhrÎÚzhZnolimsZ lo_caps_xyzZ hi_caps_xyzZcap_loZcap_hiZxh0Zyh0Zzh0Zxl0Zyl0Zzl0ZerrlinerkrfrgrhrirjrkZerrorbar_containerrN)rcrgr`rbrjrhrirOÚerrorbar. sk ÿ"  ÿ                   ÿÿ ÿÿ þ     þ ÿþÿþ      ÿ    ü zAxes3D.errorbar)Úfor_layout_onlyc s^tƒj||||d}|g}|jrR|j ¡D]&}| ¡r*t ||¡}|r*| |¡q*t j   |¡S)N)Úcall_axes_locatorÚbbox_extra_artistsrs) r6Ú get_tightbboxrQr¼r½r±ÚmartistZ_get_tightbbox_for_layout_onlyr³r¦r Úunion) rIrÀrtrursr"Úbatchr†Zaxis_bbrLrNrOrvr s ýÿ zAxes3D.get_tightbboxzC0-ZC0ozC3-)ÚlinefmtÚ markerfmtÚbasefmtrørtÚ orientationcsÄddlm} | ¡} tjdddg| dt |¡t |¡f} t |¡t |¡f} t |¡t |¡f}| dkrœ|| }}||}}‡fdd„t|||ƒDƒ}nf| dkrÔ|| }}||}}‡fd d„t|||ƒDƒ}n.|| }}|| }}‡fd d„t|||ƒDƒ}t |ƒ\}}}|d kr$t j d }|j |||ˆ| d d\}t j|||d d}| |¡|j ||||d d\}| |||f|d}| |¡t  ||ˆˆg| ¡\}}}| || •|| •||•| ¡|S)aö Create a 3D stem plot. A stem plot draws lines perpendicular to a baseline, and places markers at the heads. By default, the baseline is defined by *x* and *y*, and stems are drawn vertically from *bottom* to *z*. Parameters ---------- x, y, z : array-like The positions of the heads of the stems. The stems are drawn along the *orientation*-direction from the baseline at *bottom* (in the *orientation*-coordinate) to the heads. By default, the *x* and *y* positions are used for the baseline and *z* for the head position, but this can be changed by *orientation*. linefmt : str, default: 'C0-' A string defining the properties of the vertical lines. Usually, this will be a color or a color and a linestyle: ========= ============= Character Line Style ========= ============= ``'-'`` solid line ``'--'`` dashed line ``'-.'`` dash-dot line ``':'`` dotted line ========= ============= Note: While it is technically possible to specify valid formats other than color or color and linestyle (e.g. 'rx' or '-.'), this is beyond the intention of the method and will most likely not result in a reasonable plot. markerfmt : str, default: 'C0o' A string defining the properties of the markers at the stem heads. basefmt : str, default: 'C3-' A format string defining the properties of the baseline. bottom : float, default: 0 The position of the baseline, in *orientation*-coordinates. label : str, default: None The label to use for the stems in legends. orientation : {'x', 'y', 'z'}, default: 'z' The direction along which stems are drawn. data : indexable object, optional DATA_PARAMETER_PLACEHOLDER Returns ------- `.StemContainer` The container may be treated like a tuple (*markerline*, *stemlines*, *baseline*) Examples -------- .. plot:: gallery/mplot3d/stem3d_demo.py r)Ú StemContainerrrr)r}cs&g|]\}}}ˆ||f|||fg‘qSrNrN©r²ZthisxZthisyZthisz©rørNrOrŸÓ sÿzAxes3D.stem..cs&g|]\}}}|ˆ|f|||fg‘qSrNrNrr€rNrOrŸØ sÿcs&g|]\}}}||ˆf|||fg‘qSrNrNrr€rNrOrŸÝ sÿNzlines.linestylerW)rRrƒrt)Z linestylesÚcolorsrt)rt)Úmatplotlib.containerr~r†rrˆrŒrLrr²r r.r/r‰r rÐr¼Z add_containerr‹râ)rIrrrrzr{r|rørtr}r~rárrrZbasexZbasexlimZbaseyZbaseylimrŒr\Z linemarkerZ linecolorZbaselineZ stemlinesZ markerlineZstem_containerZjxÚjyZjzrNr€rOÚstem‚ s^B     ÿ    ÿ    ÿ   ÿÿ  ÿ ÿ z Axes3D.stem)N)N)N)NN)NN)NN)NN)NNF)N)TrÒN)NN)NTTT)NNTF)NNNr)N)r rrrs)reNFF)NN)T)rÒ)NN)N)ru)FrurN)rN)rN)rr)rrrNT)rr)NrTN)Nr)NNNrEFr NNNNFFFFFF)NTN)”Ú__name__Ú __module__Ú __qualname__Ú__doc__rCZ _axis_namesrZGrouperrr1rZdeprecate_privatize_attributeÚdistrrZsxZsyr7r8r9rUrCr[r]rZget_zgridlinesZget_zticklinesÚ deprecatedÚpropertyZw_xaxisZw_yaxisZw_zaxisrdrbrormrqrpr‰r‹r™r«rwZallow_rasterizationr¸rÂrÃrØr×rÇrÑrÚrârÐreZmake_keyword_onlyrúr)Zset_xlimr“Zset_ylimr”r•rürýrþZ get_zscaleZ set_xscaleZ set_yscaleZ set_zscaleÚmaprBZ get_zticksZ set_zticksZget_zmajorticklabelsZget_zminorticklabelsZget_zticklabelsZset_zticklabelsZ zaxis_dateÚtextwrapÚdedentrÿr/r)rrºr<r%r&r'rr-rArBr>rArDrSr@rVr rmrhrXrsrurvrxr0ryr|r~r}rèrZtext3DZtext2Dr‰Zplot3DrÅrÒrØrårérìrërròZ contour3Dr÷rúrûZ contourf3DrürÞr Z scatter3Dr rr r>Zquiver3Dr@rrrvr„Zstem3DÚ __classcell__rNrNrLrOr*sÖ      ÿýt    ÿ ÿ ÿ   U& 7  $ # ÿ > ÿ     ò     þ  F C   (=$ÿ 0&      )ÿ Dvÿ a 0ÿ  ÿ%ÿ4 0! þG6ÿ !ÿ ÿ Rý EÿÿÿtrrcCs°t dd|¡}}t ||¡\}}t |d|d d¡dtj}t |ddd|ddd d¡dtjdd}||}|d}|d}|d}|||fS) z,Return a tuple X, Y, Z with a test data set.gÀg@rrr gø?rriô)rŒrJZmeshgridÚexprc)rërrrÞrßZZ1ZZ2ràrNrNrOÚ get_test_dataú s&*ÿr‘)r)8rˆrÚrr5r¶r+rZnumpyrŒZ matplotlibr.rrrrZmatplotlib.artistr³rwZmatplotlib.axesZaxesr)Zmatplotlib.collectionsr­Zmatplotlib.colorsrr¬Zmatplotlib.imageÚimageräZmatplotlib.linesrŒråZmatplotlib.patchesr r¯r‚Ú containerrnZmatplotlib.transformsZ transformsr¦rZmatplotlib.axes._baserr r Zmatplotlib.tri._triangulationr rEr rrZinterpdZdefine_aliasesrr‘rNrNrNrOÚsr                ÿf