B
&[_~H @ s d dl mZ d dlmZ ddlmZ d dlmZ d dl Z ddl
m Z m Z m
Z
d dlZddlmZmZ d dlZyd dlZdZW n ek
r d ZY nX yd d
lmZ dZW n ek
r d ZY nX g ZdddZ dddZ dd Z dddi dddddddddddddd d
fddZ dS ) N)ColorscaleValidator )apply_default_cascade)BytesIO)rescale_intensity_integer_ranges_integer_types)Writer
from_arrayTF)Imagepil pngc C s |dk s|dkrt dd}| jdkr,d}nH| jdkrJ| jd dkrJd }n*| jdkrl| jd d
krld }d }nt d
|dkrtrdnd}|dkr|dkrt d|dkrD| j}| j}|dkr| |d |d |d f} t|d |d |dk||d}t| |d} d}
t ,} || | j |
t
|
d } W dQ R X nxtsRtdt| }
|dksp|dkrzd }
d}nd}
d}t .} |
j| ||d |
t
|
d } W dQ R X | S ) a Converts a numpy array of uint8 into a base64 png string.
Parameters
----------
img: ndarray of uint8
array image
backend: str
'auto', 'pil' or 'pypng'. If 'auto', Pillow is used if installed,
otherwise pypng.
compression: int, between 0 and 9
compression level to be passed to the backend
ext: str, 'png' or 'jpg'
compression format used to generate b64 string
r z*compression level must be between 0 and 9.F L ZRGBr
ZRGBATzInvalid image shapeautor Zpypngr z6jpg binary strings are only available with PIL backendr )Z greyscalealpha compression)modezdata:image/png;base64,zutf-8Nzapillow needs to be installed to use `backend='pil'. Pleaseinstall pillow or use `backend='pypng'.ZjpgZjpegzdata:image/jpeg;base64,)formatZcompress_level)
ValueErrorndimshape pil_importedZreshaper r
r writeZrowsbase64 b64encodegetvaluedecode ImportErrorr Z fromarraysave)imgbackendr extr r r shwZimg_pngprefixstreamZ
base64_stringZpil_img r+ :/tmp/pip-install-l29rncou/plotly/plotly/express/_imshow.py_array_to_b64str sP
$
"r- maxc C s |dkr dnd}| d kr | S t | r6| gd |g S t| dkrTt| d |g S t| dkrnt| |g S t| dkr~| S tdt| d S )Nr. r r r r
z\zmax can be a scalar, or an iterable of length 1, 3 or 4. A value of %s was passed for zmax.)npZisscalarlenlistr str)zr r r+ r+ r, _vectorize_zvalue\ s
r5 c C sj | j j}d}|tkr t| d S | t| }|d| krBdS |d| krRdS |d| krbdS dS d S )Ng?r r/ i l )dtypetyper r r0 isfiniter. )r$ dtZrtolZim_maxr+ r+ r, _infer_zmax_from_typeo s r: r c & s> t }t| | }tr&t tjr&|r6td jd jd }}x<||gD ]0}t
j | j t j
rV j | t j |< qVW |dkr j | }|dkr j | }|dkrd}|dddkr||d< |dddkr||d< |dddkr:tjj |d< |d d d
|d< nt d rt jd r|dkrP j}|dddkrt jd
r jjp|d|d< t drt jd r|dkr j}|dddkrt jd
rވ jjpd|d< |dddkrd|d< |dddkrd|d< |dddkr,d|d< |dkr:d}t tjr\|rVtdd}nd}t |dkr j dko| } j t j krd t j | dk r| d | d |dkr܈ j dkrdnd}|dkr,dk s|rdkr dk s|rddkrd n8dkrL j t j krLt! dkrddk rdd j dkrh|sh|dk r j"d t#|krtd|dk r j"d t#|krtdt$j%|| dd }|d krdnd }t&t&|d d }|dkr t&dd!d"|d#< d!|d$ d%< t'd&d'} t&| (|d( |d)|d< |d rt&|d d*|d d+< nL j dkr j"d, d-ks j dkr|rd} dk rȈdk rt)d.d/t)d0d/ |rZdkrdkr } d} nP j dkrt* d d ft j d1} n&t + fd2d3t, j"d, D } t-| |||d4} t$j.| d5}n* j"d, dkrnd6nd7} t$j. | d8}i }|d krt&dd |d$< ntd9t j" t& }!x$d:D ] }"||" r||" |!|"< qW |d; r|d; |!d<< n |d= j/j0j1dkrd>d?i|!d@< t$j2||dA}#|#3|! |rn| rnt 4 | ksndB|d pZd|d pfdf }$nz|dC dDkrdE}%n8 j dkrdF}%n& j dkr j"d, dkrdG}%ndE}%dH|d pd|d pd|d pd|%f }$|#j5|$dI |d r|#j6|d d* |d r(|#j7|d d* |#j3|d= ddJ |#S )Ka
Display an image, i.e. data on a 2D regular raster.
Parameters
----------
img: array-like image, or xarray
The image data. Supported array shapes are
- (M, N): an image with scalar data. The data is visualized
using a colormap.
- (M, N, 3): an image with RGB values.
- (M, N, 4): an image with RGBA values, i.e. including transparency.
zmin, zmax : scalar or iterable, optional
zmin and zmax define the scalar range that the colormap covers. By default,
zmin and zmax correspond to the min and max values of the datatype for integer
datatypes (ie [0-255] for uint8 images, [0, 65535] for uint16 images, etc.). For
a multichannel image of floats, the max of the image is computed and zmax is the
smallest power of 256 (1, 255, 65535) greater than this max value,
with a 5% tolerance. For a single-channel image, the max of the image is used.
Overridden by range_color.
origin : str, 'upper' or 'lower' (default 'upper')
position of the [0, 0] pixel of the image array, in the upper left or lower left
corner. The convention 'upper' is typically used for matrices and images.
labels : dict with str keys and str values (default `{}`)
Sets names used in the figure for axis titles (keys ``x`` and ``y``),
colorbar title and hoverlabel (key ``color``). The values should correspond
to the desired label to be displayed. If ``img`` is an xarray, dimension
names are used for axis titles, and long name for the colorbar title
(unless overridden in ``labels``). Possible keys are: x, y, and color.
x, y: list-like, optional
x and y are used to label the axes of single-channel heatmap visualizations and
their lengths must match the lengths of the second and first dimensions of the
img argument. They are auto-populated if the input is an xarray.
color_continuous_scale : str or list of str
colormap used to map scalar data to colors (for a 2D image). This parameter is
not used for RGB or RGBA images. If a string is provided, it should be the name
of a known color scale, and if a list is provided, it should be a list of CSS-
compatible colors.
color_continuous_midpoint : number
If set, computes the bounds of the continuous color scale to have the desired
midpoint. Overridden by range_color or zmin and zmax.
range_color : list of two numbers
If provided, overrides auto-scaling on the continuous color scale, including
overriding `color_continuous_midpoint`. Also overrides zmin and zmax. Used only
for single-channel images.
title : str
The figure title.
template : str or dict or plotly.graph_objects.layout.Template instance
The figure template name or definition.
width : number
The figure width in pixels.
height: number
The figure height in pixels.
aspect: 'equal', 'auto', or None
- 'equal': Ensures an aspect ratio of 1 or pixels (square pixels)
- 'auto': The axes is kept fixed and the aspect ratio of pixels is
adjusted so that the data fit in the axes. In general, this will
result in non-square pixels.
- if None, 'equal' is used for numpy arrays and 'auto' for xarrays
(which have typically heterogeneous coordinates)
contrast_rescaling: 'minmax', 'infer', or None
how to determine data values corresponding to the bounds of the color
range, when zmin or zmax are not passed. If `minmax`, the min and max
values of the image are used. If `infer`, a heuristic based on the image
data type is used.
binary_string: bool, default None
if True, the image data are first rescaled and encoded as uint8 and
then passed to plotly.js as a b64 PNG string. If False, data are passed
unchanged as a numerical array. Setting to True may lead to performance
gains, at the cost of a loss of precision depending on the original data
type. If None, use_binary_string is set to True for multichannel (eg) RGB
arrays, and to False for single-channel (2D) arrays. 2D arrays are
represented as grayscale and with no colorbar if use_binary_string is
True.
binary_backend: str, 'auto' (default), 'pil' or 'pypng'
Third-party package for the transformation of numpy arrays to
png b64 strings. If 'auto', Pillow is used if installed, otherwise
pypng.
binary_compression_level: int, between 0 and 9 (default 4)
png compression level to be passed to the backend when transforming an
array to a png b64 string. Increasing `binary_compression` decreases the
size of the png string, but the compression step takes more time. For most
images it is not worth using levels greater than 5, but it's possible to
test `len(fig.data[0].source)` and to time the execution of `imshow` to
tune the level of compression. 0 means no compression (not recommended).
binary_format: str, 'png' (default) or 'jpg'
compression format used to generate b64 string. 'png' is recommended
since it uses lossless compression, but 'jpg' (lossy) compression can
result if smaller binary strings for natural images.
Returns
-------
fig : graph_objects.Figure containing the displayed image
See also
--------
plotly.graph_objects.Image : image trace
plotly.graph_objects.Heatmap : heatmap trace
Notes
-----
In order to update and customize the returned figure, use
`go.Figure.update_traces` or `go.Figure.update_layout`.
If an xarray is passed, dimensions names and coordinates are used for
axes labels and ticks.
z{It is not possible to use binary image strings for xarrays.Please pass your data as a numpy array instead using`img.values`r r Nr xycolor
z
columns__len__name indexequalz0Binary strings cannot be used with pandas arraysTFr r/ r ZminmaxZinferzZThe length of the y vector must match the length of the first dimension of the img matrix.z[The length of the x vector must match the length of the second dimension of the img matrix.Z
coloraxis1)r; r< r4 Z coloraxislowerreversed) autorange)yaxisdomain)Z scaleanchor constrainZxaxisrH rJ
colorscaleimshowcolor_continuous_scale)rK ZcmidZcminZcmax)
title_textZcolorbarr )r r
min)r r. )in_range out_rangec s0 g | ](}t d |f | | ftjdqS ).)rP rQ )r r0 uint8).0ch)r$ zmaxzminr+ r,
s zimshow..)r% r r& )sourceZrgbZrgba256)r4 rV rU
colormodelz_px.imshow only accepts 2D single-channel, RGB or RGBA images. An image of shape %s was provided)heightwidthtitlerN templatet< margin)datalayoutz%%s: %%{x}
%s: %%{y}r7 Zheatmapz%{z}z%{z[0]}z[%{z[0]}, %{z[1]}, %{z[2]}]z/%s: %%{x}
%s: %%{y}
%s: %s)
hovertemplate)r] overwrite)8localsr copyxarray_imported
isinstancexarrayZ DataArrayr Zdimsr0 Z
issubdtypeZcoordsr6 Z
datetime64Zastyper3 getZplotutilsZlabel_from_attrsreplacehasattrr? rA rC pdZ DataFrameZ
asanyarrayr boolrR r. rO r: r r1 goZHeatmapdictr Zvalidate_coercer5 r Zdstackranger- r rb r` r^ ZFigureZ
update_layoutallZ
update_tracesZ update_xaxesZ update_yaxes)&r$ rV rU originlabelsr; r< rM Zcolor_continuous_midpointZ range_colorr\ r] r[ rZ ZaspectZcontrast_rescalingZ
binary_stringZbinary_backendZbinary_compression_levelZ
binary_formatargsZy_labelZx_labelZaxZ is_dataframetracerG rb Zcolorscale_validatorZ
rescale_imageZ img_rescaledZimg_strrY Z layout_patch attr_nameZfigrc Z
hover_namer+ )r$ rU rV r, rL s.
.
rL )r r
r )r. ) Zplotly.graph_objsZ
graph_objsrp Z _plotly_utils.basevalidatorsr Z_corer ior r Z imshow_utilsr r r Zpandasrn r r r
Znumpyr0 ri rg r" ZPILr r Z _float_typesr- r5 r: rL r+ r+ r+ r, sR
A