ó ^{UZc@s»dgZddlZddlmZmZmZddlmZerVddl Z nd„Z d„Z d„Z de fd „ƒYZeeƒZd „Zd e_d „Zd „ZdS(tbsiÿÿÿÿN(t have_pandast no_picklingtassert_no_pickling(tstateful_transformcCs¼yddlm}Wntk r3tdƒ‚nXtjtj|dtƒƒ}|jdksgt‚|j ƒt |ƒ}tj|ƒ}|jdkrÇ|j ddkrÇ|dd…df}n|jdksÜt‚tj |ƒtj |ƒkstj |ƒtj |ƒkr'tdƒ‚nt|ƒ|d}tj|j d|fdtƒ}xXt|ƒD]J}tj|fƒ}d||<|||||fƒ|dd…|fs 5cCsƒd„}|ddgddƒ|ddgddƒ|ddgddgdd gƒ|ttdƒƒddgd d gƒdS( NcSs%tjt||ƒ|ƒs!t‚dS(N(R tallcloseR)R(RR'texpected((s,lib/python2.7/site-packages/patsy/splines.pyttFsi igà?ig333333Ó?i gffffffæ?igš™™™™™@g333333@(tlistR(R,((s,lib/python2.7/site-packages/patsy/splines.pyttest__R_compat_quantileEs  "tBScBs\eZdZd„Zdddeddd„Zd„Zdddeddd„Ze Z RS(s3 bs(x, df=None, knots=None, degree=3, include_intercept=False, lower_bound=None, upper_bound=None) Generates a B-spline basis for ``x``, allowing non-linear fits. The usual usage is something like:: y ~ 1 + bs(x, 4) to fit ``y`` as a smooth function of ``x``, with 4 degrees of freedom given to the smooth. :arg df: The number of degrees of freedom to use for this spline. The return value will have this many columns. You must specify at least one of ``df`` and ``knots``. :arg knots: The interior knots to use for the spline. If unspecified, then equally spaced quantiles of the input data are used. You must specify at least one of ``df`` and ``knots``. :arg degree: The degree of the spline to use. :arg include_intercept: If ``True``, then the resulting spline basis will span the intercept term (i.e., the constant function). If ``False`` (the default) then this will not be the case, which is useful for avoiding overspecification in models that include multiple spline terms and/or an intercept term. :arg lower_bound: The lower exterior knot location. :arg upper_bound: The upper exterior knot location. A spline with ``degree=0`` is piecewise constant with breakpoints at each knot, and the default knot positions are quantiles of the input. So if you find yourself in the situation of wanting to quantize a continuous variable into ``num_bins`` equal-sized bins with a constant effect across each bin, you can use ``bs(x, num_bins - 1, degree=0)``. (The ``- 1`` is because one degree of freedom will be taken by the intercept; alternatively, you could leave the intercept term out of your model and use ``bs(x, num_bins, degree=0, include_intercept=True)``. A spline with ``degree=1`` is piecewise linear with breakpoints at each knot. The default is ``degree=3``, which gives a cubic b-spline. This is a stateful transform (for details see :ref:`stateful-transforms`). If ``knots``, ``lower_bound``, or ``upper_bound`` are not specified, they will be calculated from the data and then the chosen values will be remembered and re-used for prediction from the fitted model. Using this function requires scipy be installed. .. note:: This function is very similar to the R function of the same name. In cases where both return output at all (e.g., R's ``bs`` will raise an error if ``degree=0``, while patsy's will not), they should produce identical output given identical input and parameter settings. .. warning:: I'm not sure on what the proper handling of points outside the lower/upper bounds is, so for now attempting to evaluate a spline basis at such points produces an error. Patches gratefully accepted. .. versionadded:: 0.2.0 cCsi|_d|_d|_dS(N(t_tmptNonet_degreet _all_knots(tself((s,lib/python2.7/site-packages/patsy/splines.pyt__init__ˆs  ic CsÅi|d6|d6|d6|d6|d6|d6}||jd= %ssAdf=%s with degree=%r implies %s knots, but %s knots were providediiÿÿÿÿR8R9s#lower_bound > upper_bound (%r > %r)sknots must be 1 dimensionals1some knot values (%s) fall below lower bound (%r)s1some knot values (%s) fall above upper bound (%r)(R0R<RR2R t concatenateR1RtlinspaceR)RRR R tanyRR3( R4ttmpR:RR!t n_inner_knotstknot_quantilest inner_knotsR8R9t all_knots((s,lib/python2.7/site-packages/patsy/splines.pytmemorize_finish£sp                 c Csƒt||j|jƒ}|s=|dd…dd…f}ntrt|tjtjfƒrtj|ƒ}|j|_qn|S(Ni( R R3R2Rt isinstancetpandastSeriest DataFrametindex( R4RR6RRR7R8R9R((s,lib/python2.7/site-packages/patsy/splines.pyt transformìsN( t__name__t __module__t__doc__R5R1tFalseR?RHRNRt __getstate__(((s,lib/python2.7/site-packages/patsy/splines.pyR/Ms:   I  cCsæddlm}ddlm}m}m}|jdƒ}d}|jdƒ}x}trÏ||dksmPn|d7}|jd|ƒ}|||!}i} x0|D](} | jd dƒ\} } | | |         ic Cstjdddƒ}t|dddgdddtƒ}|jdd ksRt‚tjdƒ}d||dks   ,  ­  -