Sindbad~EG File Manager
�
Mٜg� � �X � d dl mZ d dlmZ d dlZd dlmZ dd�Z G d� d� Z y) � )� annotations)�
TYPE_CHECKINGN)�import_optional_dependencyc � �� t rddl�n t d� ��j | ||�� d�fd� � }|S )ak
Generate a numba jitted groupby ewma function specified by values
from engine_kwargs.
Parameters
----------
nopython : bool
nopython to be passed into numba.jit
nogil : bool
nogil to be passed into numba.jit
parallel : bool
parallel to be passed into numba.jit
Returns
-------
Numba function
r N�numba)�nopython�nogil�parallelc � �� t j | j � }| d j � } t j | � j t j � }
t j |
|k\ | t j � |d< t dt | � � D �] } | | } t j | � }
|
|
j t j � z
}
�j t | � � D ]� }t j | | � se|
| s|r�#||xx |||dz
z z cc< |
| s�?| | | | k7 r || | | z || | z z || |z z | |< |r||xx |z
cc< �zd||< ��|
| s��| | | |< �� t j |
|k\ | t j � || < �� ||fS )z�
Compute online exponentially weighted mean per column over 2D values.
Takes the first observation as is, then computes the subsequent
exponentially weighted mean accounting minimum periods.
r � � �?) �np�empty�shape�copy�isnan�astype�int64�where�nan�range�len�prange)�values�deltas�minimum_periods�
old_wt_factor�new_wt�old_wt�adjust� ignore_na�result� weighted_avg�nobs�i�cur�is_observations�jr s ��D/usr/local/lib/python3.12/site-packages/pandas/core/window/online.py� online_ewmaz4generate_online_numba_ewma_func.<locals>.online_ewma% s� �� �" ���&�,�,�'�� �a�y�~�~�'� ����,�'�'�/�/����9���H�H�T�_� 4�l�B�F�F�K��q� ��q�#�f�+�&�A���)�C� !�x�x��}�n�O� �O�*�*�2�8�8�4� 4�D��\�\�#�c�(�+���x�x� �Q��0�&�q�)�� �q� �]�f�Q��U�m�%C�C� �*�1�-� +�A��#�a�&� 8�%+�A�Y� �a� �%@�V�c�RS�f�_�$U�%+�A�Y��%7�39� �Q�� &� &�q� �V� 3� �,/��q� �$�Q�'�&)�!�f�L��O�# ,�& ���� �!8�,����O�F�1�I�/ '�2 �v�~� � )r �
np.ndarrayr r, r �intr �floatr r. r r, r �boolr! r/ )r r r �jit)r r r
r* r s @r) � generate_online_numba_ewma_funcr1
s� �� �, ��*�7�3��
�Y�Y����Y�A�. ��. ��. � �. � � . �
� . � �
. � �. � �. � B�. �` �r+ c �" � e Zd Zdd�Zd� Zdd�Zy)� EWMMeanStatec �� � dd|z z }|| _ || _ || _ || _ |rdn|| _ d|z
| _ t
j | j | j dz
� | _ d | _ y )Nr
r )
�axisr r r! r r r �onesr �last_ewm)�self�comr r! r5 r �alphas r) �__init__zEWMMeanState.__init__Z sk � ��s�S�y�!���� ���
� �� �"��� #�c��� � �5�[����g�g�d�j�j����Q� � 7�8�� � ��
r+ c
� � ||||| j | j | j | j | j � \ }}|| _ |d | _ |S )N���)r r r r r! r7 )r8 r# r � min_periods�ewm_funcr" r s r) �run_ewmzEWMMeanState.run_ewme sW � �!� � � � � � � �K�K� �K�K� �K�K� �N�N�
���� �� � �r�
��
��
r+ c �z � t j | j | j dz
� | _ d | _ y )Nr )r r6 r r5 r r7 )r8 s r) �resetzEWMMeanState.resett s* � ��g�g�d�j�j����Q� � 7�8�� � ��
r+ N)�return�None)�__name__�
__module__� __qualname__r; r@ rB � r+ r) r3 r3 Y s � � �
� r+ r3 )r r/ r r/ r
r/ )
�
__future__r �typingr �numpyr �pandas.compat._optionalr r1 r3 rH r+ r) �<module>rM s? �� "� � � >�L��L� �L� �L�^ � r+
Sindbad File Manager Version 1.0, Coded By Sindbad EG ~ The Terrorists