o hh@sZdZddlmZddlZddlZddlZddlZddlZddl Z ddl m Z m Z m Z mZddlmZddlmZddlmZddlmZddlmZddlmZddl m!Z"ddl#m$Z%ddl&m'Z'ddl(m)Z)m*Z*ddl#m+Z+ddl,m-Z-d d l.m/Z/d d l.m0Z0d d l.m1Z1e j2e 3d gdgdgdGddde'Z4dddZ5GdddZ6dS)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)_apicbook _docstring_preprocess_data)Axes)_axis_method_wrapper_process_plot_format)Bbox) Triangulation)art3d)proj3d)axis3dxlim3dylim3dzlim3d)xlimylimzlimc seZdZdZdZdZeej d<eej d< d ddd ddd ddd d fd d Z ddZ ddZ ddZ ddZddZddZeddZeddZddZd fd!d" Zd#d$Zd%d&d'd(Zd d)d*Zejfd+d,Zd-d.Zd/d0Zedd1Zedd2Zd3d4Z d5d6Z!dddd d7d8d9Z"d d;d<Z#d d=d>Z$  dd?d@Z%dAdBZ& ddCdDZ'ddEdFZ(ddGdHZ)ddIdJZ*d d d ddddKdLdMZ+d d d ddddNdOdPZ,d d d ddddQdRdSZ-d d d ddddTdUdVZ.e,Z/e-Z0e.Z1dWdXZ2dYdZZ3d[d\Z4edd]Z5ed^d_Z6ed`d_Z7edd_Z8e9daj:gd\e6_e7_e8_eddbZ;eddcZeddfZ?eddgdhdiidjZ@eddkZAeAjrfeAjeBCdl7_dmdnZD  ddodpZEd dqdrZF ddsdtdueGdveHjIfdwdxZJdydzZKdd}d~ZLddZMddZNddZOddZPddZQfddZRddZSddZTddZUddZVddZWd ddZXddZYddZZddZ[d ddZ\de]de]dveHjIfddZ^ddZ_ddZ`ddZa dddZbddZcddZdddZed ddZfddZgdZhdZidddZjdfdd ZkddZleddZmddZnd d dfdd ZoeoZpejoZqdd dœfddĄ ZrerZsddddd dƜddȄZtddddd dɜdd˄Zud ddd̈́Zvdddddd dΜddЄZwdddӄZx  dddՄZydd dddׄZzdddلZ{e|d dddd dڜfdd܄ Z}e}Z~e|d dddd dڜfddބ ZddZe|ddd dfdd ZeZe|ddd dfdd Zdd dfdd Ze|gdd dd dfdd ZeZe|dd dfdd Ze|  dd dddZdfdd Ze|d%ddd d dddZeZddd dd dddZe|gdd  dddZd d dd dfdd Ze|dddd ddd ddd ZeZZS( 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`. 3dxyzrviewNirperspT) elevazimroll shareviewsharez proj_type focal_length box_aspectcomputed_zorderc s|durgd}||_||_||_||| | |_t|_t|_d}t||gd|d|gg|_ t|_ | |j|j|j||_ |durW|j d||d|_||_|durg|j d||| dd rqtd tj||g| Rd | d | t|d|_d|_d |_|d|_||jd d}|jj d|j!|jj d|j"|jj d|j#|$|j%&d|j'()ddg}|d|d\|_*|_+|j,dd-d dS)a 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. shareview : Axes3D, optional Other Axes to share view angles with. Note that it is not possible to unshare axes. sharez : Axes3D, optional Other Axes to share z-limits with. Note that it is not possible to unshare axes. proj_type : {'persp', 'ortho'} The projection type, default 'persp'. 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) 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. **kwargs Other optional keyword arguments: %(Axes3D:kwdoc)s N)r(?r)F]tE?r rdatalimrauto_add_to_figureFzSauto_add_to_figure is no longer supported for Axes3D. Use fig.add_axes(ax) instead.T)frameonr&UUUUUU?rootmotion_notify_eventbutton_press_eventbutton_release_eventrrr)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_axesjoin _adjustable _shareviewpopAttributeErrorsuper__init__ set_axis_off set_axis_onMinvM _view_marginautoscale_view fmt_zdata mouse_init get_figurecanvas callbacks_connect_picklable_on_move _button_press_button_release set_top_viewpatch set_linewidth transLimitsinverted transform _pseudo_w _pseudo_hspines set_visible)selffigrectrr r!r"r#r$r%r&r'argskwargsxymargin pseudo_bbox __class__o/var/www/html/construction_image-detection-poc/venv/lib/python3.10/site-packages/mpl_toolkits/mplot3d/axes3d.pyrG<st?        zAxes3D.__init__cCsd|_d|_dS)NFT _axis3donstalerarjrjrkrH zAxes3D.set_axis_offcCsd|_d|_dSNTrlrorjrjrkrIrpzAxes3D.set_axis_oncCs |j|S)zs For artists in an Axes, if the zaxis has units support, convert *z* using zaxis unit type )zaxis convert_units)rarrjrjrkconvert_zunitss zAxes3D.convert_zunitscCsNd|j}d|j}d|j}d|j}| |f|j_| |f|j_d|_dS)Ngffffff?g?T)_distviewLim intervalx intervalyrn)raxdwlxdwydwlydwrjrjrkrWs     zAxes3D.set_top_viewcCs(t||_t||_t||_dS)z5Init 3D Axes; overrides creation of regular X/Y Axes.N)rXAxisxaxisYAxisyaxisZAxisrrrorjrjrk _init_axiss  zAxes3D._init_axiscC|jS)z0Return the ``ZAxis`` (`~.axis3d.Axis`) instance.)rrrorjrjrk get_zaxisszAxes3D.get_zaxisrr get_gridlines get_ticklinesc Csb|\}}}}}}|||f|||f|||f|||f|||f|||f|||f|||fg}t||jS)z:Return cube with limits from *vals* transformed by self.M.)r _proj_pointsrJ) ravalsminxmaxxminymaxyminzmaxzxyzsrjrjrk_transformed_cubeszAxes3D._transformed_cubeFcsvtjd|dtjd|||d||_|dvr||}t|j |j |j g}tj |dd}|j dkr~tj|dd}t|||j|} | |j} t|j|j|jfD] \} } | |vr{| || | | d || | | d d d d q[d St|j} ||| |<hd |}|r|}tj|j|}tj| |}| |||9<|| d Sd 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 )autoequalequalxyequalyzequalxzaspectr)r adjustableanchorsharerrrrr axisr+@TNr view_margin>rr )r check_in_listrF set_aspect_aspect_equal_aspect_axis_indicesnparrayr~get_view_intervalrrrptprBmeanmax _box_aspect enumerate set_xlim3d set_ylim3d set_zlim3d differencerDlinalgnormset_box_aspect)rarrrr ax_indicesview_intervalsrrscaledeltasiset_limr&remaining_ax_indices remainingold_diagnew_diagrhrjrkrsL0    & zAxes3D.set_aspectcCsTg}|dkr gd}|S|dkrddg}|S|dkr ddg}|S|dkr(ddg}|S) a Get the indices for which of the x, y, z axes are constrained to have equal aspect ratios. Parameters ---------- aspect : {'auto', 'equal', 'equalxy', 'equalxz', 'equalyz'} See descriptions in docstring for `.set_aspect()`. rrr rrrr rrrrj)rarrrjrjrkrMs z!Axes3D._equal_aspect_axis_indicesr )zoomcCs~|dkr td|d|durtjdtd}ntj|td}tjd|d|d |tj|9}|j|d d |_ d |_ dS) a{ Set the Axes box aspect. The box aspect is the ratio of height to width in display units for each face of the box when viewed perpendicular to that face. This is not to be confused with the data aspect (see `~.Axes3D.set_aspect`). The default ratios are 4:4:3 (x:y:z). To simulate having equal aspect in data space, set the box aspect to match your data range in each dimension. *zoom* controls the overall size of the Axes3D in the figure. Parameters ---------- aspect : 3-tuple of floats or 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). zoom : float, default: 1 Control overall size of the Axes3D in the figure. Must be > 0. rzArgument zoom = z must be > 0N)rdtype)rrg }?Treverse) ValueErrorrasarrayfloatr check_shaperr_roll_to_verticalrrn)rarrrjrjrkrbs zAxes3D.set_box_aspectcCsp|dur |jdd}|j}tj|}|j|j}d}| }| |||}| | | |ddS)NT)originalr active) get_positionrPtransSubfigure mtransformsr r9 transformedheightwidthfrozenshrunk_to_aspect _set_positionanchored get_anchor)rapositiontransbb fig_aspectr&pbpb1rjrjrk apply_aspects   zAxes3D.apply_aspectc s`|sdS||j|d|_|}||r |||nd||_t j |j|_ dd|j D}|jrutdd|jDd}|}}t|dddd D]}t|tjrf||_|d7}qVt|tjrs||_|d7}qVn |D]}|qw|jr|jD]}||q|jD]}||q|jD]}||qt|dS) NFcss.|]}t|tjtjfr|r|VqdSN) isinstancemcoll CollectionmpatchesPatch get_visible).0artistrjrjrk s zAxes3D.draw..css|]}|VqdSr) get_zorder)rrrjrjrkrsr cSs|Sr)do_3d_projection)rrjrjrkszAxes3D.draw..T)keyr)r_unstale_viewLimrXdraw_frameonget_axes_locatorrget_projrJrrinvrK _childrenr'r _axis_mapvaluessortedrrrzorderrrrrm draw_pane draw_gridrF) rarendererlocatorcollections_and_patches zorder_offsetcollection_zorder patch_zorderrrrhrjrkrsR         z Axes3D.drawcCs`||}|dd|ddk}|dd|ddk}|dd|ddk}|||fS)Nr rrr)r get_w_lims)ratcxhighyhighzhighrjrjrkget_axis_positions  zAxes3D.get_axis_positioncKdS)zK Not implemented in `~mpl_toolkits.mplot3d.axes3d.Axes3D`. Nrj)raxysrerjrjrkupdate_datalimszAxes3D.update_datalim_get_autoscale_on_set_autoscale_oncCr)z Retrieve autoscaling margin of the z-axis. .. versionadded:: 3.9 Returns ------- zmargin : float See Also -------- mpl_toolkits.mplot3d.axes3d.Axes3D.set_zmargin )_zmarginrorjrjrk get_zmarginszAxes3D.get_zmargincCs*|dkrtd||_|dd|_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 gz margin must be greater than -0.5rTN)rr _request_autoscale_viewrn)ramrjrjrk set_zmargins   zAxes3D.set_zmargin)rrrtightcGs|r|dus|dus|durtdt|dkr!|d}}}nt|dkr-|\}}}n|r3td|durT|durT|durT|durLtd|d |j|j|jfS|dur]|||durf|||duro| ||j ||du|du|dud 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 rrzYMust pass a single positional argument for all margins, or one for each margin (x, y, z).Tzignoring tight=z in get mode)rscalexscaleyscalez) TypeErrorlenr warn_external_xmargin_ymarginr  set_xmargin set_ymarginrrM)rarrrrmarginsrjrjrkrs,       zAxes3D.marginsbothcCs|dur d}d}d}n0|dvr|||}nd}|dvr)|||}nd}|dvr9|||}nd}|rD|jd|d|rM|jd |d|rX|jd |ddSdS) 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)rrF)rrrrr)rrr)set_autoscalex_onget_autoscalex_onset_autoscaley_onget_autoscaley_onset_autoscalez_onget_autoscalez_onr)raenablerrrrrrjrjrk autoscale2s.      zAxes3D.autoscalecCst|t|kr|jtt|t|g| n|j|| |j|| |dur:|j|| | dSr) rshaper<update_from_data_xy column_stackravelupdate_from_data_xupdate_from_data_yr=rM)raXYZhad_datarjrjrkauto_scale_xyzUs zAxes3D.auto_scale_xyzcCs|dur&|j}|s%|jD]}t|tjrd}q t|tjtjfr$d}nq nt |}|_|rl| rl|j j \}}|j } | ||\}}|jdkrZ|||j} || 8}|| 7}|sd| ||\}}||||j|r|r|j j\} } |j} | | | \} } |jdkr| | |j} | | 8} | | 7} |s| | | \} } || | |j|r|r|jj \}}|j}|||\}}|jdkr|||j} || 8}|| 7}|s|||\}}||||jdSdSdS)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. NTFr)_tightrrmimage AxesImagemlinesLine2Drrboolr r<rwr~get_major_locator nonsingularr view_limits set_xboundrLr"rxrr set_yboundr$r=rrr  set_zbound)rarrrrr2rx0x1xlocatordeltay0y1ylocatorz0z1zlocatorrjrjrkrMcs\               zAxes3D.autoscale_viewcCs4|\}}|\}}|\}}||||||fS)zGet 3D world limits.) get_xlim3d get_ylim3d get_zlim3d)rarrrrrrrjrjrkrs   zAxes3D.get_w_limsc Cs`|dur t|r |\}}|\}}|dur|}|dur|}|t||ft|dd|ddS)z% Set 3D axis bounds. Nrr)riterablerr7) ra get_boundr axis_invertedlowerupperr old_lower old_upperrjrjrk _set_bound3ds  zAxes3D._set_bound3dcC||j|j|j|||dS)a Set the lower and upper numerical bounds of the x-axis. This method will honor axis inversion regardless of parameter order. It will not change the autoscaling setting (`.get_autoscalex_on()`). Parameters ---------- lower, upper : float or None The lower and upper bounds. If *None*, the respective axis bound is not modified. view_margin : float or None The margin to apply to the bounds. If *None*, the margin is handled by `.set_xlim`. See Also -------- get_xbound get_xlim, set_xlim invert_xaxis, xaxis_inverted N)rR get_xboundset_xlimxaxis_invertedrarNrOrrjrjrkr;zAxes3D.set_xboundcCrS)a Set the lower and upper numerical bounds of the y-axis. This method will honor axis inversion regardless of parameter order. It will not change the autoscaling setting (`.get_autoscaley_on()`). Parameters ---------- lower, upper : float or None The lower and upper bounds. If *None*, the respective axis bound is not modified. view_margin : float or None The margin to apply to the bounds. If *None*, the margin is handled by `.set_ylim`. See Also -------- get_ybound get_ylim, set_ylim invert_yaxis, yaxis_inverted N)rR get_yboundset_ylimyaxis_invertedrWrjrjrkr<rXzAxes3D.set_yboundcCrS)a Set the lower and upper numerical bounds of the z-axis. This method will honor axis inversion regardless of parameter order. It will not change the autoscaling setting (`.get_autoscaley_on()`). Parameters ---------- lower, upper : float or None The lower and upper bounds. If *None*, the respective axis bound is not modified. view_margin : float or None The margin to apply to the bounds. If *None*, the margin is handled by `.set_zlim`. See Also -------- get_zbound get_zlim, set_zlim invert_zaxis, zaxis_inverted N)rR get_zboundset_zlimzaxis_invertedrWrjrjrkr=szAxes3D.set_zboundemitrraxminaxmaxc Cs|durt|r|\}}n |dur|d}|dur&|dur&|d}|dur4|dur2td|}|durB|dur@td|}t|sLt|rWtd|d|d|durftjd rd|j}nd}|||} || 8}|| 7}|j ||||d S) z% Set 3D axis limits. Nr rz"Cannot pass both 'lower' and 'min'z"Cannot pass both 'upper' and 'max'z Axis limits z, z cannot be infinitezaxes3d.automarginr`r) rrKrrisinfrmplrcParamsrL_set_lim) rarrNrOr`rrrarbrArjrjrk _set_lim3ds2      zAxes3D._set_lim3d)r`rrxminxmaxc C|j|j|||||||dS)aX Set the 3D x-axis view limits. Parameters ---------- left : float, optional The left xlim in data coordinates. Passing *None* leaves the limit unchanged. The left and right xlims may also be passed as the tuple (*left*, *right*) as the first positional argument (or as the *left* keyword argument). .. ACCEPTS: (left: float, right: float) right : float, optional The right xlim in data coordinates. Passing *None* leaves the limit unchanged. emit : bool, default: True Whether to notify observers of limit change. auto : bool or None, default: False Whether to turn on autoscaling of the x-axis. *True* turns on, *False* turns off, *None* leaves unchanged. view_margin : float, optional The additional margin to apply to the limits. xmin, xmax : float, optional They are equivalent to left and right respectively, and it is an error to pass both *xmin* and *left* or *xmax* and *right*. Returns ------- left, right : (float, float) The new x-axis limits in data coordinates. See Also -------- get_xlim set_xbound, get_xbound invert_xaxis, xaxis_inverted Notes ----- The *left* value may be greater than the *right* value, in which case the x-axis values will decrease from *left* to *right*. Examples -------- >>> set_xlim(left, right) >>> set_xlim((left, right)) >>> left, right = set_xlim(left, right) One limit may be left unchanged. >>> set_xlim(right=right_lim) Limits may be passed in reverse order to flip the direction of the x-axis. For example, suppose ``x`` represents depth of the ocean in m. The x-axis limits might be set like the following so 5000 m depth is at the left of the plot and the surface, 0 m, is at the right. >>> set_xlim(5000, 0) r_)rhr~)raleftrightr`rrrirjrjrjrkrU EzAxes3D.set_xlim)r`rryminymaxc Crk)aV Set the 3D y-axis view limits. Parameters ---------- bottom : float, optional The bottom ylim in data coordinates. Passing *None* leaves the limit unchanged. The bottom and top ylims may also be passed as the tuple (*bottom*, *top*) as the first positional argument (or as the *bottom* keyword argument). .. ACCEPTS: (bottom: float, top: float) top : float, optional The top ylim in data coordinates. Passing *None* leaves the limit unchanged. emit : bool, default: True Whether to notify observers of limit change. auto : bool or None, default: False Whether to turn on autoscaling of the y-axis. *True* turns on, *False* turns off, *None* leaves unchanged. view_margin : float, optional The additional margin to apply to the limits. ymin, ymax : float, optional They are equivalent to bottom and top respectively, and it is an error to pass both *ymin* and *bottom* or *ymax* and *top*. Returns ------- bottom, top : (float, float) The new y-axis limits in data coordinates. See Also -------- get_ylim set_ybound, get_ybound invert_yaxis, yaxis_inverted Notes ----- The *bottom* value may be greater than the *top* value, in which case the y-axis values will decrease from *bottom* to *top*. Examples -------- >>> set_ylim(bottom, top) >>> set_ylim((bottom, top)) >>> bottom, top = set_ylim(bottom, top) One limit may be left unchanged. >>> set_ylim(top=top_lim) Limits may be passed in reverse order to flip the direction of the y-axis. For example, suppose ``y`` represents depth of the ocean in m. The y-axis limits might be set like the following so 5000 m depth is at the bottom of the plot and the surface, 0 m, is at the top. >>> set_ylim(5000, 0) r_)rhr)rabottomtopr`rrrorprjrjrkrZhrnzAxes3D.set_ylim)r`rrzminzmaxc Crk)aV Set the 3D z-axis view limits. Parameters ---------- bottom : float, optional The bottom zlim in data coordinates. Passing *None* leaves the limit unchanged. The bottom and top zlims may also be passed as the tuple (*bottom*, *top*) as the first positional argument (or as the *bottom* keyword argument). .. ACCEPTS: (bottom: float, top: float) top : float, optional The top zlim in data coordinates. Passing *None* leaves the limit unchanged. emit : bool, default: True Whether to notify observers of limit change. auto : bool or None, default: False Whether to turn on autoscaling of the z-axis. *True* turns on, *False* turns off, *None* leaves unchanged. view_margin : float, optional The additional margin to apply to the limits. zmin, zmax : float, optional They are equivalent to bottom and top respectively, and it is an error to pass both *zmin* and *bottom* or *zmax* and *top*. Returns ------- bottom, top : (float, float) The new z-axis limits in data coordinates. See Also -------- get_zlim set_zbound, get_zbound invert_zaxis, zaxis_inverted Notes ----- The *bottom* value may be greater than the *top* value, in which case the z-axis values will decrease from *bottom* to *top*. Examples -------- >>> set_zlim(bottom, top) >>> set_zlim((bottom, top)) >>> bottom, top = set_zlim(bottom, top) One limit may be left unchanged. >>> set_zlim(top=top_lim) Limits may be passed in reverse order to flip the direction of the z-axis. For example, suppose ``z`` represents depth of the ocean in m. The z-axis limits might be set like the following so 5000 m depth is at the bottom of the plot and the surface, 0 m, is at the top. >>> set_zlim(5000, 0) r_)rhrr)rarqrrr`rrrsrtrjrjrkr]rnzAxes3D.set_zlimcC t|jjSr)tupler:rwrorjrjrkget_xlim zAxes3D.get_xlimcCrur)rvr:rxrorjrjrkget_ylimrxzAxes3D.get_ylimcCru)a Return the 3D z-axis view limits. Returns ------- left, right : (float, float) The current z-axis limits in data coordinates. See Also -------- set_zlim set_zbound, get_zbound invert_zaxis, zaxis_inverted Notes ----- The z-axis may be inverted, in which case the *left* value will be greater than the *right* value. )rvr;rwrorjrjrkget_zlims zAxes3D.get_zlim get_scaler~_set_axes_scalera 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. get_ticklocs set_ticksget_majorticklabelsget_minorticklabelsget_ticklabelsset_ticklabelszAxis.set_tickszAxes3D.set_zticks)doc_sub 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. cOr)z:Currently not implemented for 3D Axes, and returns *None*.Nrj)rardrerjrjrkclabelCz Axes3D.clabelcCsd|_|dur |j}|dur|j}|dur|j}tjddt|jD|d}|r7dd|jd |D}n|g}|D]}||_ ||_ ||_ ||_ qsz$Axes3D.view_init..) vertical_axiscSsh|]}|qSrjrj)rsiblingrjrjrk sz#Axes3D.view_init..r)rur6r5r7r check_getitemr _axis_namesr@ get_siblingsrr r!_vertical_axis)rarr r!rraxesaxrjrjrkr>Gs,4zAxes3D.view_initcCsxtjddg|d|dkr%|durd}n |dkr td|d||_dS|dtjfvr6td|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) rortho)r$Nr rzfocal_length = z must be greater than 0z must be None for proj_type = )rrr _focal_lengthrinf)rar$r%rjrjrkr8s   zAxes3D.set_proj_typearrznp.typing.ArrayLikerreturncCs,|r t||jddSt||jdS)z Roll arrays to match the different vertical axis. Parameters ---------- arr : ArrayLike Array to roll. reverse : bool, default: False Reverse the direction of the roll. r)rr!r)rarrrjrjrkrs zAxes3D._roll_to_verticalcCsN||j}tjg|||Rd|i}d|}t|j }t|j }t |t |}t |t |}t |}||||g} ||j | } || \} } } | |_| |_| |_|jtjkr|t| | | | }t|j |j }n||j | |j}t| | | |}t|j |j |j}t||}t||}|S)z?Create the projection matrix from the current viewing position. pb_aspect?)rrrworld_transformationrHrIrJrdeg2radrr cossinru_calc_view_axes_view_u_view_v_view_wrr_view_transformation_uvw_ortho_transformation_persp_transformationdot)rar&worldMRelev_radazim_radp0p1p2pseyeuvwviewMprojM eye_focalM0rJrjrjrkrsD        zAxes3D.get_projrrcCs: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_1dtolist _rotate_btn_pan_btn _zoom_btn)ra rotate_btnpan_btnzoom_btnrjrjrkrOs zAxes3D.mouse_initcCs|jgggddS)zrr r!)rarrrjrjrkr"1s  zAxes3D.shareviewcsxt|jtjkrtjd|_nd|_d}t||gd|d|gg|_ t |_ d|_ | |tjddS)Nz axes.zmarginr(r*r r.z axes3d.grid)rFclearrrrrerfr r r<r9r=rLrMgrid)rarfrhrjrkrBs   z Axes3D.clearcCsl|j|kr2|j|_|j|j|_|_|jddjj }|r&| dur&| |r4| | |dSdSdSNTr/)inaxesbuttonrxdataydata_sx_syrPrQtoolbar _nav_stack push_current set_message_mouse_event_to_messageraeventrrjrjrkrUTs zAxes3D._button_presscCsJd|_|jddjj}|r|dur||r#|||dSdSr)rrPrQrget_navigate_moderrrrrjrjrkrV^szAxes3D._button_releasecCs:||||||d|j|j|jffS)N)r autoscalex_onr autoscaley_onr autoscalez_on) rwr ryr"rzr$rr r!rorjrjrk _get_viewhs   zAxes3D._get_viewcCs4|\}\}}}|jdi|||_||_||_dS)Nrj)setrr r!)rarpropsrr r!rjrjrk _set_viewps  zAxes3D._set_viewc Cs<z||WSttfy|jj}||}|YSw)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 )rNrErrrget_major_formatterformat_data_short)rarfuncvalrjrjrk format_zdataxs  zAxes3D.format_zdatacCs8d}|j|jvr|}|S|jdur||||}|S)z Return a string giving the current view rotation angles, or the x, y, z coordinates of the point on the nearest axis pane underneath the mouse cursor, depending on the mouse button pressed. N)rr_rotation_coordsrJ_location_coords)raxvyvrcoordsrjrjrk format_coords  zAxes3D.format_coordcCsNt|j}t|j}t|j}d|dd|dd|dddd}|S)z9 Return the rotation angles as a string. z elevation=z.0fu °, azimuth=u °, roll=°-u−)r _norm_anglerr r!replace)ra norm_elev norm_azim norm_rollrrjrjrkrs    zAxes3D._rotation_coordsc Cs||||\}}||d}||d}||d}|dkr/d|d|d|} | S|dkr@d|d|d|} | S|dkrOd|d|d |} | S) zY Return the location on the axis pane underneath the cursor as a string. rr rzx pane=z, y=z, z=zx=z , y pane=z , z pane=) _calc_coord format_xdata format_ydatar) rarrrrpane_idxxsyszsrrjrjrkrszAxes3D._location_coordsc Csn|\}}}}}}t|||g}t|||g}|jtjkr#d} n|j} ||j|j||j| } | S)zJ Returns the current camera location in data coordinates. geA)_get_w_centers_rangesrrrrrrur) racxcyczdxdydzcrr%rrjrjrk_get_camera_locs zAxes3D._get_camera_loccCs|jtjkr d}nd|j}tt||||j}||}g}|j D] }| \} } | | q)t d} tdD]} || dkrNtj| | <q@|| || || | | <q@tt| } | | }|||}|| fS)z Given the 2D view coordinates, find the point on the nearest axis pane that lies directly below those coordinates. Returns a 3D point in data coordinates. r rrr)rrrrr inv_transformrKr*rrr active_paneappendzerosrangeargminabs)rarrrzvrvec pane_locsrrlocscalesrrrrrjrjrkrs$          zAxes3D._calc_coordrrc Cstjdd}tjd|}||}||}||||}t|}d|}|d|d}d|d|} || krJttd|||g} | S||kro||} t|t|| || ||g} | tj| } | Std||||g} | S)au Convert a point (x, y) to a point on a virtual trackball. This is Ken Shoemake's arcball (a sphere), modified to soften the abrupt edge (optionally). See: Ken Shoemake, "ARCBALL: A user interface for specifying three-dimensional rotation using a mouse." in Proceedings of Graphics Interface '92, 1992, pp. 151-156, https://doi.org/10.20380/GI1992.18 The smoothing of the edge is inspired by Gavin Bell's arcball (a sphere combined with a hyperbola), but here, the sphere is combined with a section of a cylinder, so it has finite support. axes3d.trackballsizerzaxes3d.trackballborderr r)rerfrsqrtrrr) rarrsbr2rraaripdrrjrjrk_arcballs& &zAxes3D._arcballcCs|jsdS|dur dS|jdurdS|j|j}}|dus$|j|kr&dS||j||j}}|j}|j }|j|j vr|dkrH|dkrHdSt j d}|dkrt |j} || dt | ||dt | } || dt | ||dt | } |j| } |j| } |j} nvtjt |j|j|jf}|dkrt d| |||g}t j|}|t j d}tt ||t ||}n,||j||j|}|||||}|dkrt||}n td|td| }||}t |\} } } |j|j}|j| | | |d d d |_ n=|j|j!vrF|j"#|j|jg\}}|$||d |%d d|j&|j'|(n|j|j)vrZ|||}|*||||||_|_|j+d d j,-dS) z Mouse moving. By default, button-1 rotates, button-2 pans, and button-3 zooms; these buttons can be modified via `mouse_init`. Nrzaxes3d.mouserotationstyleazel trackballrsphereTrrr/).rrrJrrrrrr]r^rrerfrrr!rrrr _Quaternionfrom_cardan_anglesrrrrrotate_from_torad2degas_cardan_anglesrrr>rnr transDatar\ start_pandrag_panrrend_panr_scale_axis_limitsrPrQ draw_idle)rarrrrrrhstyler!delevdazimrr qknkthdq current_vecnew_vecrpxpyrrjrjrkrTsr    ..       zAxes3D._on_movecCs\|j}|j||f|j|jfg\\}}\}} |||_|_|||||||| } } d} |dkr7d} n|dkr=d} | dkrG| dkrGdSt |j |j |j g} | |j |j} | jt | | | g}|\}}}}}}|||d}|||d}|||d}|j||||dd|j||||dd|j||||dddS)Nrrrr rr) _pan_start trans_inverser\rrrrr(rrrrrrruTrrrr)rarrrrrrr xdata_start ydata_startdudvdwrduvw_projectedrrrrrrrrrrjrjrkr)fs0zAxes3D.drag_panc Cs~tt|j}tt|j}d||j}td}t |tj dkr*dnd||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 rrrrr )rrr rrr!rrr r pirr _view_axes) rarrroll_radrVrrrrjrjrkrs   zAxes3D._calc_view_axesincCs|\}}}} |dkr|jjd}|jjd} n|dkr'|jjd}|jjd}tt||g|jjd|jjd\}}tt|| g|jjd|jjd\}} ||d} || d} |jjd|jjdd} |jjd|jjdd} || | d|dd| | |t ||}t || }||jjd|jjd}||jjd|jjd}t||}|dkrd|}| |||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 rrrNout) bboxminrrcliprr(r)r*r _zoom_data_limits)rarJ directionmodetwinxtwinystart_xstart_ystop_xstop_y zoom_center_x zoom_center_y ax_center_x ax_center_yrrscale_uscale_vrrjrjrk_set_view_from_bboxs8         zAxes3D._set_view_from_bboxc Cst|||g}t||sJt|j|j|jg}|td}tjj|j |dd}|j dvrJ| |j }t t ||d}|||||<||d|d|ddS)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). rr rrrrN)rrallcloserrrrrrr=rrr r r+) rarZr[scale_wrrSax_idxs min_ax_idxsrjrjrkrMs   zAxes3D._zoom_data_limitsc Cs|\}}}}}} |j|||d|||ddd|j|||d|||ddd|j|| |d|| |ddddS)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. rNr:)rrrr) rascale_xscale_yscale_zrrrrrrrjrjrkr+s((,zAxes3D._scale_axis_limitsc Cs`|\}}}}}}||d}||d}||d} ||} ||} ||} ||| | | | fS)z%Get 3D world centers and axis ranges.r)r) rarrrrrrrrrrrrrjrjrkr s   zAxes3D._get_w_centers_rangescKs&|dur||j_|jj||fi|S)zI Set zlabel. See doc for `.set_ylabel` for description. N)rrlabelpadset_label_text)razlabelfontdictrererjrjrk set_zlabelszAxes3D.set_zlabelcCs|jj}|S)z. Get the z-label text string. )rrlabelget_text)rarjrjrjrk get_zlabel#szAxes3D.get_zlabelcKst|rd}||_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 _draw_gridrn)ravisiblererjrjrkr1s  z Axes3D.gridc stjgd|d|dvrtj|fi||dvrBt|}|dd|dd|dd|d d|jjd i|dSdS) 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. )rrrrr)rrrrrrNrqlabeltop labelbottomrj)rrrF tick_paramsdictrDrrset_tick_params)rarrezkwrhrjrkrqAs    zAxes3D.tick_paramscCs |\}}|j||dddS)z Invert the z-axis. See Also -------- zaxis_inverted get_zlim, set_zlim get_zbound, set_zbound Nr:)rzr])rarqrrrjrjrk invert_zaxis_s zAxes3D.invert_zaxis get_invertedcCs$|\}}||kr||fS||fS)z Return the lower and upper z-axis bounds, in increasing order. See Also -------- set_zbound get_zlim, set_zlim invert_zaxis, zaxis_inverted )rz)rarNrOrjrjrkr\ns zAxes3D.get_zbound) axlim_clipc  s,tj|||fi|}t|||||S)a Add the text *s* to the 3D Axes at location *x*, *y*, *z* in data coordinates. Parameters ---------- x, y, z : float The position to place the text. s : str The text. zdir : {'x', 'y', 'z', 3-tuple}, optional The direction to be used as the z-direction. Default: 'z'. See `.get_dir_vector` for a description of the values. axlim_clip : bool, default: False Whether to hide text that is outside the axes view limits. .. versionadded:: 3.10 **kwargs Other arguments are forwarded to `matplotlib.axes.Axes.text`. Returns ------- `.Text3D` The created `.Text3D` instance. )rFtextr text_2d_to_3d) rarrrrzdirrwrerxrhrjrkrx~sz Axes3D.text)rzrwc s|}|rt|dts|^}}d|vrtdn|dd}t|||\}}}tj||g|Ri|} | D] } t j | |||dq:t ||||\}}}| ||||| 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. axlim_clip : bool, default: False Whether to hide data that is outside the axes view limits. .. versionadded:: 3.10 **kwargs Other arguments are forwarded to `matplotlib.axes.Axes.plot`. rrz,plot() for multiple values for argument 'zs'rrzrw) has_datarstrrrDr_broadcast_with_masksrFplotr line_2d_to_3d juggle_axesr1) rarrrzrwrdrer0rlineslinerhrjrkrs z Axes3D.plotr)whererO facecolorsshaderwc KsPtjgd|d|} t||||||\}}}}}}| dur'|jg} tt | } |dur5d}nt j |t d}|j |j krOtd|j d|j d|t |@}|d krtjt ||||ft ||||ft ||||fd d rd }nd }| dur|d krd} nd} g}t|D]\}}|||}|||}|||}|||}|||}|||}t|sq|d krSt|d}t |ddf}t |dd|dd|ddf|dddddf<t |dd|dd|ddf|dddddf<t |dd|dd|ddf|dddddf<t |dd|dd|ddf|dddddf<|g|}q|d krt |||f}t |ddd|ddd|dddf}t j||fdd}||qtj|f| | | d| }|||||g||g||g| |S)a Fill the area between two 3D curves. The curves are defined by the points (*x1*, *y1*, *z1*) and (*x2*, *y2*, *z2*). This creates one or multiple quadrangle polygons that are filled. All points must be the same length N, or a single value to be used for all points. Parameters ---------- x1, y1, z1 : float or 1D array-like x, y, and z coordinates of vertices for 1st line. x2, y2, z2 : float or 1D array-like x, y, and z coordinates of vertices for 2nd line. where : array of bool (length N), optional Define *where* to exclude some regions from being filled. The filled regions are defined by the coordinates ``pts[where]``, for all x, y, and z pts. More precisely, fill between ``pts[i]`` and ``pts[i+1]`` if ``where[i] and where[i+1]``. Note that this definition implies that an isolated *True* value between two *False* values in *where* will not result in filling. Both sides of the *True* position remain unfilled due to the adjacent *False* values. mode : {'quad', 'polygon', 'auto'}, default: 'auto' The fill mode. One of: - 'quad': A separate quadrilateral polygon is created for each pair of subsequent points in the two lines. - 'polygon': The two lines are connected to form a single polygon. This is faster and can render more cleanly for simple shapes (e.g. for filling between two lines that lie within a plane). - 'auto': If the points all lie on the same 3D plane, 'polygon' is used. Otherwise, 'quad' is used. facecolors : list of :mpltype:`color`, default: None Colors of each individual patch, or a single color to be used for all patches. shade : bool, default: None Whether to shade the facecolors. If *None*, then defaults to *True* for 'quad' mode and *False* for 'polygon' mode. axlim_clip : bool, default: False Whether to hide data that is outside the axes view limits. .. versionadded:: 3.10 **kwargs All other keyword arguments are passed on to `.Poly3DCollection`. Returns ------- `.Poly3DCollection` A `.Poly3DCollection` containing the plotted polygons. )rquadpolygon)rONTrz where size (z) does not match size ()rg-q=)atolrrFr rrrrrr)rrrw)rrr|rr~_get_patches_for_fillget_next_colorlistmcolors to_rgba_arrayrrr7sizerisnanr _all_points_on_plane concatenatecontiguous_regionsremptyr)r Poly3DCollectionadd_collectionr1)rar?rCrFx2y2z2rrOrrrwrer0polysidx0idx1x1iy1iz1ix2iy2iz2i n_polys_ipolys_iline1line2polypolycrjrjrk fill_betweenst>             :::: .   zAxes3D.fill_between)rvminvmax lightsourcerwc" s|} |jdkr tdt|}t|||\}}}|j\} } d| vp(d| v} d| vp0d| v}| r9|r9td| dd| dd| dd }| dd }t j d rY|}n| }|rvt t t | |d t t t | |d | d d }| dd }| d|d u}|d urtdg}| d dkr| d dkr|d urtjfdd|||fDdd}n\ttd| d | d g}ttd| d | d g}g}t|D]6\t|D],\fdd|||fD}t|j}|||d ur||qqt|tjr&t|s\g}g}t||D]"\}}t|t|jd d}t|rQ||||q0|}|d ur\|}|d urqtj|f|||||d| }nn|rtj|fd|i| }t|tjr|djdd} n tdd|D} | | |d us|d ur|!|||d ur|"|n&| dd }!|!d ur|j#$}!tt%&|!}!tj|f|!|||d| }|'||(|||| |S)a 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 : :mpltype:`color` Color of the surface patches. cmap : Colormap, optional Colormap of the surface patches. facecolors : list of :mpltype:`color` Colors of each individual patch. norm : `~matplotlib.colors.Normalize`, optional Normalization for the colormap. vmin, vmax : float, optional 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`, optional The lightsource to use when *shade* is True. axlim_clip : bool, default: False Whether to hide patches with a vertex outside the axes view limits. .. versionadded:: 3.10 **kwargs Other keyword arguments are forwarded to `.Poly3DCollection`. r!Argument Z must be 2-dimensional.rstridecstridercountccount.Cannot specify both stride and count argumentsr2_internal.classic_moder rNcmaprzshade cannot be None.rcsg|] }t|qSrj)r_array_patch_perimetersrr)rrrjrk sz'Axes3D.plot_surface..rrc s.g|]}t|ddfqS)r )r_array_perimeterr)cscs_nextrsrs_nextrjrkrs ) edgecolorsrrrrwrw).rcSs g|] }|dddfqSNr)r)rrrjrjrkrs color)rrrrw))r|ndimrr_to_unmasked_float_arrayrbroadcast_arraysr'rDrerfintrceilgetstackrr  itertoolspairwiserr=r rndarrayisfiniteall zip_longestrr rr set_arrayset_climset_norm _get_linesrrto_rgbarr1)"rar-r.r/rrrrrwrer0rowscols has_stride has_countrrcompute_stridesfcolorsrrcolsetrrow_indscol_indsr new_polys new_colsetrcolnew_polyravg_zrrj)rrrrrrrk plot_surfaceMsK                           zAxes3D.plot_surfacec s|}jdkr tdt\j\}}d|vp#d|v} d|vp+d|v} | r4| r4td|dd} |dd} |dd } |dd }tjd rt| rs| rat t t || dnd } |rqt t t ||dnd } n"| s| rt t t || dnd } |rt t t ||dnd } t t t | rt td || }|d kr|d |dkr||dg7}ng}| rt td || }|d kr|d |dkr||dg7}ng}| d kr| d krtd jd krg}g}fdd|D}fdd|D}fdd|D}fdd|D}fdd|D}fdd|D}ddt|||Dddt|||D}tj|fd|i|}|||||S)at 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. axlim_clip : bool, default: False Whether to hide lines and patches with vertices outside the axes view limits. .. versionadded:: 3.10 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`. rrrrrrrr rrrrz*Either rstride or cstride must be non zerocg|]}|qSrjrjrr)r-rjrkr} z)Axes3D.plot_wireframe..crrjrjr)r.rjrkr~ rcrrjrjr)r/rjrkr rcrrjrjr)tXrjrkr rcrrjrjr)tYrjrkr rcrrjrjr)tZrjrkr rcS"g|] \}}}tt|||qSrjrziprxlylzlrjrjrkr cSrrjrrrjrjrkr rrw)r|rrrrr'rDrerfrrr transposerr rrr Line3DCollectionrr1)rar-r.r/rwrer0rrrrrrrrriiciixlinesylineszlinestxlinestylinestzlinesrlinecrj)r-r.r/rrrrkplot_wireframe sp,           "    zAxes3D.plot_wireframe)rrrrrrwcOs|} |dur |j}tt|}|dd} |d| du} t j |i|\} }}z|d} Wn t yB|^} }Ynwt | } | }| j|}| j|}| |}tj|||fdd}| rtj|g|Rd|i|}|dddddfjd d}|||dus|dur||||dur||ntj|g|R| |||d |}|||| j| j| | |S) a Plot a triangulated surface. The (optional) triangulation can be specified in one of two ways; either:: plot_trisurf(triangulation, ...) where triangulation is a `~matplotlib.tri.Triangulation` object, or:: plot_trisurf(X, Y, ...) plot_trisurf(X, Y, triangles, ...) plot_trisurf(X, Y, triangles=triangles, ...) in which case a Triangulation object will be created. See `.Triangulation` for an explanation of these possibilities. The remaining arguments are:: plot_trisurf(..., Z) where *Z* is the array of values to contour, one per point in the triangulation. Parameters ---------- X, Y, Z : array-like Data values as 1D arrays. color Color of the surface patches. cmap A colormap for the surface patches. norm : `~matplotlib.colors.Normalize`, optional An instance of Normalize to map values to colors. vmin, vmax : float, optional Minimum and maximum value to map. shade : bool, default: True Whether to shade the facecolors. Shading is always disabled when *cmap* is specified. lightsource : `~matplotlib.colors.LightSource`, optional The lightsource to use when *shade* is True. axlim_clip : bool, default: False Whether to hide patches with a vertex outside the axes view limits. .. versionadded:: 3.10 **kwargs All other keyword arguments are passed on to :class:`~mpl_toolkits.mplot3d.art3d.Poly3DCollection` Examples -------- .. plot:: gallery/mplot3d/trisurf3d.py .. plot:: gallery/mplot3d/trisurf3d_2.py Nrrr/rrrwrr )rrrrw)r|rrrrrrrrDr get_from_args_and_kwargsKeyErrorrget_masked_trianglesrrrr rrrrrrr1)rarrrrrrwrdrer0rrtrir trianglesxtytztvertsrrrjrjrk plot_trisurf sZ9           zAxes3D.plot_trisurfc s"|jd|jdd}g}g}t|jD]f\}}||}g|} ||} t| ||t| ||tdsBqtt td|d} tdd| d| fddt t | dD| | gt | dq| tj t|||dd|dS) z4 Extend a contour in 3D by creating r rrc s`g|],}dt|dt|ddt|ddt|fqSrr )roundrbotstepsizerrrjrkr s ((z-Axes3D._3d_extend_contour..T)rrrN)levelsr get_paths_iter_connected_components get_edgecolorr _paths_to_3d_segmentsrrr extendr add_collection3drrrremove) racsetstrider polyvertscolorsrlevelpathsubpathsrnstepsrjr rk_3d_extend_contour s,     zAxes3D._3d_extend_contourcCs@d|}|r|||dStj||dur|n|j||ddS)Nrr{)rr collection_2d_to_3dr )rarextend3drrzoffsetrwrjrjrkadd_contour_set s zAxes3D.add_contour_setcCs|j||||ddS)Nrzr rw)_add_contourf_set)rarrzr rwrjrjrkadd_contourf_set s  zAxes3D.add_contourf_setcCsd|}|jddt|jd}|jr.|jdt|jddd}t|d|}|jrH|jdt|jddd}t||}tj||durQ|n|||d|S)z Returns ------- levels : `numpy.ndarray` Levels at which the filled contours are added. rNrrrr{) r rdiff _extend_mininsert _extend_maxr r r)rarrzr rw midpoints min_level max_levelrjrjrkr#! s"" zAxes3D._add_contourf_set)rrrzr rwcsd|} t||||\} } }tj| | |g| Ri| }|||||||||||| |S)a Create a 3D contour plot. Parameters ---------- X, Y, Z : array-like, Input data. See `.Axes.contour` for supported data shapes. extend3d : bool, default: False Whether to extend contour in 3D. stride : int, default: 5 Step size for extending contour. 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*. axlim_clip : bool, default: False Whether to hide lines with a vertex outside the axes view limits. .. versionadded:: 3.10 data : indexable object, optional DATA_PARAMETER_PLACEHOLDER *args, **kwargs Other arguments are forwarded to `matplotlib.axes.Axes.contour`. Returns ------- matplotlib.contour.QuadContourSet )r|r rotate_axesrFcontourr!r1)rar-r.r/rrrzr rwrdrer0jXjYjZrrhrjrkr.8 s "zAxes3D.contourcs|}tj|i|\} }}| j} | j} d|vr|d} n|^} }t| | | |\} }}t| || j| j } t j | |g|Ri|}| ||||||| | | | ||S)a Create a 3D 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.tricontour` for supported data shapes. extend3d : bool, default: False Whether to extend contour in 3D. stride : int, default: 5 Step size for extending contour. 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*. axlim_clip : bool, default: False Whether to hide lines with a vertex outside the axes view limits. .. versionadded:: 3.10 data : indexable object, optional DATA_PARAMETER_PLACEHOLDER *args, **kwargs Other arguments are forwarded to `matplotlib.axes.Axes.tricontour`. Returns ------- matplotlib.tri._tricontour.TriContourSet r/)r|r rrrrDr r-rmaskrF tricontourr!r1)rarrrzr rwrdrer0rr-r.r/r/r0r1rrhrjrkr3e s"%  zAxes3D.tricontourcs>d|d|d|||ifdddD}|jg||RdS)Nrrrcs(g|]}t|t|fqSrjrnanminnanmax)rdimdim_valsrjrkr s z/Axes3D._auto_scale_contourf..r)r1)rar-r.r/rzr r0limitsrjr8rk_auto_scale_contourf s  zAxes3D._auto_scale_contourfr"csd|} t||||\} } } tj| | | g|Ri|} || |||}||||||| | S)aG 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*. axlim_clip : bool, default: False Whether to hide lines with a vertex outside the axes view limits. .. versionadded:: 3.10 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-rFcontourfr#r;)rar-r.r/rzr rwrdrer0r/r0r1rr rhrjrkr< s zAxes3D.contourfcs|}tj|i|\}}}|j}|j} d|vr|d} n|^} }t|| | |\} } } t| | |j|j }t j || g|Ri|}| ||||}| || | ||||S)a 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. axlim_clip : bool, default: False Whether to hide lines with a vertex outside the axes view limits. .. versionadded:: 3.10 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 rrrrDr r-rr2rF tricontourfr#r;)rarzr rwrdrer0rr-r.r/r/r0r1rr rhrjrkr= s"   zAxes3D.tricontourfc s|}t|}|jrt|nd}t|tjur)tj ||||d| |n+t|tj ur?tj ||||d| |nt|tj urTtj||||d| ||rt|tjrk|jt|jd|int|tjr|j|jddd|int|tjr 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, *zs* and *zdir*. Supported 2D collection types are: - `.PolyCollection` - `.LineCollection` - `.PatchCollection` (currently not supporting *autolim*) Parameters ---------- col : `.Collection` A 2D collection object. zs : float or array-like, default: 0 The z-positions to be used for the 2D objects. zdir : {'x', 'y', 'z'}, default: 'z' The direction to use for the z-positions. autolim : bool, default: True Whether to update the data limits. axlim_clip : bool, default: False Whether to hide the scatter points outside the axes view limits. .. versionadded:: 3.10 rr{r0Nr)r|rrrrKtyperPolyCollectionr poly_collection_2d_to_3d set_sort_zposLineCollectionline_collection_2d_to_3dPatchCollectionpatch_collection_2d_to_3drrr1r _segments3drr_vecPatch3DCollectionrFr) rarrrzautolimrwr0zvalszsortval collectionrhrjrkr s>           zAxes3D.add_collection3d) rrrrrr facecolorrr) replace_namesc s|} |} t|||\}}}tj|}t|||||| dd\}}}}}} | ddur5| | d<t| |r?| }t j ||g| R||d| }t j |||||d|jdkri|jdkri|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 : :mpltype:`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. axlim_clip : bool, default: False Whether to hide the scatter points outside the axes view limits. .. versionadded:: 3.10 data : indexable object, optional DATA_PARAMETER_PLACEHOLDER **kwargs All other keyword arguments are passed on to `~.axes.Axes.scatter`. Returns ------- paths : `~matplotlib.collections.PathCollection` rN)rr)rrz depthshaderw皙?r)r|rr~rmar*delete_masked_pointsrmay_share_memorycopyrFscatterr rEr rrr1)rarrrrzrrrPrwrdrer0zs_origrpatchesrhrjrkrVH s(7  "  zAxes3D.scattercs|}tj||g|Ri|} tj|t|dd}g} g} t| |D]+\} } t| }| | 7} | | gt|7} t | | ||d|vrP| |dq%t| dkr^t| \}}ngg}}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, default: 0 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'). axlim_clip : bool, default: False Whether to hide bars with points outside the axes view limits. .. versionadded:: 3.10 data : indexable object, optional DATA_PARAMETER_PLACEHOLDER **kwargs Other keyword arguments are forwarded to `matplotlib.axes.Axes.bar`. Returns ------- mpl_toolkits.mplot3d.art3d.Patch3DCollection T)subokalphar)r|rFbarr broadcast_torrr _get_patch_vertsrpatch_2d_to_3d set_alpharr1)rarlrrrzrwrdrer0rXrverts_zsrrvsrrrhrjrkr[ s&    z Axes3D.baraveragec Os|}tt||||||\}}}}}}t|}t||}t|}t||}t|}t||}tgd}t|j|j}d||fd||fd||ffD]%\}}}|dtj tj f}|dtj tj f}|||d|f|d|f<qY| d|jdd}g}|dur|j g}t t|}t|t|kr|D] }||gdqn|}t|t|kr|dt|9}tj|g| R||| | | d | }|||||f||f||f||S) aR 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 : {'average', 'min', 'max'}, default: 'average' 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`, optional The lightsource to use when *shade* is True. axlim_clip : bool, default: False Whether to hide the bars with points outside the axes view limits. .. versionadded:: 3.10 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. ))rrrrr rr r rr rr)rrr r rr r r r rr r )rcrfrhrg)rdrjrire)rcrgrjrd)rfrerirhrr r.)rN)zsortrrrrw)r|rrrrKrrrr'newaxisreshaperrrrrrrr rrr1)rarrrrrrrrlrrrwrdrer0rrrrrrcuboidrrrdprrrrjrjrkbar3d sTD   .&  z Axes3D.bar3dcenterc s<tj|f||d|}|j\}}|jd||S)N)rhrgq= ףp?)rF set_titletitlerset_y)rarjrhrreretrrrhrjrkrs szAxes3D.set_titleg333333?tail)lengtharrow_length_ratiopivot normalizerwc Ksdd} |}tj||||||dd}tdd|Dr-tjgfi| }|||Stjd|gt d}||}t j gd | d | d krL||8}n | d krV||d 8}t |dd}t |dd t }| rtjj|dd}d||dk<||d}t|dkr|tj||dd}| |}|ddddftj||}|t|ddf}|dd}g||ddd |ddd }ng}tj|fd| i| }||||dddf|dddf|ddd f||S)a 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*. axlim_clip : bool, default: False Whether to hide arrows with points outside the axes view limits. .. versionadded:: 3.10 data : indexable object, optional DATA_PARAMETER_PLACEHOLDER **kwargs Any additional keyword arguments are delegated to :class:`.Line3DCollection` c SsX|dddf}|dddf}tjj|ddddfdd}tj|||dkt|d}tj| ||dkt|d}td}t|}t |}||} ||} ||d|} t ||dd|| | g| ||dd|| g| | t ||gg} | } | gdgdfd 9<t d | |}t d | |}tj||gddS) Nrr rr)rrI)rr rr)rrrr rzij...,...j->...i)rrrdivide zeros_like ones_likemathradiansrrr full_likerUeinsumr)UVWrrrx_py_pranglerrr13r32r12RposRneg Rpos_vecs Rneg_vecsrjrjrk calc_arrows s,    z"Axes3D.quiver..calc_arrowsT)compresscss|] }t|dkVqdS)rN)r)rrrjrjrkr z Axes3D.quiver..r(r)rwmiddletip)rzrwrrNrr rr)rr rrw)r|rr~anyr rrrrrrrr)astyperrrnrmultiplyouterswapaxesr1)rar-r.r/UrGWrxryrzr{rwrerr0 input_argsrshaft_dtarrow_dtXYZrrshafts head_dirsheadsrrjrjrkquiver sD0     " & 4z Axes3D.quiverrrrrrwc2 st|dkr dd}ndd}||i|\} }jdkr"tdtjjtjd} t| d| dur>t\} } } n fd d | D\} } } fd d }|durY|j }||d }||d}| | | | tjgdgdgdgdgtjd}t t }dd}|dD]}|j| \}}}t|}t|}t|}||j}|ddd}|D]}|D]}|||dg}t|}|r||||t|D]C\}} ||||g}!|||| g}"t|!}#t|"}$|#r|$s||#|"|q׈|#s|$r||$|"|q||||dg}%||||g}&t|%}'|'r?||'|&|qqqi}(|D]u\})}*| durT|*}+nHg}+|*D]C},|,dddf|,dddf|,dddff}-t|,j}.| |-|.dddf<| |-|.dddf<| |-|.dddf<|+|.qX||)}/||)}0tj|+f|/|0|||d|}1||1|1|(|)<qH|(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`, optional The lightsource to use when *shade* is True. axlim_clip : bool, default: False Whether to hide voxels with points outside the axes view limits. .. versionadded:: 3.10 **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 rc[s|||f||fSrrj) _Axes3D__x _Axes3D__y _Axes3D__zfilledrerjrjrkvoxelsX szAxes3D.voxels..voxelsc[s d||fSrrj)rrerjrjrkr[ s z%Argument filled must be 3-dimensionalrr Nc3s|] }t|VqdSr)rr\)rr) coord_shaperjrkrj rz Axes3D.voxels..cspt|dvrt|jt|St|dvr0t|ddjkr.td|d|Std|d)Nr)rrrzWhen multidimensional, z must match the shape of filledzInvalid z argument)rrr\r'r)rr)rrjrk_broadcast_color_argl s z+Axes3D.voxels.._broadcast_color_argrrrcrfrerdcss:tj|tjd}t|D] }|Vtj|ddd}q dS)z%Generate cyclic permutation matrices.rr rrN)rrintpr r!)nmatrrjrjrkpermutation_matrices s  z+Axes3D.voxels..permutation_matricesrrrr)rrrrrr'rrvindicesrrr1rrr=raranger rritemsrr rr)2rarrrrrwrdrerxyzrrrrrsquare voxel_facesrpermutepcqcrcpindsqindsrindssquare_rot_possquare_rot_negrr1ri0r1rrri1i2pkpk2ikpolygonscoord faces_indsfaces face_indsindfacerM edgecolorrrj)rrrkr s L             " .   z Axes3D.voxels)rrrxerryerrzerrrcD sX|}t|tj}dd|D}|dd|jd|fd|fd|fg|dd t |r2|n|g}t |r<|n|g}t |rF|n|g}t |t |kr\t |ksat d t d | || | d d }d |d <|jj||dkr}||fn|||f|dd\\}}tj|||d|r||ddn ||dd|dkr||nd }| dd|vrd|d<| d ur|d} dD]}| |d qi|d| i}| r| |d<n d|vr|d|d<dD] }||vr||||<qi|ddi}| d urtjd} | dkrd| |d<| d ur"| |d<| |d<d d!}fd"d#ggg}}} g}!d$d$d%d&}"dd'dd(|dtjd)d}#|#|jdd*jd+9}#|jd,|#|#fg}#ttj|#dd-}#tj|dddd.tj !|"}$Wd n 1swYt#|$|#dddgd'}#|#d/9}#i||#d'd0}%|% dd t$gd(|||g|||g|||g|||gD]\}&}'t%|&|&}(d urqt sgt |'t&t't |'(t)t't |'(t)fd1d2t*|||gD})|)\\}*}+\},}-\}.}/B}0|0+ry| dkry||*|,|.g|0@}1||+|-|/g|0@}2tj,|1d|"|(|d3|}3tj,|2d|"|(|d3|}4||3||4|-|3|-|4+r||+|-|/g@\}5}6}7|j.|5|6|7gRi|%+r||*|,|.g@\}8}9}:|j.|8|9|:g Ri|%tj/t0|)j1fd4|i|};|2|;|-|;|!-|)qt0|!}!fd5d6}<|<|!d\}=}>|<|!d\}?}@|<|!d\}A}B|3|=|>f|?|@f|A|Bf|t4j5|t6|t6|f|d up|d u|d u|d7}C|j7-|C||| fS)8ag 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 : :mpltype:`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, `~.set_xlim`, `~.set_ylim`, or `~.set_zlim` 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. axlim_clip : bool, default: False Whether to hide error bars that are outside the axes limits. .. versionadded:: 3.10 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|] \}}|dur||qSrrj)rr2rrjrjrkrVsz#Axes3D.errorbar..rrrrrF)convertz)'x', 'y', and 'z' must have the same sizerjN _nolegend_rT) return_kwargs)rrwg?nonerC0) marker markersizemarkerfacecolormarkeredgewidthmarkeredgecolor markevery linestyle fillstyle drawstyle dash_capstyledash_joinstylesolid_capstylesolid_joinstyle linewidth)r\rZr rasterizedrNonezerrorbar.capsizerrrrcsfdd|DS)Ncsg|] }gt|qSrj)rr)rrr2rjrkrsz8Axes3D.errorbar.._apply_mask..rj)arraysr2rjrrk _apply_masksz$Axes3D.errorbar.._apply_maskcsZt|jdkr |\}}n||}}t|B|||}t|B|||}||fSr)rr'rr)errdatalomaskhimasklow_errhigh_errlowshighs) everymaskrjrk _extract_errss   z&Axes3D.errorbar.._extract_errs|_rr rzlines.markersizer/Hr4r)rr r!gV&,t=?)rxrycs&g|]\}}||qSrjrj)rrr)r dir_vectorrlolimsuplimsrjrkrsz#Axes3D.errorbar..)lsrrwrwc sLt|dd|ddddft|dd|ddddffSrr4)err_arr coord_label)i_xyzrjrk_digout_minmaxs$$z'Axes3D.errorbar.._digout_minmax)has_xerrhas_yerrrj)8r|rnormalize_kwargsr5r6r setdefault_process_unit_inforrKrr_errorevery_to_maskrDr _plot_argsr r set_zorderrNadd_linererfrrPdpi transAxesr[r\rr& _setattr_cmrrrrrget_dir_vectorrr\rr7rrLine3Dr rrrr=rr1 mcontainerErrorbarContainerrv containers)Drarrrrrrfmt barsabove erroreveryecolor elinewidthcapsizecapthickxlolimsxuplimsylolimsyuplimszlolimszuplimsrwrer0rj data_line base_stylereb_lines_style eb_cap_stylererrlinescaplineslimmarks coorderrs capmarker quiversizerKeb_quiver_stylerzri_zdircoorderrrxhryhrzhnolims lo_caps_xyz hi_caps_xyzcap_locap_hixh0yh0zh0xl0yl0zl0errlinerrrrrrrerrorbar_containerrj)rrrrrrrrkerrorbar sq                                       zAxes3D.errorbarcall_axes_locatorbbox_extra_artistsfor_layout_onlyc s^tj||||d}|g}|jr)|jD]}|r(t||}|r(||qt j |S)Nr9) rF get_tightbboxrmrrrmartist_get_tightbbox_for_layout_onlyr rr union) rarr:r;r<rvbatchraxis_bbrhrjrkr=.s   zAxes3D.get_tightbboxzC0-C0ozC3-)linefmt markerfmtbasefmtrqrj orientationrwcsddlm} |} tjgd| dt|t|f} t|t|f}t|t|f}| dkrM||}}||}}fddt|||D}n3| dkri|| }}||}}fd dt|||D}n|| }}||}}fd dt|||D}t |\}}}|d urt j d }|j |||| d d\}t j|||d | d}|||j ||||d d\}| |||f|d}||t ||g| \}}}|g|| g||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, optional The label to use for the stems in legends. orientation : {'x', 'y', 'z'}, default: 'z' The direction along which stems are drawn. axlim_clip : bool, default: False Whether to hide stems that are outside the axes limits. .. versionadded:: 3.10 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) StemContainerr)rGrcs&g|]\}}}||f|||fgqSrjrjrthisxthisythiszrqrjrkrzAxes3D.stem..rcs&g|]\}}}||f|||fgqSrjrjrIrMrjrkrrNcs&g|]\}}}||f|||fgqSrjrjrIrMrjrkrrNNzlines.linestyler)rrzrj) linestylesrrjrw)rj)matplotlib.containerrHr|rrrrKrrr rerfrr rr add_containerrr1)rarrrrDrErFrqrjrGrwrHr0rrrbasexbasexlimbaseybaseylimrr linemarker linecolorbaseline stemlines markerlinestem_containerjxjyjzrjrMrkstem>sX G                  (z Axes3D.stemr)NNF)TrN)NN)NTTT)NNN)NNNrF)F)r rr)rHNFF)T)r)r)FrrNF)rN)rNF)rrT)rrrONT)rr)NrbTN)Nrr)NNNrFr NNNNFFFFFFF)__name__ __module__ __qualname____doc__rrrGrouperrr@rGrHrIrtrWrrrget_zgridlinesget_zticklinesrrrrrr>allow_rasterizationrrr r$r#r rrr&r1rMrrRr;r<r=rhrUrZr]rrrrwryrz get_zscale set_xscale set_yscale set_zscalemapformat get_zticks set_zticksget_zmajorticklabelsget_zminorticklabelsget_zticklabelsset_zticklabels zaxis_datetextwrapdedentrr>r8r7rrrrrOrrrr#r"rrUrVrrrrrrrrrrrTr)rr\rMr+rrirl get_frame_on set_frame_onrrqrur^r\rxtext3Dtext2Drplot3Drrrrrr!r$r#rr. contour3Dr3r;r< contourf3Dr=rrV scatter3Dr[rqrsrquiver3Drr8r=r_stem3D __classcell__rjrjrhrkr)s      V );   $ # ;      H H H            M  ;      %!^$ 0&    ,  H| f   * 9 $4 CP;(  XN  zrrQcCstdd|}}t||\}}t|d|d ddtj}t|ddd|ddd ddtjdd}||}|d}|d}|d}|||fS) z,Return a tuple X, Y, Z with a test data set.gg@rr g?rri)rrmeshgridexprD)rArrr-r.Z1Z2r/rjrjrk get_test_datas&* rc@seZdZdZddZddZddZdd Zed d Z d d Z ddZ ddZ e Z ddZddZddZeddZeddZddZdS)r"zb Quaternions consisting of scalar, along 1, and vector, with components along i, j, k cCs||_t||_dSr)scalarrrvector)rarrrjrjrkrGsz_Quaternion.__init__cCs||j |j Srrirrrorjrjrk__neg__sz_Quaternion.__neg__cCsH||j|jt|j|j|j|j|j|jt|j|jS)a Product of two quaternions i*i = j*j = k*k = i*j*k = -1 Quaternion multiplication can be expressed concisely using scalar and vector parts, see )rirrrrcrossrarrjrjrk__mul__s z_Quaternion.__mul__cCs||j|j S)z5The conjugate quaternion -(1/2)*(q+i*q*i+j*q*j+k*q*k)rrorjrjrk conjugatesz_Quaternion.conjugatecCs|j|jt|j|jS)zThe 2-norm, q*q', a scalar)rrrrrorjrjrkrsz_Quaternion.normcCs$t|j}||j||j|S)zScaling such that norm equals 1)rrrrirrrarrjrjrkr{s z_Quaternion.normalizecCs |j}||j||j |S)z.The reciprocal, 1/q = q'/(q*q') = q' / norm(q))rrirrrrjrjrk reciprocalsz_Quaternion.reciprocalcCs ||Sr)rrrjrjrk__div__s z_Quaternion.__div__cCs|d|}|||}|jS)Nr)rir)rarrjrjrkrotates  z_Quaternion.rotatecCs|j|jko |j|jkjSr)rrrrrjrjrk__eq__sz_Quaternion.__eq__cCsdt|jt|jS)Nz_Quaternion({}, {}))rmreprrrrorjrjrk__repr__sz_Quaternion.__repr__cCst||}tj|}t|t||}|d}|dkrWt||dkrNtdtd}d|t ||<t||}|tj|}|d|}|S|dgd}|S|t ||t ||}|S)a The quaternion for the shortest rotation from vector r1 to vector r2 i.e., q = sqrt(r2*r1'), normalized. If r1 and r2 are antiparallel, then the result is ambiguous; a normal vector will be returned, and a warning will be issued. rrz6Rotation defined by anti-parallel vectors is ambiguousrr rc) rrrrarctan2rwarningswarnr r rr)clsrrr2r3r4r1rjrjrkr$ s"      z_Quaternion.rotate_from_tocCst|dt|d}}t|dt|d}}t|dt|d}} |||||| } ||| |||} |||||| } ||| |||} || | | | gS)z Converts the angles to a quaternion q = exp((roll/2)*e_x)*exp((elev/2)*e_y)*exp((-azim/2)*e_z) i.e., the angles are a kind of Tait-Bryan angles, -z,y',x". The angles should be given in radians, not degrees. r)rrr)rrr r!casacesecrsrqwqxqyqzrjrjrkr#%sz_Quaternion.from_cardan_anglescCs|j}|jdddf\}}}td| |||||||||||}ttd||||||||||||dd}td||||||||||||}|||fS)z The inverse of `from_cardan_angles()`. Note that the angles returned are in radians, not degrees. The angles are not sensitive to the quaternion's norm(). .Nrrr )rrrrarcsinrL)rarrrrr rr!rjrjrkr&7s :D8 z_Quaternion.as_cardan_anglesN)r`rarbrcrGrrrpropertyrr{rr __truediv__rrr classmethodr$r#r&rjrjrjrkr"s(     r")rQ)7rc collectionsrrrrurnumpyr matplotlibrerrrrmatplotlib.artistrr>matplotlib.collectionsrmatplotlib.colorsrrmatplotlib.imageimager3matplotlib.linesrr5matplotlib.patchesrXrrP containerr matplotlib.transforms transformsrmatplotlib.axesrmatplotlib.axes._baserr r matplotlib.tri._triangulationr rr rrinterpddefine_aliasesrrr"rjrjrjrks|                 /