from __future__ import absolute_import, division, unicode_literals
import param
import numpy as np
from ...core.ndmapping import sorted_context
from .chart import AreaPlot, ChartPlot
from .path import PolygonPlot
from .plot import AdjoinedPlot
[docs]class DistributionPlot(AreaPlot):
"""
DistributionPlot visualizes a distribution of values as a KDE.
"""
bandwidth = param.Number(default=None, doc="""
The bandwidth of the kernel for the density estimate.""")
cut = param.Number(default=3, doc="""
Draw the estimate to cut * bw from the extreme data points.""")
filled = param.Boolean(default=True, doc="""
Whether the bivariate contours should be filled.""")
[docs]class BivariatePlot(PolygonPlot):
"""
Bivariate plot visualizes two-dimensional kernel density
estimates. Additionally, by enabling the joint option, the
marginals distributions can be plotted alongside each axis (does
not animate or compose).
"""
bandwidth = param.Number(default=None, doc="""
The bandwidth of the kernel for the density estimate.""")
cut = param.Number(default=3, doc="""
Draw the estimate to cut * bw from the extreme data points.""")
filled = param.Boolean(default=False, doc="""
Whether the bivariate contours should be filled.""")
levels = param.ClassSelector(default=10, class_=(list, int), doc="""
A list of scalar values used to specify the contour levels.""")
[docs]class BoxPlot(ChartPlot):
"""
BoxPlot plots the ErrorBar Element type and supporting
both horizontal and vertical error bars via the 'horizontal'
plot option.
"""
style_opts = ['notch', 'sym', 'whis', 'bootstrap',
'conf_intervals', 'widths', 'showmeans',
'show_caps', 'showfliers', 'boxprops',
'whiskerprops', 'capprops', 'flierprops',
'medianprops', 'meanprops', 'meanline']
_nonvectorized_styles = style_opts
_plot_methods = dict(single='boxplot')
[docs] def get_extents(self, element, ranges, range_type='combined'):
return super(BoxPlot, self).get_extents(
element, ranges, range_type, 'categorical', element.vdims[0]
)
def get_data(self, element, ranges, style):
if element.kdims:
with sorted_context(False):
groups = element.groupby(element.kdims).data.items()
else:
groups = [(element.label, element)]
data, labels = [], []
for key, group in groups:
if element.kdims:
label = ','.join([d.pprint_value(v) for d, v in zip(element.kdims, key)])
else:
label = key
data.append(group[group.vdims[0]])
labels.append(label)
style['labels'] = labels
style = {k: v for k, v in style.items()
if k not in ['zorder', 'label']}
style['vert'] = not self.invert_axes
format_kdims = [kd.clone(value_format=None) for kd in element.kdims]
return (data,), style, {'dimensions': [format_kdims, element.vdims[0]]}
[docs] def init_artists(self, ax, plot_args, plot_kwargs):
artists = ax.boxplot(*plot_args, **plot_kwargs)
artists['artist'] = artists['boxes']
return artists
[docs] def teardown_handles(self):
for g in ('whiskers', 'fliers', 'medians', 'boxes', 'caps', 'means'):
for v in self.handles.get(g, []):
v.remove()
[docs]class SideBoxPlot(AdjoinedPlot, BoxPlot):
bgcolor = param.Parameter(default=(1, 1, 1, 0), doc="""
Make plot background invisible.""")
border_size = param.Number(default=0, doc="""
The size of the border expressed as a fraction of the main plot.""")
xaxis = param.ObjectSelector(default='bare',
objects=['top', 'bottom', 'bare', 'top-bare',
'bottom-bare', None], doc="""
Whether and where to display the xaxis, bare options allow suppressing
all axis labels including ticks and xlabel. Valid options are 'top',
'bottom', 'bare', 'top-bare' and 'bottom-bare'.""")
yaxis = param.ObjectSelector(default='bare',
objects=['left', 'right', 'bare', 'left-bare',
'right-bare', None], doc="""
Whether and where to display the yaxis, bare options allow suppressing
all axis labels including ticks and ylabel. Valid options are 'left',
'right', 'bare' 'left-bare' and 'right-bare'.""")
def __init__(self, *args, **kwargs):
super(SideBoxPlot, self).__init__(*args, **kwargs)
if self.adjoined:
self.invert_axes = not self.invert_axes
[docs]class ViolinPlot(BoxPlot):
"""
BoxPlot plots the ErrorBar Element type and supporting
both horizontal and vertical error bars via the 'horizontal'
plot option.
"""
bandwidth = param.Number(default=None, doc="""
Allows supplying explicit bandwidth value rather than relying
on scott or silverman method.""")
inner = param.ObjectSelector(objects=['box', 'medians', None],
default='box', doc="""
Inner visual indicator for distribution values:
* box - A small box plot
* stick - Lines indicating each sample value
* quartiles - Indicates first, second and third quartiles
""")
_plot_methods = dict(single='violinplot')
style_opts = ['showmeans', 'facecolors', 'showextrema', 'bw_method',
'widths', 'stats_color', 'box_color', 'alpha', 'edgecolors']
_nonvectorized_styles = [
s for s in style_opts if s not in ('facecolors', 'edgecolors', 'widths')
]
[docs] def init_artists(self, ax, plot_args, plot_kwargs):
box_color = plot_kwargs.pop('box_color', 'black')
stats_color = plot_kwargs.pop('stats_color', 'black')
facecolors = plot_kwargs.pop('facecolors', [])
edgecolors = plot_kwargs.pop('edgecolors', 'black')
labels = plot_kwargs.pop('labels')
alpha = plot_kwargs.pop('alpha', 1.)
showmedians = self.inner == 'medians'
bw_method = self.bandwidth or 'scott'
artists = ax.violinplot(*plot_args, bw_method=bw_method,
showmedians=showmedians, **plot_kwargs)
if self.inner == 'box':
box = ax.boxplot(*plot_args, positions=plot_kwargs['positions'],
showfliers=False, showcaps=False, patch_artist=True,
boxprops={'facecolor': box_color},
medianprops={'color': 'white'}, widths=0.1,
labels=labels)
artists.update(box)
for body, color in zip(artists['bodies'], facecolors):
body.set_facecolors(color)
body.set_edgecolors(edgecolors)
body.set_alpha(alpha)
for stat in ['cmedians', 'cmeans', 'cmaxes', 'cmins', 'cbars']:
if stat in artists:
artists[stat].set_edgecolors(stats_color)
artists['bodies'] = artists['bodies']
return artists
def get_data(self, element, ranges, style):
if element.kdims:
with sorted_context(False):
groups = element.groupby(element.kdims).data.items()
else:
groups = [(element.label, element)]
data, labels, colors = [], [], []
elstyle = self.lookup_options(element, 'style')
for i, (key, group) in enumerate(groups):
if element.kdims:
label = ','.join([d.pprint_value(v) for d, v in zip(element.kdims, key)])
else:
label = key
data.append(group[group.vdims[0]])
labels.append(label)
colors.append(elstyle[i].get('facecolors', 'blue'))
style['positions'] = list(range(len(data)))
style['labels'] = labels
style['facecolors'] = colors
if element.ndims > 0:
element = element.aggregate(function=np.mean)
else:
element = element.clone([(element.aggregate(function=np.mean),)])
new_style = self._apply_transforms(element, ranges, style)
style = {k: v for k, v in new_style.items()
if k not in ['zorder', 'label']}
style['vert'] = not self.invert_axes
format_kdims = [kd.clone(value_format=None) for kd in element.kdims]
ticks = {'yticks' if self.invert_axes else 'xticks': list(enumerate(labels))}
return (data,), style, dict(dimensions=[format_kdims, element.vdims[0]], **ticks)
[docs] def teardown_handles(self):
box_artists = ('cmedians', 'cmeans', 'cmaxes', 'cmins', 'cbars', 'bodies')
violin_artists = ('whiskers', 'fliers', 'medians', 'boxes', 'caps', 'means')
for group in box_artists+violin_artists:
for v in self.handles.get(group, []):
v.remove()