a Sic@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 gdgdgdGddde)Z6dddZ7dS)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)axis3dZxlim3dZylim3dZzlim3dxlimylimzlimcseZdZdZdZdZeej d<e dZ dddd dd dd dd fd d Z ddZddZddZddZddZddZeddZeddZe jddd d!ed"d#Ze jdd$d d!ed%d#Ze jddd d!ed&d#Zdd'd(Zdd)d*Zdd+d,Zdfd.d/ Zd0d1d2d3Z dd4d5Z!e"j#fd6d7Z$d8d9Z%d:d;Z&edd<Z'edd=Z(d>d?Z)dddd d@dAdBZ*ddDdEZ+ddFdGZ,ddHdIZ-dJdKZ.e /ddLddddMdNdOZ0e1jj2Z3e1jj4Z5e0Z6dPdQZ7dRdSZ8dTdUZ9eddVZ:eddWZ;ed$dWZdXj?gd\e;_e<_e=_eddYZ@eddZZAedd[ZBedd\ZCedd]ZDedd^d_d`idaZEeddbZFeFjreFjeGHdc7_dddeZIddfdgZJddhdiZKdjdkZLdldmZMddodpZNdqdrZOdsdtZPdudvZQdwdxZRfdydzZSd{d|ZTd}d~ZUddZVddZWddZXddZYddZZdddZ[ddZ\ddZ]ddZ^e _ddddddZ`dfdd ZaddZbddZcddZddddZedfdd ZfefZgejfZhddfdd ZieiZjdddddddZkddZldddZmddZnddddddddZodddZpdddZqdddZrdddZsetd-ddddfdd ZueuZvetd-ddddfdd ZwddZxetdddfddÄ ZyeyZzetdddfddń Z{dfddDŽ Z|etgdȢdɍdfdd̄ Z}e}Z~etdfdd΄ ZetdddфZdfddԄ Zetd0ddd-dלddلZeZddd ddڜdd܄ZetgdݢdɍdddZdd-dfdd Zetdddd dddddZeZZS(Axes3Dz 3D Axes object. 3dxyzrz3.6NirperspT)elevazimrollsharez proj_type box_aspectcomputed_zorder focal_lengthcs|durgd}||_||_||_||| | |_t|_t|_t|_ t|_ | |j|j|j||_ |dur|j d||d|_| dd} tj||g| Rd|d| t|d|_d|_||jjjd |j|jjjd |j|jjjd |j||j !d |j"#$d dg}|d|d \|_%|_&|j'dd(d| rt)j*dddd|+|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. 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. auto_add_to_figure : bool, default: False Prior to Matplotlib 3.4 Axes3D would add themselves to their host Figure on init. Other Axes class do not do this. This behavior is deprecated in 3.4, the default is changed to False in 3.6. The keyword will be undocumented and a non-False value will be an error in 3.7. 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)r$?r%rdatalimauto_add_to_figureFT)frameonr!motion_notify_eventbutton_press_eventbutton_release_eventrrr)r r r z3.4z3.7aPAxes3D(fig) adding itself to the figure is deprecated since %(since)s. Pass the keyword argument auto_add_to_figure=False and use fig.add_axes(ax) to suppress this warning. The default value of auto_add_to_figure is changed to False in mpl3.6 and True values will no longer work %(removal)s. This is consistent with other Axes classes.)removalmessage), initial_azim initial_elev initial_roll set_proj_typer"r unit xy_viewLim zz_viewLim xy_dataLim zz_dataLim view_init_sharez _shared_axesjoin _adjustablepopsuper__init__ set_axis_off set_axis_onM fmt_zdata mouse_initfigurecanvas callbacks_connect_picklable_on_move _button_press_button_release set_top_viewpatch set_linewidth transLimitsinverted transform _pseudo_w _pseudo_hspines set_visiblerwarn_deprecatedadd_axes)selffigrectrrrrr r!r"r#argskwargsr'Z pseudo_bbox __class__W/var/www/html/django/DPS/env/lib/python3.9/site-packages/mpl_toolkits/mplot3d/axes3d.pyr?8sd@            zAxes3D.__init__cCsd|_d|_dS)NFT _axis3donstalerXr_r_r`r@szAxes3D.set_axis_offcCsd|_d|_dS)NTrardr_r_r`rAszAxes3D.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)rXrr_r_r`convert_zunitsszAxes3D.convert_zunitscCsNd|j}d|j}d|j}d|j}| |f|j_| |f|j_d|_dS)Ngffffff?g?T)_distviewLim intervalx intervalyrc)rXZxdwlZxdwZydwlZydwr_r_r`rLs    zAxes3D.set_top_viewcCs(t||_t||_t||_dS)z5Init 3D axes; overrides creation of regular X/Y axes.N)rXAxisxaxisYAxisyaxisZZAxisrerdr_r_r` _init_axiss  zAxes3D._init_axiscCs|jS)z0Return the ``ZAxis`` (`~.axis3d.Axis`) instance.rerdr_r_r` get_zaxisszAxes3D.get_zaxisre get_gridlines get_ticklinesz3.1rmz3.8) alternativer-cCs|jSN)rmrdr_r_r`zAxes3D.rocCs|jSrv)rordr_r_r`rwrxcCs|jSrvrqrdr_r_r`rwrxc Cs\|p |\}}}}}}|||f|||f|||f|||f|||f|||f|||f|||fgSrv) get_w_lims)rXvalsminxmaxxminymaxyminzmaxzr_r_r` unit_cubeszAxes3D.unit_cubecCs(|dur|j}||}t||}|Srv)rBrrZ proj_points)rXrzrBZxyzsZtcuber_r_r` tunit_cubes   zAxes3D.tunit_cubecCs|||}|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 )r)rXrzrBtcedgesr_r_r` tunit_edgess zAxes3D.tunit_edgesFcs6tjd|dtjd|||d||_|dvr2|dkrDgd}n4|dkrVd d g}n"|d krhd d g}n|d krxd d g}t|j|j |j g}tj |d d}tj |d d}t ||} |j|| kd } | |j| } t|j|j|jfD]:\} } | |vr| || | | d|| | | dqdS)a} 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 Currently ignored by Axes3D 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 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 adjustableanchorshare)rrrrrrr rrrr rrraxis@N)r check_in_listr> set_aspect_aspectnparrayrmget_view_intervalroremeanptpmax _box_aspect enumerate set_xlim3d set_ylim3d set_zlim3d)rXrrrrZ ax_indicesZview_intervalsrrdeltascaledeltasiset_limr]r_r`rs>2      zAxes3D.set_aspectr )zoomcCst|dkrtd|d|dur0tjdtd}ntj|td}tjd|d|d |tj|9}||_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 (which for Axes3D is always 'auto'). 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)rrrdtype)rrgIr|E?T) ValueErrorrasarrayfloatr check_shapelinalgnormrrc)rXrrr_r_r`set_box_aspect`szAxes3D.set_box_aspectcCsp|dur|jdd}|j}tj|}|j|j}d}| }| |||}| | | |ddS)NT)originalr active) get_position get_figuretransSubfigure mtransformsr r3 transformedheightwidthfrozenshrunk_to_aspect _set_positionanchored get_anchor)rXpositiontransbb fig_aspectr!pbpb1r_r_r` apply_aspects   zAxes3D.apply_aspectc s:|s dS||j|d|_|}||r@|||nd||_dd|j D}|j rt dd|j Dd}|}}t|dddd D]:}t|tjr||_|d7}qt|tjr||_|d7}qn|D] }|q|jr*|j D]}||q|j D]}||qt|dS) NFcss,|]$}t|tjtjfr|r|VqdSrv) isinstancemcoll CollectionmpatchesPatch get_visible).0artistr_r_r` szAxes3D.draw..css|]}|VqdSrv) get_zorder)rrr_r_r`rsr cSs|Srv)do_3d_projection)rr_r_r`rwrxzAxes3D.draw..T)keyreverse)r_unstale_viewLimrMdraw_frameonget_axes_locatorrget_projrB _childrenr"r _axis_mapvaluessortedrrrzorderrrrrbZ draw_paner>) rXrendererlocatorZcollections_and_patchesZ zorder_offsetZcollection_zorderZ patch_zorderrrr]r_r`rsH         z Axes3D.drawcCsh|}|||j}|dd|ddk}|dd|ddk}|dd|ddk}|||fS)Nr rrr)ryrrB)rXrzrZxhighZyhighZzhighr_r_r`get_axis_positions zAxes3D.get_axis_positioncKsdSrvr_)rXxysr\r_r_r`update_datalimszAxes3D.update_datalim_get_autoscale_on_set_autoscale_oncCs*|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)r_zmargin_request_autoscale_viewrc)rXmr_r_r` set_zmargins  zAxes3D.set_zmargin)rrrtightcGs|r&|dus|dus|dur&tdnBt|dkrD|d}}}n$t|dkr\|\}}}n |rhtd|dur|dur|dur|durtd|d |j|j|jfS|dur|||dur|||dur| ||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_ymarginrautoscale_view)rXrrrrmarginsr_r_r`rs,       zAxes3D.marginsbothcCs|durd}d}d}nl|dvr6|t||}nd}|dvrZ|t||}nd}|dvr~|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)rrF)rrrrr)rrr)set_autoscalex_onboolget_autoscalex_onset_autoscaley_onget_autoscaley_onset_autoscalez_onget_autoscalez_onr)rXenablerrrrrr_r_r` autoscales,   zAxes3D.autoscalecCst|t|kr<|jtt|t|g| n |j|| |j|| |durt|j|| | dSrv) rshaper6update_from_data_xy column_stackravelupdate_from_data_xupdate_from_data_yr7r)rXXYZhad_datar_r_r`auto_scale_xyz;szAxes3D.auto_scale_xyzcCs|durL|j}|sZ|jD]0}t|tjr.d}qt|tjtjfrd}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)_tightrrmimage AxesImagemlinesLine2Drrrrr:cleanr6rjrmget_major_locator nonsingularr view_limits set_xboundrrkror set_yboundrr7rer set_zbound)rXrrrrrrx0x1Zxlocatorry0y1Zylocatorz0z1Zzlocatorr_r_r`rIs\               zAxes3D.autoscale_viewcCs4|\}}|\}}|\}}||||||fS)zGet 3D world limits.) get_xlim3d get_ylim3d get_zlim3d)rXr{r|r}r~rrr_r_r`rys   zAxes3D.get_w_limsemit)zminzmaxcCsf|durt|r|\}}|dur6|dur2td|}|durR|durNtd|}|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&r)riterablerre_set_lim)rXbottomtopr&rr'r(r_r_r`set_zlimszAxes3D.set_zlimcCs t|jjSrv)tupler4rjrdr_r_r`get_xlimszAxes3D.get_xlimcCs t|jjSrv)r/r4rkrdr_r_r`get_ylimszAxes3D.get_ylimcCs t|jjS)zGet 3D z limits.)r/r5rjrdr_r_r`get_zlimszAxes3D.get_zlim get_scale_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. get_ticklocs set_ticksget_majorticklabelsget_minorticklabelsget_ticklabels_set_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. cOsdS)z:Currently not implemented for 3D axes, and returns *None*.Nr_)rXr[r\r_r_r`clabelsz Axes3D.clabelcCsld|_|dur|j|_n||_|dur0|j|_n||_|durH|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 rr) vertical_axis) rhr0rr/rr1rr check_getitemdict_vertical_axis)rXrrrr?r_r_r`r8s1   zAxes3D.view_initcCsvtjddg|d|dkrH|dur(d}n|dkr@td|d||_n*|dtjfvrjtd|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)rXr r#r_r_r`r2&s zAxes3D.set_proj_typecCst||jdS)z1Roll arrays to match the different vertical axis.r)rrrB)rXarrr_r_r`_roll_to_verticalBszAxes3D._roll_to_verticalcCs||j}tjg|||Rd|i}d|}tt |j }tt |j }tt |j }t|t|}t|t|}t|} |||| g} ||j| } | |_|| |_|jtj|j|_td} t|dtjkr dnd| |j<|jtjkrHt| || |} 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 )rGrrZworld_transformationr#r$r%rdeg2radr _norm_anglerrrcossinrheyeZvvecrrzerosabspirBrDrEZview_transformationZortho_transformationZpersp_transformationdot)rXr!ZworldMRZelev_radZazim_radZroll_radp0p1p2psrNVZviewMZprojMZ eye_focalM0rBr_r_r`rFsH    "   zAxes3D.get_projrcCs*d|_t||_t||_dS)aa 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. 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 _zoom_btn)rX rotate_btnzoom_btnr_r_r`rDs zAxes3D.mouse_initcCs|jggddS)z2Disable mouse buttons for 3D rotation and zooming.)r_r`N)rDrdr_r_r`disable_mouse_rotationszAxes3D.disable_mouse_rotationcCsdS)z Return whether this Axes supports the zoom box button functionality. Axes3D objects do not use the zoom box button. Fr_rdr_r_r`can_zoomszAxes3D.can_zoomcCsdS)z Return whether this Axes supports the pan/zoom button functionality. Axes3d objects do not use the pan/zoom button. Fr_rdr_r_r`can_panszAxes3D.can_pancCstjtjj|d|jdur.||jur.td|jd||||_|j j |j _ |j j |j _ | \}}|j ||d|d|j j|j _dS)z Share the z-axis with *other*. This is equivalent to passing ``sharex=other`` when constructing the Axes, and cannot be used if the z-axis is already being shared with another Axes. )otherNzz-axis is already sharedrFr))rcheck_isinstancemaxes_base _AxesBaser9rr:r;remajorminorr2r.r_scale)rXrdr!r"r_r_r`rs   z Axes3D.sharezcs>t|jtjkr$tjd|_nd|_|tjddS)Nz axes.zmarginr$z axes3d.grid) r>clearrDrrEmplrcParamsrgridrdr]r_r`rls   z Axes3D.clearcCsT|j|krP|j|_|j|j|_|_t|jj d}|rP| durP|jj j dSNtoolbar) inaxesbuttonrZxdataydatasxsygetattrrErF _nav_stackrq push_currentrXeventrqr_r_r`rJs  zAxes3D._button_presscCs*d|_t|jjd}|r&|jjjdSrp)rZrxrErFrqrzr{r_r_r`rKszAxes3D._button_releasecCs"||||j|j|jfSrv)r0r1r2rrrrdr_r_r` _get_views zAxes3D._get_viewcCs6|\}}}}}}|j|||d||_||_||_dS)Nr)setrrr)rXviewrrrrrrr_r_r` _set_views zAxes3D._set_viewc Cs@z ||WSttfy:|jj}||}|YS0dS)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)rCAttributeErrorrreget_major_formatterformat_data_short)rXrfuncvalr_r_r` format_zdatas   zAxes3D.format_zdatacs"|jdurdS|j|jvrdt|j}t|j}t|j}d|dd|dd|dddd 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. Nz elevation=z.0fu °, azimuth=u °, roll=°-u−cst|d|dfS)Nrr )rZ_line2d_seg_dist)edgexdydr_r`rw sz%Axes3D.format_coord..)rzx=%s, y=%s, z=%s)rBrZr]r rKrrrreplaceminrrhypotrZ inv_transform format_xdata format_ydatar)rXrrZ norm_elevZ norm_azimZ norm_rollrTrUrrr!rr r"d0d1dtrrrxsyszsr_rr` format_coords6            zAxes3D.format_coordcCs|js dS|jdurdS|j|j}}|dur2dS||j||j}}|j}|j}|||_|_|j|jvr"|dkr|dkrdSt |j }|| dt |||dt |} || dt |||dt |} |j| |_|j| |_|d|_|jjn|jdkrZ|dkrF|dkrFdS|\} } } }}}d|||}d|||}t |j}t |j}| | |t |t ||t |}|| | t ||t |t |}||| t |}|| || ||| ||||||||||jjn|j|jvr|\} } } }}}d|||}| | |}|| |}|||}|| || ||| ||||||||||jjdS)z Mouse moving. By default, button-1 rotates and button-3 zooms; these buttons can be modified via `mouse_init`. NrTrr )rZrBrtrurvrwrRrSr]rrJrrLrMrrrrcrErF draw_idleryrrrr^)rXr|rrdxdywhrZdelevZdazimr{r|r}r~rrrrdxxZdyyZdzzdfdzr_r_r`rIsb  ..     .0   zAxes3D._on_movecKs&|dur||j_|jj||fi|S)zI Set zlabel. See doc for `.set_ylabel` for description. N)relabelpadset_label_text)rXZzlabelfontdictrr\r_r_r` set_zlabeliszAxes3D.set_zlabelcCs|j}|S)z. Get the z-label text string. )re get_labelget_text)rXlabelr_r_r` get_zlabelqs zAxes3D.get_zlabelcCs|jS)z)Get whether the 3D axes panels are drawn.)rrdr_r_r` get_frame_onzszAxes3D.get_frame_oncCst||_d|_dS)zs Set whether the 3D axes panels are drawn. Parameters ---------- b : bool TN)rrrc)rXbr_r_r` set_frame_on~s zAxes3D.set_frame_onz3.5rvisiblecKst|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)rZ _draw_gridrc)rXrr\r_r_r`ros z Axes3D.gridc stjgd|d|dvr.tj|fi||dvrt|}|dd|dd|dd|d d|jjfi|dS) 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)rrrrr-Nr,labeltop labelbottom)rrr> tick_paramsrAr=reset_tick_params)rXrr\Zzkwr]r_r`rs    zAxes3D.tick_paramscCs |\}}|j||dddS)z$ Invert the z-axis. Nr)r2r.rXr,r-r_r_r` invert_zaxiss zAxes3D.invert_zaxiscCs|\}}||kS)z9 Returns True if the z-axis is inverted. r2rr_r_r`zaxis_inverteds zAxes3D.zaxis_invertedcCs(|\}}||kr||fS||fSdS)zP Return the lower and upper z-axis bounds, in increasing order. Nrrr_r_r` get_zbounds zAxes3D.get_zboundcCsd|durt|r|\}}|\}}|dur2|}|dur>|}|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)rr)rr*rr.rrr)rXlowerupper old_lower old_upperr_r_r`rs   zAxes3D.set_zboundc s*tj|||fi|}t||||S)z Add text to the plot. kwargs will be passed on to Axes.text, except for the *zdir* keyword, which sets the direction to be used as the z direction. )r>textr Z text_2d_to_3d)rXrrrszdirr\rr]r_r`rsz Axes3D.textrc s|}|r4t|dts4|^}}d|vr@tdn |dd}t|t|}tj ||g|Ri|}|D]} t j | ||dqrt ||||\}}}| |||||S)a0 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 ('x', 'y' or 'z'). **kwargs Other arguments are forwarded to `matplotlib.axes.Axes.plot`. rrz+plot() for multiple values for argument 'z'rr)has_datarstrrr=r broadcast_torr>plotr line_2d_to_3d juggle_axesr) rXrrrr[r\rrlinesliner]r_r`rs  z Axes3D.plot)rvminvmax lightsourcec! s2|} |jdkrtdt|}t|||\}}}|j\} } d|vpPd|v} d|vp`d|v} | rr| rrtd|dd|dd|dd }|dd }t j d r| }n| }|rt t t | |d t t t | |d d |vr|d }n4|d d}|dur"|j}tt|}d}|dd}|d|du}|durdtdg}| d dkr| d dkr|durtjfdd|||fDdd}nttd| d | d g}ttd| d | d g}g}t|dd|d dD]\t|dd|d dD]Z\fdd|||fD}t|j}|||dur8||q8qt|tjrt|r$g}g}t||D]F\}}t|t|jd d}t |r||||q|}|dur$|}t!j"|fi|}|durp|rZ|#||$||}|%||&|n|rt|tjr|dj'dd} ntdd|D} |(| |dus|dur|)|||dur|*|n*|r|#||$||}n|}|%||+||,|||| |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 : 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 arguments are forwarded to `.Poly3DCollection`. r!Argument Z must be 2-dimensional.rstridecstridercountccount.Cannot specify both stride and count argumentsr>2_internal.classic_moder facecolorscolorNcmapshadezshade cannot be None.rcsg|]}t|qSr_)r_array_patch_perimetersra)rrr_r` sz'Axes3D.plot_surface..rIrc s.g|]&}t|ddfqS)r )r_array_perimeterr)cscs_nextrsrs_nextr_r`rs).rcSs g|]}|dddfqSNr)r)rrWr_r_r`rrx)-rndimrr_to_unmasked_float_arrayrbroadcast_arraysrr=rmrnintrceil _get_linesget_next_colorrmcolorsto_rgbagetstacklistrangezipTappendrndarrayisnanany itertools zip_longestrr Poly3DCollection _shade_colors_generate_normalsset_facecolorsset_edgecolorsr set_arrayset_climset_normadd_collectionr)!rXr r rrrrrr\rrowscols has_stride has_countrrZcompute_stridesZfcolorsrrrcolsetpolysZrow_indscol_indsrWZ new_polysZ new_colsetpcolZnew_polypolycavg_zr_)rrrrrrr` plot_surfacesF                ""                 zAxes3D.plot_surfacec Cs:t|tjr||jd}d|dd|d}}}|d|ddf|d|ddf}|d|ddf|d|ddf}ntt|df}tt|df}t|D]\}} t| }d|dd|d}}}| |ddf| |ddf||ddf<| |ddf| |ddf||ddf<qt||S)a Compute the normals of a list of polygons. Normals point towards the viewer for a face with its vertices in counterclockwise order, following the right hand rule. Uses three points equally spaced around the polygon. This normal of course might not make sense for polygons with more than three points not lying in a plane, but it's a plausible and fast approximation. Parameters ---------- polygons : list of (M_i, 3) array-like, or (..., M, 3) array-like A sequence of polygons to compute normals for, which can have varying numbers of vertices. If the polygons all have the same number of vertices and array is passed, then the operation will be vectorized. Returns ------- normals : (..., 3) array A normal vector estimated for the polygon. rrr.N)rrrremptyrrcross) rXpolygonsni1i2Zi3v1v2Zpoly_irWr_r_r`rs  $&,.zAxes3D._generate_normalsc s|durtjddd}tjdd,|tjj|ddd |j}Wdn1sT0Yt|}|rt d dt d dj fd d }d||<t |}|dddf}||ddtj f|}||dddf<nt |}|S)z Shade *color* using normal vectors given by *normals*. *color* can also be an array of the same length as *normals*. Ng -x3@)azdegaltdegignore)invalidr T)rkeepdimsrI333333?cs |Srvr_)rZin_normZout_normr_r`r#sz"Axes3D._shade_colors..normrr)r LightSourcererrstaterr directionrr Normalizeinverse to_rgba_arraynewaxis asanyarraycopy) rXrnormalsrrmaskralphacolorsr_rr`rs$"    zAxes3D._shade_colorsc s|}jdkrtdt\j\}}d|vpFd|v}d|vpVd|v} |rh| rhtd|dd} |dd} |dd } |dd } tjd r| r| rt t t || dnd } | rt t t || dnd } nJ|s2| r t t t || dnd } | r.t 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|fi|}|||||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 arguments are forwarded to `.Line3DCollection`. rrrrrrrr rrrrIz*Either rstride or cstride must be non zerocsg|] }|qSr_r_rr)r r_r`rrxz)Axes3D.plot_wireframe..csg|] }|qSr_r_r+)r r_r`rrxcsg|] }|qSr_r_r+)rr_r`rrxcsg|] }|qSr_r_r+)tXr_r`rrxcsg|] }|qSr_r_r+)tYr_r`rrxcsg|] }|qSr_r_r+)tZr_r`rrxcSs"g|]\}}}tt|||qSr_rrrxlylzlr_r_r`rscSs"g|]\}}}tt|||qSr_r/r0r_r_r`rs)rrrrrrr=rmrnrrr transposerrsizerr Line3DCollectionrr)rXr r rr\rrrrrrrrrZriiZciixlinesZylinesZzlinesZtxlinesZtylinesZtzlinesrlinecr_)r r rr,r-r.r`plot_wireframe4sj&        """"    zAxes3D.plot_wireframe)rrrrrcOs|}|dur|j}tt|}|dd} |d| du} t j |i|\} }}z|d} Wnt y|^} }Yn0t | } | } | j| }| j| }| | }tj|||fdd}tj|g|Ri|}| rD|dddddfjdd}|||dus"|dur.||||durr||n.| rd||}||||}n|}|||||| j| j| ||S) aT 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 a 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 : Normalize An instance of Normalize to map values to colors. vmin, vmax : float, default: None 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` The lightsource to use when *shade* is True. **kwargs All other arguments are passed on to :class:`~mpl_toolkits.mplot3d.art3d.Poly3DCollection` Examples -------- .. plot:: gallery/mplot3d/trisurf3d.py .. plot:: gallery/mplot3d/trisurf3d_2.py NrrrrIrrr )rrrrrrrrr=r get_from_args_and_kwargsKeyErrorrget_masked_trianglesrrrr rrrrrrrrrr)rXrrrrrr[r\rrrtrir trianglesxtytZztvertsr r r'rr_r_r` plot_trisurfsD5             zAxes3D.plot_trisurfrc Cs|j}|j}|d|dd}t||D]V\}}|}|sBq*t|||} t|||} |d} g} g} tt| d|}|dkrt| ddkr*d}nq*t| dd|d}t t t|dD]Z}t t||}t t|d|}| | d|| d|| d|| d|gqt | } || } || | }|| | }tj| ||d}||||q*|D]}|qdS)z4 Extend a contour in 3D by creating r rrr edgecolorsN)levels collectionsr get_pathsr Z_paths_to_3d_segments get_edgecolorroundrrrrrrrrr set_sort_zposadd_collection3dremove)rXcsetstriderEcollsrrr8pathsZtopvertsZbotvertsrZ polyvertsr'Znstepsstepsizerrrr*Zcolors2Zpolycolr r_r_r`_3d_extend_contoursL          zAxes3D._3d_extend_contourcCsRd|}|r|||n4t|j|jD]$\}}|dur<|}tj|||dq(dS)Nrr)rRrrErFr line_collection_2d_to_3d)rXrMextend3drNroffsetrr8r_r_r`add_contour_setBszAxes3D.add_contour_setcCs|j|||ddS)NrrU)_add_contourf_set)rXrMrrUr_r_r`add_contourf_setMszAxes3D.add_contourf_setc Csd|}|jddt|jd}|jr\|jdt|jddd}t|d|}|jr|jdt|jddd}t||}t||jD].\}}|dur|}t j |||d| |q|S)z Returns ------- levels : numpy.ndarray Levels at which the filled contours are added. rNrIrrr r) rErdiff _extend_mininsert _extend_maxrrrFr poly_collection_2d_to_3drJ) rXrMrrUZ midpointsZ min_level max_levelrr8r_r_r`rXPs""  zAxes3D._add_contourf_set)rTrNrrUcsb|} t||||\} } } tj| | | g|Ri| }||||||||||| |S)a9 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 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. data : indexable object, optional DATA_PARAMETER_PLACEHOLDER *args, **kwargs Other arguments are forwarded to `matplotlib.axes.Axes.contour`. Returns ------- matplotlib.contour.QuadContourSet )rr rotate_axesr>contourrVr)rXr r rrTrNrrUr[r\rjXjYjZrMr]r_r`rais 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 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. data : indexable object, optional DATA_PARAMETER_PLACEHOLDER *args, **kwargs Other arguments are forwarded to `matplotlib.axes.Axes.tricontour`. Returns ------- matplotlib.tri.tricontour.TriContourSet r)rr r:rrr=r r`r>r(r> tricontourrVr)rXrTrNrrUr[r\rr=r r rrbrcrdrMr]r_r`res"   zAxes3D.tricontourcs>d|d|d|||ifdddD}|jg||RdS)Nrrrcs(g|] }t|t|fqSr_rnanminnanmax)rdimZdim_valsr_r`rsz/Axes3D._auto_scale_contourf..r)r)rXr r rrrErlimitsr_rjr`_auto_scale_contourfs  zAxes3D._auto_scale_contourfrWcsb|}t||||\} } } tj| | | g|Ri|} || ||} |||||| || 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 )rr r`r>contourfrXrl)rXr r rrrUr[r\rrbrcrdrMrEr]r_r`rms zAxes3D.contourfcs|}tj|i|\}}}|j}|j}d|vr>|d} n|^} }t||| |\} } } t| | |j|j }t j || g|Ri|} | | ||}| ||| |||| S)aV 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)rr r:rrr=r r`r>r(r> tricontourfrXrl)rXrrUr[r\rr=r r rrbrcrdrMrEr]r_r`rns"  zAxes3D.tricontourfcst|}|jrt|nd}t|tjurHtj|||d| |nRt|tj urrtj |||d| |n(t|tj urtj |||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)rr[r5rtyperPolyCollectionr r^rJLineCollectionrSPatchCollectionpatch_collection_2d_to_3dr>r)rXr rrZzvalsZzsortval collectionr]r_r`rK#s     zAxes3D.add_collection3d) rrrrrDc facecolorrr) replace_namesc s|} |} tjdd|||fD\}}}tj|}t|||||\}}}}}t| |rh|}t j ||g|R||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 arguments are passed on to `~.axes.Axes.scatter`. Returns ------- paths : `~matplotlib.collections.PathCollection` cSs"g|]}ttj|tjqSr_)rr mafillednan)rtr_r_r`rzrxz"Axes3D.scatter..)rru)rr depthshade皙?r)rrrryr rdelete_masked_pointsmay_share_memoryr&r>scatterr rsrr5rr) rXrrrrrrur}r[r\rZzs_origpatchesr]r_r`rDs"2   "  zAxes3D.scattercs|}tj||g|Ri|}t|t|}g} g} t||D]T\} } t| } | | 7} | | gt| 7} t | | |d|vrF| |dqFt| dkrt| \}}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 arguments are forwarded to `matplotlib.axes.Axes.bar`. Returns ------- mpl_toolkits.mplot3d.art3d.Patch3DCollection r)r)rr>barrrrrr Z_get_patch_vertsr\Zpatch_2d_to_3d set_alpharr)rXleftrrrr[r\rrrAZverts_zsrrvsrrr]r_r`rs$    z Axes3D.baraveragec Os|} tt||||||\}}}}}}t|}t||}t|}t||}t|}t||}tgd}t|j|j}d||fd||fd||ffD]J\}}}|dtj tj f}|dtj tj f}|||d|f|d|f<q| d|jdd}g}|dur0|j g}t t|}t|t|krn|D]}||gdqTn&|}t|t|kr|dt|9}| r||}|||| }n|}tj|g| R||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. ))rrrrr rr r rr rr)rrr r rr r r r rr r )rrrr)rrrr)rrrr)rrrrrr r.)rINr)zsortrv)rrrr[rrrrrr$reshape_get_patches_for_fillrrrr#rextendrrr rrr)rXrrrrrrrrrrr[r\rr{r|r}r~rrZcuboidrrrdprrur'Z sfacecolorsr r_r_r`bar3dsT?   .&    z Axes3D.bar3dcenterc s<tj|f||d|}|j\}}|jd||S)N)rlocgq= ףp?)r> set_titletitlerset_y)rXrrrr\retrrr]r_r`rl szAxes3D.set_titlertail)lengtharrow_length_ratiopivot normalizecsddd}|}d} t|| kr6td| t|f|d| } dd| D} tjg| | R} | d| } | | d} | rttj| fd d| D} nd d| D} td d | Drt j gg|| dRi|} | | | Stj d |gt d}||}tjgd|d|dkr0||8}n|dkrF||d8}t| dd}t| d| t }tjj|dd}|dk|}|r||d}n|}t|dkr6|tj||dd}||}|ddddftj||}|t|ddf}|dd}g||}ng}t j |g|| dRi|} | | ||dddf|dddf|dddf|| S)a ax.quiver(X, Y, Z, U, V, W, /, length=1, arrow_length_ratio=.3, pivot='tail', normalize=False, **kwargs) Plot a 3D field of arrows. The arguments could be array-like or scalars, so long as they 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:`~matplotlib.collections.LineCollection` c Ssd|dddf}|dddf}tjj|ddddfdd}tj|||dkt|d}tj| ||dkt|d}t|}t|}t |} t ||dd|||d||| g||d|||dd|| | g| | || t ||gg} | } | gdgdfd9<t d | |} t d | |} tj| | gdd}|S) Nrr rr)whereout)rr rr)rrrr rIzij...,...j->...i)rrrdivide zeros_like ones_likemathradiansrLrMr full_liker&einsumr)UVWanglerrrZx_py_prarurZRposZRnegZ Rpos_vecsZ Rneg_vecs head_dirsr_r_r` calc_arrows s(    (*z"Axes3D.quiver..calc_arrowsrz-Wrong number of arguments. Expected %d got %dNcSs g|]}t|tjjr|jqSr_)rrry MaskedArrayr(rkr_r_r`r sz!Axes3D.quiver..cs g|]}tjj|dqS)r()rryr compressedrrr_r`r scSsg|]}t|qSr_)rr rr_r_r`r rxcss|]}t|dkVqdS)rN)r)rvr_r_r`r rxz Axes3D.quiver..r$r)rmiddletip)rrrrrr rr)rIr rI)r)rrrrr functoolsreduce logical_orrr r6rrrrrrastyperrrmultiplyouterswapaxesr)rXrrrrr[r\rrZargi input_argsmasksbcastr8Zshaft_dtZarrow_dtZXYZrrZshaftsrheadsrr_rr`quivers sb.             "   4z Axes3D.quiver)rrDrrc2 st|dkrdd}ndd}||i|\}}jdkrDtdtjjtjd} t| d|dur|t\} } } nfd d |D\} } } fd d } |dur|j }| |d }| |d}| | | | tjgdgdgdgdgtjd}t t }dd}|dD]}|j| \}}}t|}t|}t|}||j}|ddd}|D]4}|D]&}|||dg}t|}|r||||t|dd|ddD]\}}||||g} ||||g}!t| }"t|!}#|"r&|#s&||"|!|n&|"sƈ|#r||#|!|q||||dg}$||||g}%t|$}&|&rr||&|%|qrqhqi}'|D]\}(})|dur|)}*ng}*|)D]}+|+dddf|+dddf|+dddff},t|+j}-| |,|-dddf<| |,|-dddf<| |,|-dddf<|*|-q||(}.||(}/|r||*}0||.|0|}.|/dur||/|0|}/tj|*f|.|/d|}1||1|1|'|(<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 ndarray of color names, with each item the color for the corresponding voxel. The size must match the voxels. - A 4D ndarray of rgb/rgba data, with the components along the last axis. 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 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||fSrvr_)Z _Axes3D__xZ _Axes3D__yZ _Axes3D__zrzr\r_r_r`voxelsV szAxes3D.voxels..voxelsc[s d||fSrvr_)rzr\r_r_r`rY sz%Argument filled must be 3-dimensionalrr Nc3s|]}t|VqdSrv)rr)rru) coord_shaper_r`rh rxz Axes3D.voxels..cspt|dvr&t|jt|St|dvr^t|ddjkrZtd||Std|dS)N)rr )rrrz8When multidimensional, {} must match the shape of filledzInvalid {} argument)rrrrrformat)rname)rzr_r`_broadcast_color_argj sz+Axes3D.voxels.._broadcast_color_argrrDrrrrcss8tj|tjd}t|D]}|Vtj|ddd}qdS)z%Generate cyclic permutation matrices.rr rrN)rrNintprr)rmatrr_r_r`permutation_matrices s z+Axes3D.voxels..permutation_matricesrIrrrC)rrrrrrrr/indicesrrrrrrrRarangerritemsrrrr rrK)2rXrrDrrr[r\rZxyzr5rrrrsquareZ voxel_facesrpermutepcZqcrcZpindsZqindsZrindsZsquare_rot_posZsquare_rot_negrqrTi0r1r2rUrVrrpkZpk2ikrcoordZ faces_indsZfacesZ face_indsindfacerv edgecolorr'polyr_)rrzr`r sH                "  .     z Axes3D.voxels)rrrxerryerrzerrrcC sX|}t|tj}dd|D}|dd|jd|fd|fd|fg|dd t |rd|n|g}t |rx|n|g}t |r|n|g}t |t |krt |ksnt d | || | d d }d |d <|jj|dkr||fn|||f|dd\\}}tj||d|r2||ddn||dd|dkr^||nd }| dd|vr~d|d<| d ur|d} dD]}| |d qi|d| i}| r| |d<nd|vr|d|d<dD]}||vr||||<qi|ddi}| d ur$tjd} | dkr:d| |d<| d urL| |d<| |d<d d!}fd"d#ggg}}}g} d$d$d%d&}!dd'dd(|dtjd)d}"|"|jjd*9}"|jd+|"|"fg}"ttj|"dd,}"tj|dddd- tj !|"}#Wd n1s 0Yt#|#t$|"dddgd'}"|"d.9}"i||"d'd/}$|$ dd t%gd(|||g|||g|||g|||gD]\}%}&t&|%|%}'d urʐqt sgt |&t't(t |&)t*t(t |&)t*fd0d1t+|||gD}(|(\\})}*\}+},\}-}.B}/|/,r| dkr||)|+|-g|/@}0||*|,|.g|/@}1tj-|0d|!|'d2|}2tj-|1d|!|'d2|}3||2||3|.|2|.|3,r6||*|,|.g@\}4}5}6|j/|4|5|6gRi|$,rx||)|+|-g@\}7}8}9|j/|7|8|9g Ri|$tj0t$|(j1fi|}:|2|:|.|:| .|(qt$| } fd3d4};|;| d\}<}=|;| d\}>}?|;| d\}@}A|3|<|=f|>|?f|@|Af|t4j5|t6|t6|f|d up4|d u|d u|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|]\}}|dur||qSrvr_)rrrr_r_r` S rxz#Axes3D.errorbar..rrrrrF)convertz)'x', 'y', and 'z' must have the same sizerN _nolegend_rT) return_kwargs)rg?nonerC0) marker markersizemarkerfacecolormarkeredgewidthmarkeredgecolor markevery linestyle fillstyle drawstyle dash_capstyledash_joinstylesolid_capstylesolid_joinstyle linewidth)rQr)r rasterizedrNonezerrorbar.capsizerrrrcsfdd|DS)Ncsg|]}gt|qSr_)rcompress)rrrr_r`r rxz8Axes3D.errorbar.._apply_mask..r_)arraysr(r_rr` _apply_mask sz$Axes3D.errorbar.._apply_maskcsZt|jdkr|\}}n ||}}t|B|||}t|B|||}||fSr)rrrr)errdataZlomaskZhimaskZlow_errZhigh_errZlowsZhighs) everymaskr_r` _extract_errs s   z&Axes3D.errorbar.._extract_errs|_rr rzlines.markersizeHr,r)rrrgV&,t=?)rrcs&g|]\}}||qSr_r_)rrr)r dir_vectorrlolimsuplimsr_r`r sz#Axes3D.errorbar..)lsrc sLt|dd|ddddft|dd|ddddffSrvrf)Zerr_arrZ coord_label)i_xyzr_r`_digout_minmax s$$z'Axes3D.errorbar.._digout_minmax)has_xerrhas_yerrr)8rrnormalize_kwargsrrr setdefault_process_unit_inforr*rr_errorevery_to_maskr=r _plot_argsr r set_zorderradd_linermrnrrEdpi transAxesrPrQrrZ _setattr_cmrinvrrRrrZget_dir_vectorr[rrrrrZLine3Drrr6rrr mcontainerErrorbarContainerr/ containers)CrXrrrrrrfmt barsabove erroreveryecolor elinewidthcapsizecapthickxlolimsxuplimsZylolimsZyuplimsZzlolimsZzuplimsr\rr data_line base_stylereb_lines_style eb_cap_stylerZerrlinescaplinesZlimmarksZ coorderrsZ capmarkerZ quiversizeZinvMZeb_quiver_stylerrZi_zdirZcoorderrr1xhr2Zyhr3zhnolimsZ lo_caps_xyzZ hi_caps_xyzcap_locap_hiZxh0Zyh0Zzh0Zxl0Zyl0Zzl0Zerrlinerr{r|r}r~rrerrorbar_containerr_)rrrrrrrr`errorbar s k "                    0                  zAxes3D.errorbar)for_layout_onlyc s^tj||||d}|g}|jrR|jD]&}|r*t||}|r*||q*t j |S)N)call_axes_locatorbbox_extra_artistsr-) r> get_tightbboxrbrrrmartist_get_tightbbox_for_layout_onlyrrr union) rXrr.r/r-rbatchrZaxis_bbr]r_r`r0( s zAxes3D.get_tightbboxzC0-ZC0ozC3-)linefmt markerfmtbasefmtr,r orientationcsddlm} |} tjgd| dt|t|f} t|t|f} t|t|f}| dkr|| }}||}}fddt|||D}nf| dkr|| }}||}}fd dt|||D}n.|| }}|| }}fd dt|||D}t |\}}}|d ur"t 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, 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) StemContainerr)r8rcs&g|]\}}}||f|||fgqSr_r_rthisxthisyZthiszr,r_r`r szAxes3D.stem..rcs&g|]\}}}||f|||fgqSr_r_r:r=r_r`r scs&g|]\}}}||f|||fgqSr_r_r:r=r_r`r sNzlines.linestyler)rrr) linestylesr*r)r)matplotlib.containerr9rrrrrrrr rmrnrr r6r add_containerrr)rXrrrr5r6r7r,rr8r9rrrrbasexZbasexlimbaseyZbaseylimrr linemarker linecolorbaseline stemlines markerlinestem_containerZjxjyZjzr_r=r`stem8 sVB                    (z Axes3D.stem)N)N)NN)NN)NNF)N)TrN)NN)NTTT)NNTF)NNNr)N)r r)NN)T)r)NN)N)N)r)FrrN)rN)rN)rr)rrrxNT)rr)NrTN)Nr)NNNrFr NNNNFFFFFF)NTN)__name__ __module__ __qualname____doc__r _axis_namesrGrouperrr:rdeprecate_privatize_attributedistr?r@rArgrLrprrrZget_zgridlinesZget_zticklines deprecatedpropertyZw_xaxisZw_yaxisZw_zaxisrrrrrrr1allow_rasterizationrrrrrrrrrrrymake_keyword_onlyr.rfset_xlimrset_ylimrrr0r1r2Z get_zscale set_xscale set_yscaleZ set_zscalemaprZ get_zticksZ set_zticksZget_zmajorticklabelsZget_zminorticklabelsZget_zticklabelsZset_zticklabelsZ zaxis_datetextwrapdedentr=r8r2rGrrDrarbrcrrlrJrKr}rrrrIrrrrrename_parameterrorrrrrrZtext3DZtext2DrZplot3Dr rrr9rBrRrVrYrXrraZ contour3DrerlrmZ contourf3DrnrKrZ scatter3DrrrrZquiver3Drr,r0rJZstem3D __classcell__r_r_r]r`r*sf          Q' 7  $ #  >             F E   (L      ) J+ $v e 4  %4 0! G5 & Y  Etrr~cCstdd|}}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?rHr>i)rrmeshgridexprQ)rrrr r ZZ1ZZ2rr_r_r` get_test_data s&*rb)r~)8rNrFrrrrr\numpyr matplotlibrmrrrrmatplotlib.artistrr1matplotlib.axesaxesrfmatplotlib.collectionsrmatplotlib.colorsr*rmatplotlib.imageimagermatplotlib.linesrrmatplotlib.patchesrrr? containerrmatplotlib.transforms transformsrrmatplotlib.axes._baserr r Zmatplotlib.tri.triangulationr rr rrinterpddefine_aliasesrrbr_r_r_r`sn