U
|
ý°dö ã @ sH d dl Zd dlZd dlmZ dd Zdd Zdd Z dd Z
|
d
|
d Z dS ) é Nc C sf t dddgdddgd¡ d¡} | jdgd d
|
d d ¡}| jdgd d
|
d d ¡}t ||¡ d S )NZAliceZBobZCarlé é )ÚnameÚager r F©Z
|
group_keysc S s | S ©N© ©Úgroupr r ú]d:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/groupby/test_apply_mutate.pyÚ<lambda> ó z$test_group_by_copy.<locals>.<lambda>c S s | ¡ S r )Úcopyr r r r r r )ÚpdÚ DataFrameZ set_indexÚgroupbyÚapplyÚtmÚassert_frame_equal)ÚdfZgrp_by_same_valueZ grp_by_copyr r r Útest_group_by_copy s þÿûÿr c C s t dgd dgd dgd dgd dgd d gd dgd dgd dgd d
|
d td d D tjjdddd¡} dd }dd }| d¡ |¡}| d¡ |¡}t ||¡ d S )NÚaé Úbé Úcé ÚdÚeÚfc S s g | ]}d | qS )Úgr )Ú.0Úxr r r Ú
|
<listcomp>$ s z&test_mutate_groups.<locals>.<listcomp>é é éd é )Úsize)Úcat1Úcat2Zcat3Úvalc S s, | ¡ } | jjdd| d< | d¡d ¡ S ©NÚmin)ÚmethodÚrankr* )r r+ r/ r r- ©r" r r r Úf_copy) s z"test_mutate_groups.<locals>.f_copyc S s$ | j jdd| d< | d¡d ¡ S r, )r+ r/ r r- r0 r r r Ú f_no_copy. s z%test_mutate_groups.<locals>.f_no_copyr) )
|
r r ÚrangeÚnpÚrandomÚrandintr r r Úassert_series_equal)r r1 r2 Z
|
grpby_copyZ grpby_no_copyr r r Útest_mutate_groups s0 ÿþýüûúöÿr8 c C sf t dddddddddg tdd¡} | jddd d d
|
¡}| jddd d d
|
¡}t ||¡ d S ) Nr$ r é é )ÚkeyÚvaluer; Tr c S s | d d
j S r ©r; r0 r r r r = r z/test_no_mutate_but_looks_like.<locals>.<lambda>c S s | j S r r= r0 r r r r > r )r r r3 r r r r7 )r Zresult1Zresult2r r r Ú test_no_mutate_but_looks_like7 s &r> c
|
C s t ddddddgddddddgd ¡} d
|
d }| jd gd d |¡}t jddddddgt j ddddddg¡dd}t ||¡ d S )NÚAÚBr$ r r9 é é r )Úcol1Úcol2c S s d| j | jd df< | jS )Nr éÿÿÿÿrD )ÚlocÚindexrD r0 r r r ÚfnH s z-test_apply_function_with_indexing.<locals>.fnrC F)Zas_indexr )r r )r r$ )r r )r$ r9 )r$ rA )r$ rB rD )rG r ) r r r r ZSeriesÚ
|
MultiIndexÚ from_tuplesr r7 )r rH ÚresultÚexpectedr r r Ú!test_apply_function_with_indexingB s ÿÿûrM c C sÜ t jdddgdddgdddgdddgdddgdddgdt j ddddd d
|
g¡d } d d }| jddd |¡}t jddddddddgddddddddgddddddddggt j ddddddddg¡d }t ||¡ d S )Nr$ r r9 )©ÚCÚjulian©r@ Úgeoffrey©r? rP ©r@ rP ©r? rR ©rO rR rS rU rT rQ rN rV )Úcolumnsc S s& | j d d }| jdd| d|f< | S )Nr r$ )ÚaxisÚsum)rW rY )Zgroupedr r r r Ú
|
add_columnn s z8test_apply_mutate_columns_multiindex.<locals>.add_column)ÚlevelrX r r: )rR r? rR )rR r@ rR )rR rO rR )rR rY rR )rP r? rP )rP r@ rP )rP rO rP )rP rY rP )r r rI rJ r r r r )r rZ rK rL r r r Ú$test_apply_mutate_columns_multiindexW s\ úúÿ÷øýøÿñ r\ ) Únumpyr4 Zpandasr Zpandas._testingZ_testingr r r8 r> rM r\ r r r r Ú<module> s
|
