← Back
Editing: lambertw.cpython-311.pyc
� d�c� � �p � d Z ddlZ ddlZddlmZ n # e$ r Y nw xY wd� Zd� Z e dk r e � � dS dS )z�Compute a Pade approximation for the principal branch of the Lambert W function around 0 and compare it to various other approximations. � Nc �n � d� t d� � D � � } t j | dd� � \ }}||fS )Nc �P � g | ]#}t j t j d |�� � ��$S )r )�n)�mpmath�diff�lambertw)�.0r s �D/usr/lib/python3/dist-packages/scipy/special/_precompute/lambertw.py� <listcomp>z!lambertw_pade.<locals>.<listcomp> s+ � � E� E� E�q�f�k�&�/�1��2�2�2� E� E� E� � � � )�ranger �pade)�derivs�p�qs r � lambertw_pader s; � � E� E�E�!�H�H� E� E� E�F� �;�v�q�!� $� $�D�A�q� �a�4�Kr c �� � t t � � t j d� � 5 t � � \ } }| d d d� |d d d� }} t d� | � � � � t d� |� � � � d d d � � n # 1 swxY w Y t j ddd� � t j ddd� � }}t j ||� � \ }}|d|z z }g }|� � � D ]6}|� t t j |� � � � � � �7t j |� � � |j � � }t! j d d � � � \ }}t j d � | D � � � � } t j d � |D � � � � }t j | |� � t j ||� � z } t' | |z � � } |d � ||| � � t j |� � t j t j |� � � � z } t' | |z � � } |d � ||| � � t j dt j d � � |z d z z � � } d| z | dz d z z } t' | |z � � }|d � |||� � }|� ||� � � � � � �� � t! j � � t! j d d � � � \ }}| | k }|� |||� � }t j dd� � }|� dt' |� � z dz |d� � |� ||�� � t! j � � d S )N�2 ���zp = {}zq = {}g ��g �?�K y �?r � )�nrows�ncolsc �, � g | ]}t |� � ��S � ��float)r �p0s r r zmain.<locals>.<listcomp>' � � �(�(�(��%��)�)�(�(�(r c �, � g | ]}t |� � ��S r r )r �q0s r r zmain.<locals>.<listcomp>( r" r r r )�axg333333ӿg333333�?g �g�������?�r) �print�__doc__r �workdpsr �format�np�linspace�meshgrid�flatten�append�complexr �array�reshape�shape�plt�subplots�polyval�abs� pcolormesh�log�sqrt�exp�colorbar�ravel�tolist�show�plot)r r �x�y�z� lambertw_std�z0�fig�axes� pade_approx�pade_err� asy_approx�asy_err� series_approx� series_err�imr% � pade_better�ts r �mainrQ s� � � �'�N�N�N� � �� � � "� "� �����1����2��w��$�$�B�$��1�� �h�o�o�a� � �!�!�!� �h�o�o�a� � �!�!�!� "� "� "� "� "� "� "� "� "� "� "���� "� "� "� "� �;�t�S�"� %� %�r�{�4��b�'A�'A�q�A� �;�q�!� � �D�A�q� �B�q�D��A��L��i�i�k�k� :� :�����G�F�O�B�$7�$7� 8� 8�9�9�9�9��8�L�)�)�1�1�!�'�:�:�L�� �1�A�.�.�.�I�C�� ��(�(�a�(�(�(�)�)�A� ��(�(�a�(�(�(�)�)�A��*�Q��"�"�2�:�a��#3�#3�3�K��;� �-�.�.�H���G���q�!�X�&�&�&������R�V�B�F�1�I�I� .� .�.�J��*�|�+�,�,�G���G���q�!�W�%�%�%� ���2�6�!�9�9�Q�;��?�#�$�$�A���F�Q��T�!�V�O�M��]�\�1�2�2�J� �a�� � �A�q�*� -� -�B��L�L��� � � � �+�+�-�-�L�.�.�.��H�J�J�J��l��!�,�,�,�G�C���W�$�K� � � �q�!�[� )� )�B� � �D�#� � �A��G�G�D��Q���K�#� �q�#�&�&�&��L�L���L�����H�J�J�J�J�Js �A,B!�!B%�(B%�__main__) r( �numpyr+ r �matplotlib.pyplot�pyplotr4 � ImportErrorr rQ �__name__r r r �<module>rX s� ��� � � � � � ��M�M�M�#�#�#�#�#�#�#��� � � ��D� ����� � � +� +� +�\ �z����D�F�F�F�F�F� �s � ��
Save File
Cancel