mirror of https://github.com/Kozea/pygal.git
Python to generate nice looking SVG graph
http://pygal.org/
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
59 lines
1.9 KiB
59 lines
1.9 KiB
# -*- coding: utf-8 -*- |
|
# This file is part of pygal |
|
# |
|
# A python svg graph plotting library |
|
# Copyright © 2012 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/>. |
|
""" |
|
Interpolation using scipy |
|
""" |
|
from pygal.util import ident |
|
|
|
scipy = None |
|
try: |
|
import scipy |
|
from scipy import interpolate |
|
except ImportError: |
|
pass |
|
|
|
KINDS = ['cubic', 'univariate', 'quadratic', 'slinear', 'nearest', 'zero'] |
|
|
|
|
|
def interpolation(x, y, kind): |
|
"""Make the interpolation function""" |
|
assert scipy is not None, ( |
|
'You must have scipy installed to use interpolation') |
|
order = None |
|
if len(y) < len(x): |
|
x = x[:len(y)] |
|
|
|
x, y = zip(*filter(lambda t: None not in t, zip(x, y))) |
|
|
|
if len(x) < 2: |
|
return ident |
|
if isinstance(kind, int): |
|
order = kind |
|
elif kind in KINDS: |
|
order = {'nearest': 0, 'zero': 0, 'slinear': 1, |
|
'quadratic': 2, 'cubic': 3, 'univariate': 3}[kind] |
|
if order and len(x) <= order: |
|
kind = len(x) - 1 |
|
if kind == 'krogh': |
|
return interpolate.KroghInterpolator(x, y) |
|
elif kind == 'barycentric': |
|
return interpolate.BarycentricInterpolator(x, y) |
|
elif kind == 'univariate': |
|
return interpolate.InterpolatedUnivariateSpline(x, y) |
|
return interpolate.interp1d(x, y, kind=kind, bounds_error=False)
|
|
|