ó
Śs§]c           @   s"   d  d l  Z d   Z d   Z d S(   i˙˙˙˙Nc         C   s  t  |   } t j | d  } t j | d  } x| t |  D]n } xe t |  D]W } |  | | f } | d k rR | | } | | d c | 7<| | d c d 7<qR qR Wq? Wt j | d  } x; t |  D]- \ } }	 |	 d k rŃ | | |	 | | <qŃ qŃ Wt j |   }
 xv t |  D]h } x_ t |  D]Q } |  | | f } | | } | | d } | d k r1| | |
 | | f <q1q1WqW|
 S(   Ni   i    (   t   lent   npt   zerost   ranget	   enumeratet
   zeros_like(   t   matt   nt   totst   countst   it   jt   observedt   st   avgst   countt   oe_matt   expected(    (    si   /oak/stanford/groups/akundaje/marinovg/Symbiodinium/2019-09-27-figures/multi_mds/miniMDS-master/hic_oe.pyt   oe   s.    

c   
      C   s  t  |   } t j | d  } t j | d  } x| t |  D]n } xe t |  D]W } |  | | f } | d k rR | | } | | d c | 7<| | d c d 7<qR qR Wq? Wt j | d  } x; t |  D]- \ } }	 |	 d k rŃ | | |	 | | <qŃ qŃ W| S(   Ni   i    (   R    R   R   R   R   (
   R   R   R   R	   R
   R   R   R   R   R   (    (    si   /oak/stanford/groups/akundaje/marinovg/Symbiodinium/2019-09-27-figures/multi_mds/miniMDS-master/hic_oe.pyt   get_expected"   s    
(   t   numpyR   R   R   (    (    (    si   /oak/stanford/groups/akundaje/marinovg/Symbiodinium/2019-09-27-figures/multi_mds/miniMDS-master/hic_oe.pyt   <module>   s   	