ó
þubVc           @   s.   d  d l  Z  d  d l Z d „  Z d „  Z d S(   iÿÿÿÿNc         C   s³  g  g  } } d } xGt  d t |  ƒ ƒ D]0} |  | } d g | d } t | t ƒ r— xr t  d | d ƒ D]# } t t | d | ƒ ƒ | | <qm Wn7 x4 t  d | d ƒ D] } t t | | ƒ ƒ | | <q« Wd }	 t | t ƒ rö t | d ƒ }	 n t | ƒ }	 | d k r9| | j | ƒ | | j t |	 ƒ ƒ q) | j | ƒ | j t |	 ƒ ƒ q) Wt j | ƒ t j | ƒ j	 ƒ  }
 } |
 j	 ƒ  |
 j
 ƒ  |
 j	 ƒ  | } t j | ƒ S(   Ni    g        i   (   t   ranget   lent
   isinstancet   listt   floatt   powt   appendt   numpyt   matrixt	   transposet   getIt   asarray(   t   data2Dt   degreet   at   bt   jt   kt   datat   lnt   it   yt   At   Bt   coeff(    (    sl   /home/widom/html/widom/teaching/33-342/IsingData/2D/WangLandau/Jaymz07/wang-landau-python-master/fitCurve.pyt   fitPolynomial   s,    
$%$c         C   sÏ  d \ } } } } } } } } t  |  ƒ }	 xÕ t d |	 ƒ D]Ä }
 |  |
 } d \ } } t | t ƒ r} | d | d } } n |
 } | } | | d } } | | d 7} | | | 7} | | d 7} | | | 7} | | | 7} | | 7} | | 7} | | 7} q: W| | d |	 8} | | | |	 8} | | d |	 8} | | | |	 8} | | | |	 8} | | | | | | | d } | | | | | | | d } | |	 | | |	 | | |	 } | | | g S(   Ng        i    i   i   (   g        g        g        g        g        g        g        g        (   g        g        (   R   R    R   R   (   R   t   s11t   s12t   s22t   sy1t   sy2t   s1t   s2t   syt   NR   R   t   xR   t   x1t   x2t   out2t   out3t   out1(    (    sl   /home/widom/html/widom/teaching/33-342/IsingData/2D/WangLandau/Jaymz07/wang-landau-python-master/fitCurve.pyt   fitParabola#   s6    


"""(   t   stringR   R   R)   (    (    (    sl   /home/widom/html/widom/teaching/33-342/IsingData/2D/WangLandau/Jaymz07/wang-landau-python-master/fitCurve.pyt   <module>   s   	