mirror of https://github.com/Kozea/pygal.git
Jean-Marc Martins
11 years ago
19 changed files with 519 additions and 53 deletions
@ -0,0 +1,208 @@
|
||||
# -*- coding: utf-8 -*- |
||||
# This file is part of pygal |
||||
# |
||||
# A python svg graph plotting library |
||||
# Copyright © 2012-2014 Kozea |
||||
# |
||||
# This library is free software: you can redistribute it and/or modify it under |
||||
# the terms of the GNU Lesser General Public License as published by the Free |
||||
# Software Foundation, either version 3 of the License, or (at your option) any |
||||
# later version. |
||||
# |
||||
# This library is distributed in the hope that it will be useful, but WITHOUT |
||||
# ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS |
||||
# FOR A PARTICULAR PURPOSE. See the GNU Lesser General Public License for more |
||||
# details. |
||||
# |
||||
# You should have received a copy of the GNU Lesser General Public License |
||||
# along with pygal. If not, see <http://www.gnu.org/licenses/>. |
||||
""" |
||||
Box plot |
||||
""" |
||||
|
||||
from __future__ import division |
||||
from pygal.graph.graph import Graph |
||||
from pygal.util import compute_scale, decorate |
||||
from pygal._compat import is_list_like |
||||
|
||||
|
||||
class Box(Graph): |
||||
""" |
||||
Box plot |
||||
For each series, shows the median value, the 25th and 75th percentiles, |
||||
and the values within |
||||
1.5 times the interquartile range of the 25th and 75th percentiles. |
||||
|
||||
See http://en.wikipedia.org/wiki/Box_plot |
||||
""" |
||||
_series_margin = .06 |
||||
|
||||
def __init__(self, *args, **kwargs): |
||||
super(Box, self).__init__(*args, **kwargs) |
||||
|
||||
@property |
||||
def _format(self): |
||||
"""Return the value formatter for this graph""" |
||||
sup = super(Box, self)._format |
||||
|
||||
def format_maybe_quartile(x): |
||||
if is_list_like(x): |
||||
if len(x) == 5: |
||||
return 'Q1: %s Q2: %s Q3: %s' % tuple(map(sup, x[1:4])) |
||||
else: |
||||
return sup(x) |
||||
return format_maybe_quartile |
||||
|
||||
def _compute(self): |
||||
""" |
||||
Compute parameters necessary for later steps |
||||
within the rendering process |
||||
""" |
||||
for serie in self.series: |
||||
serie.values = self._box_points(serie.values) |
||||
|
||||
if self._min: |
||||
self._box.ymin = min(self._min, self.zero) |
||||
if self._max: |
||||
self._box.ymax = max(self._max, self.zero) |
||||
|
||||
x_pos = [ |
||||
x / self._len for x in range(self._len + 1) |
||||
] if self._len > 1 else [0, 1] # Center if only one value |
||||
|
||||
self._points(x_pos) |
||||
|
||||
y_pos = compute_scale( |
||||
self._box.ymin, self._box.ymax, self.logarithmic, self.order_min |
||||
) if not self.y_labels else list(map(float, self.y_labels)) |
||||
|
||||
self._x_labels = self.x_labels and list(zip(self.x_labels, [ |
||||
(i + .5) / self._order for i in range(self._order)])) |
||||
self._y_labels = list(zip(map(self._format, y_pos), y_pos)) |
||||
|
||||
def _plot(self): |
||||
""" |
||||
Plot the series data |
||||
""" |
||||
for index, serie in enumerate(self.series): |
||||
self._boxf(self._serie(index), serie, index) |
||||
|
||||
def _boxf(self, serie_node, serie, index): |
||||
""" |
||||
For a specific series, draw the box plot. |
||||
""" |
||||
# Note: q0 and q4 do not literally mean the zero-th quartile |
||||
# and the fourth quartile, but rather the distance from 1.5 times |
||||
# the inter-quartile range to Q1 and Q3, respectively. |
||||
boxes = self.svg.node(serie_node['plot'], class_="boxes") |
||||
|
||||
metadata = serie.metadata.get(0) |
||||
|
||||
box = decorate( |
||||
self.svg, |
||||
self.svg.node(boxes, class_='box'), |
||||
metadata) |
||||
val = self._format(serie.values) |
||||
|
||||
x_center, y_center = self._draw_box(box, serie.values, index) |
||||
self._tooltip_data(box, val, x_center, y_center, classes="centered") |
||||
self._static_value(serie_node, val, x_center, y_center) |
||||
|
||||
def _draw_box(self, parent_node, quartiles, box_index): |
||||
""" |
||||
Return the center of a bounding box defined by a box plot. |
||||
Draws a box plot on self.svg. |
||||
""" |
||||
width = (self.view.x(1) - self.view.x(0)) / self._order |
||||
series_margin = width * self._series_margin |
||||
left_edge = self.view.x(0) + width * box_index + series_margin |
||||
width -= 2 * series_margin |
||||
|
||||
# draw lines for whiskers - bottom, median, and top |
||||
for i, whisker in enumerate( |
||||
(quartiles[0], quartiles[2], quartiles[4])): |
||||
whisker_width = width if i == 1 else width / 2 |
||||
shift = (width - whisker_width) / 2 |
||||
xs = left_edge + shift |
||||
xe = left_edge + width - shift |
||||
self.svg.line( |
||||
parent_node, |
||||
coords=[(xs, self.view.y(whisker)), |
||||
(xe, self.view.y(whisker))], |
||||
class_='reactive tooltip-trigger', |
||||
attrib={'stroke-width': 3}) |
||||
|
||||
# draw lines connecting whiskers to box (Q1 and Q3) |
||||
self.svg.line( |
||||
parent_node, |
||||
coords=[(left_edge + width / 2, self.view.y(quartiles[0])), |
||||
(left_edge + width / 2, self.view.y(quartiles[1]))], |
||||
class_='reactive tooltip-trigger', |
||||
attrib={'stroke-width': 2}) |
||||
self.svg.line( |
||||
parent_node, |
||||
coords=[(left_edge + width / 2, self.view.y(quartiles[4])), |
||||
(left_edge + width / 2, self.view.y(quartiles[3]))], |
||||
class_='reactive tooltip-trigger', |
||||
attrib={'stroke-width': 2}) |
||||
|
||||
# box, bounded by Q1 and Q3 |
||||
self.svg.node( |
||||
parent_node, |
||||
tag='rect', |
||||
x=left_edge, |
||||
y=self.view.y(quartiles[1]), |
||||
height=self.view.y(quartiles[3]) - self.view.y(quartiles[1]), |
||||
width=width, |
||||
class_='subtle-fill reactive tooltip-trigger') |
||||
|
||||
return (left_edge + width / 2, self.view.y( |
||||
sum(quartiles) / len(quartiles))) |
||||
|
||||
@staticmethod |
||||
def _box_points(values): |
||||
""" |
||||
Return a 5-tuple of Q1 - 1.5 * IQR, Q1, Median, Q3, |
||||
and Q3 + 1.5 * IQR for a list of numeric values. |
||||
|
||||
The iterator values may include None values. |
||||
|
||||
Uses quartile definition from Mendenhall, W. and |
||||
Sincich, T. L. Statistics for Engineering and the |
||||
Sciences, 4th ed. Prentice-Hall, 1995. |
||||
""" |
||||
def median(seq): |
||||
n = len(seq) |
||||
if n % 2 == 0: # seq has an even length |
||||
return (seq[n // 2] + seq[n // 2 - 1]) / 2 |
||||
else: # seq has an odd length |
||||
return seq[n // 2] |
||||
|
||||
# sort the copy in case the originals must stay in original order |
||||
s = sorted([x for x in values if x is not None]) |
||||
n = len(s) |
||||
if not n: |
||||
return 0, 0, 0, 0, 0 |
||||
else: |
||||
q2 = median(s) |
||||
# See 'Method 3' in http://en.wikipedia.org/wiki/Quartile |
||||
if n % 2 == 0: # even |
||||
q1 = median(s[:n // 2]) |
||||
q3 = median(s[n // 2:]) |
||||
else: # odd |
||||
if n == 1: # special case |
||||
q1 = s[0] |
||||
q3 = s[0] |
||||
elif n % 4 == 1: # n is of form 4n + 1 where n >= 1 |
||||
m = (n - 1) // 4 |
||||
q1 = 0.25 * s[m-1] + 0.75 * s[m] |
||||
q3 = 0.75 * s[3*m] + 0.25 * s[3*m + 1] |
||||
else: # n is of form 4n + 3 where n >= 1 |
||||
m = (n - 3) // 4 |
||||
q1 = 0.75 * s[m] + 0.25 * s[m+1] |
||||
q3 = 0.25 * s[3*m+1] + 0.75 * s[3*m+2] |
||||
|
||||
iqr = q3 - q1 |
||||
q0 = q1 - 1.5 * iqr |
||||
q4 = q3 + 1.5 * iqr |
||||
return q0, q1, q2, q3, q4 |
@ -0,0 +1,87 @@
|
||||
# -*- coding: utf-8 -*- |
||||
# This file is part of pygal |
||||
# |
||||
# A python svg graph plotting library |
||||
# Copyright © 2012-2014 Kozea |
||||
# |
||||
# This library is free software: you can redistribute it and/or modify it under |
||||
# the terms of the GNU Lesser General Public License as published by the Free |
||||
# Software Foundation, either version 3 of the License, or (at your option) any |
||||
# later version.# |
||||
|
||||
# This library is distributed in the hope that it will be useful, but WITHOUT |
||||
# ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS |
||||
# FOR A PARTICULAR PURPOSE. See the GNU Lesser General Public License for more |
||||
# details. |
||||
# |
||||
# You should have received a copy of the GNU Lesser General Public License |
||||
# along with pygal. If not, see <http://www.gnu.org/licenses/>. |
||||
""" |
||||
Pie chart |
||||
|
||||
""" |
||||
|
||||
from __future__ import division |
||||
from pygal.util import decorate |
||||
from pygal.graph.graph import Graph |
||||
from pygal.adapters import positive, none_to_zero |
||||
from math import pi |
||||
|
||||
|
||||
class Pie(Graph): |
||||
"""Pie graph""" |
||||
|
||||
_adapters = [positive, none_to_zero] |
||||
|
||||
def slice(self, serie_node, start_angle, serie, total): |
||||
"""Make a serie slice""" |
||||
dual = self._len > 1 and not self._order == 1 |
||||
|
||||
slices = self.svg.node(serie_node['plot'], class_="slices") |
||||
serie_angle = 0 |
||||
total_perc = 0 |
||||
original_start_angle = start_angle |
||||
center = ((self.width - self.margin.x) / 2., |
||||
(self.height - self.margin.y) / 2.) |
||||
radius = min(center) |
||||
for i, val in enumerate(serie.values): |
||||
perc = val / total |
||||
angle = 2 * pi * perc |
||||
serie_angle += angle |
||||
val = '{0:.2%}'.format(perc) |
||||
metadata = serie.metadata.get(i) |
||||
slice_ = decorate( |
||||
self.svg, |
||||
self.svg.node(slices, class_="slice"), |
||||
metadata) |
||||
if dual: |
||||
small_radius = radius * .9 |
||||
big_radius = radius |
||||
else: |
||||
big_radius = radius * .9 |
||||
small_radius = radius * self.config.inner_radius |
||||
|
||||
self.svg.slice( |
||||
serie_node, slice_, big_radius, small_radius, |
||||
angle, start_angle, center, val) |
||||
start_angle += angle |
||||
total_perc += perc |
||||
|
||||
if dual: |
||||
val = '{0:.2%}'.format(total_perc) |
||||
self.svg.slice(serie_node, |
||||
self.svg.node(slices, class_="big_slice"), |
||||
radius * .9, 0, serie_angle, |
||||
original_start_angle, center, val) |
||||
return serie_angle |
||||
|
||||
def _plot(self): |
||||
total = sum(map(sum, map(lambda x: x.values, self.series))) |
||||
|
||||
if total == 0: |
||||
return |
||||
current_angle = 0 |
||||
for index, serie in enumerate(self.series): |
||||
angle = self.slice( |
||||
self._serie(index), current_angle, serie, total) |
||||
current_angle += angle |
Loading…
Reference in new issue