3
¦s§]  ã               @   s   d dl Zdd„ Zdd„ ZdS )é    Nc             C   s2  t | ƒ}tj|d ƒ}tj|d ƒ}xft|ƒD ]Z}xTt|ƒD ]H}| ||f }|dkr<|| }||d   |7  < ||d   d7  < q<W q.W tj|d ƒ}x,t|ƒD ] \}}	|	dkr¤|| |	 ||< q¤W tj| ƒ}
xZt|ƒD ]N}xHt|ƒD ]<}| ||f }|| }||d  }|dkrê|| |
||f< qêW qÜW |
S )Né   r   )ÚlenÚnpÚzerosÚrangeÚ	enumerateÚ
zeros_like)ÚmatÚnÚtotsÚcountsÚiÚjÚobservedÚsÚavgsÚcountZoe_matÚexpected© r   úi/oak/stanford/groups/akundaje/marinovg/Symbiodinium/2019-09-27-figures/multi_mds/miniMDS-master/hic_oe.pyÚoe   s.    
r   c       
      C   sÌ   t | ƒ}tj|d ƒ}tj|d ƒ}xft|ƒD ]Z}xTt|ƒD ]H}| ||f }|dkr<|| }||d   |7  < ||d   d7  < q<W q.W tj|d ƒ}x,t|ƒD ] \}}	|	dkr¤|| |	 ||< q¤W |S )Nr   r   )r   r   r   r   r   )
r	   r
   r   r   r   r   r   r   r   r   r   r   r   Úget_expected"   s    r   )Únumpyr   r   r   r   r   r   r   Ú<module>   s   