U
|
ý°de ã @ sä d dl Zd dlZd dlZd dlmZmZmZmZm Z m
|
Z
|
d dl m Z dd Zej dddg¡dd Zej d
|
d d g¡d d Zej dddg¡dd Zej dddg¡ej dedddejgdeddgiedgdddfedejgdededdgiedgdddfg¡dd Zdd Zd d Zd d Zd!d" Zd#d$ Zej d%d&ge d'd(d)d*gd+d&ge d'd(d)d*gd+d&ge d'd(d)d*gd+fd&d,d&ge d-d.d)e d'd(d)e d/d0d)gd*d1d2gd+d&d,ge d-d.d)e d'd(d)gd*d1gd+d&d,ge d/d0d)e d'd(d)gd2d1gd+fg¡d3d4 Z ej d5de d6d7d)dgde d8d7d)d9gg¡d:d; Z ej d<ej d=d>gd?dej d*d1gd@de dAdBg¡ej d*d1gdCdDg¡ej dEdddFdGg¡dHdI Z!dJdK Z"ej dLdMd d gfdMd g fdMd*d d1gfdMdNd d*d1gfdOd d*gfdOd g fdOd*d*d1gfdOdNd d*d1gfg¡ej dPdg d&gd,gd&d,gg¡ej dQd=d>g¡dRdS Z#ej dTdMd d gfdMd g fdMd*d d1gfdMdNd d*d1gfdOd d*gfdOd g fdOd*d*d1gfdOdNd d*d1gfg¡dUdV Z$dWdX Z%dYdZ Z&d[d\ Z'ej d]dd^d_g¡d`da Z(ej d]dd^d_g¡dbdc Z)ddde Z*dfdg Z+ej ddddhg¡didj Z,ej dke-dd2d1d d*dldmgge-ddnd d1dmgge-dd1e-dndgd d*d1d2dldodNggd d*e-dndgd d*d1d2dldodNggg¡dpdq Z.drds Z/dtdu Z0dvdw Z1dxdy Z2dzd{ Z3ej d|ddd d*d1d2dlge4d}fdd*d d2ge4d~fddd d*d1d2dlge4d}fdd d d*d2ge4dfd*dd*d1dlge4dfd*d d*ge4d,fd dd1dlge4dfd d1dlge4dfg¡ej ddd
|
g¡d
d Z5ej 6d¡dd Z7dd Z8dS )é N)Ú DataFrameÚIndexÚ
|
MultiIndexÚSeriesÚ TimestampÚisnac C s | d¡}| ¡ }| jddgdddgf }tddgdd |_| ¡ }t ||¡ | d¡}| jddg }t ||¡ | ¡ }| jd
|
d gdddgf }tddgdd |_t ||¡ | d ¡}| j
|
d
|
d g }t ||¡ | d¡}| j
|
d dg }t ||¡ |d ¡ |d ¡ |d d¡ t j | j| d dkdf< t |d ¡ d sRtt |d ¡ d sltt |d d¡j
|
d sttdt j gddgd
|
dggddgd} | d¡}| ¡ }| j
|
dd g d¡}t ||¡ | j
|
dd g }|jddd}t ||¡ d S )NÚAé r ÚBÚCÚDÚbarÚfoo©Únameé é éÿÿÿÿé é é é ©ÚcolumnsÚany©Údropna)ÚgroupbyÚfirstÚlocr ÚindexÚ
|
sort_indexÚtmÚassert_frame_equalÚnthÚlastÚilocÚnpÚnanr ÚAssertionErrorr Ú set_index)ÚdfÚgroupedr Úexpectedr$ r% ÚgÚresult© r0 úTd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/groupby/test_nth.pyÚtest_first_last_nth sD
|
|
|
|
$
|
r2 Úmethodr r% c C s t ddddgddd|gd d¡}t|| }| dkrBddg}nddg}tj||d jd}tddgdd }t d|i|d
|
}t ||¡ d S ) Nr r r ©ÚaÚbr5 r r6 ©Údtyper ©r ) r r Úgetattrr' Úarrayr8 r r" r# )r3 Ú nulls_fixtureÚgroupsr/ ÚvaluesÚidxr- r0 r0 r1 Ú test_first_last_with_na_object@ s $
|
r@ r r c C sf t ddddgddd|gd}| d¡}| | ¡}| dkrH|jddg n |jddg }t ||¡ d S )Nr r r r4 r5 r ©r r r$ r& r" r# )r r< r+ r= r/ r- r0 r0 r1 Útest_nth_with_na_objectR s
|
|
|
$rB c C s> t dgd gd¡}|jddd}t|| }t ||¡ d S )Nr5 ©ÚidÚvaluerD F©Úas_index)r Ú from_dictr r: r" r# )r3 r+ r= r/ r0 r0 r1 Útest_first_last_with_None\ s rI z df, expectedr5 r rC rE rD r r9 r7 c C s" t | d¡| }t ||¡ d S )NrD )r: r r" r# )r3 r+ r- r/ r0 r0 r1 Ú"test_first_last_with_None_expandedg s rJ c
|
C s6 | ¡ }d|d< d|d< | d¡}| ¡ }|jddgddd ddgf }td
|
d gdd |_| ¡ }t ||¡ | ¡ }|jd dgddd ddgf }td
|
d gdd |_| ¡ }t ||¡ |
|
d¡}|j ddg }t ||¡ t t d}| d¡ tt d|dd}|jdkst|jdd ¡ } | jdks2td S )NTÚEr ÚFr r r
|
r r r r r r r r r é
|
é é ZIntCol)Údatar r Úint64©Úlevel)Úcopyr r r r r r! r" r# r% r$ r& ÚlistÚrangeÚappendr r8 r) )
|
Zdf_mixed_floatsr+ r, r r- r% r$ r? ÚsÚfr0 r0 r1 Útest_first_last_nth_dtypes{ s.
|
|
|
rZ c C s t dgttjgtdd} | d¡}| d¡j}t |j
|
¡ |¡ t |j ¡ |¡ | j}t |j d¡|¡ t |j d¡|¡ d S )Nr r7 )rP ÚnansrP r r ) r r r' r( Úobjectr r* r[ r" Úassert_series_equalr r% r$ )r+ r, r- r0 r0 r1 Ú test_first_last_nth_nan_dtype s
|
r^ c
|
C s t tdddgtdddgdddgd d
|
d gi} | d¡ ¡ }t ddd
|
ggttd td d gtdgddd}t ||¡ d S )Nz2012-01-01 00:00:00r5 r6 z2012-01-02 00:00:00ÚcÚdr ÚeZaaaarY r. z
|
2012-01-01z
|
2012-01-02r ©r r )r r r r r r" r# )Útestr/ r- r0 r0 r1 Ú test_first_strings_timestamps© s$ üÿ
|
ýrd c
|
C sh t dtjgddgddggddgd} | d¡}t | d¡| jdd g ¡ t | d¡| jdg ¡ t | d ¡| jg ¡ t | d
|
¡| jdd g ¡ t | d ¡| jdg ¡ t | d ¡| jg ¡ t |j
|
d¡| j
|
jdd g ¡ t |j
|
d¡| j
|
jdg ¡ t |dg d¡| dg jdd g ¡ t |jdd d| jdd g ¡ t |jd
|
d d| jdd g ¡ t |jdd d| jd d
¡ t |jd d d| jd d
¡ t dddddddddddddddddddd d d d dd d!d"g¡} | jdd#d$ d ¡}| jd
|
g }t ||¡ | jdd#d$ d%¡}| jg }t ||¡ t tj dd&d'¡d(d)} | d }| d }| |¡ ¡ }| |¡ d*d+ ¡}tj ||d#d, |jdks¼t|jdksÌt||dk jd }|jd |ksòt|jd |ksttjtd-d. |j|d#d/jdd0d W 5 Q R X t dtjgddgddggddgd} | d¡}|j
|
jdd1d}| j
|
jdd g }t ||¡ t dtjgdd%gddgddgddggddgd} | d¡}t | d¡| jdd%g ¡ t | dg¡| jdd%g ¡ t | ddg¡| jddd%dg ¡ t | dd
|
g¡| jdd d%dg ¡ t | ddd g¡| jddd d%dg ¡ t | ddd
|
g¡| jddd d%dg ¡ t | d g¡| jd g ¡ t | d%dg¡| jg ¡ tjd2d3dd4}t d|d5d6gd7} | jj| jjg}| j|d#d8 dd%d d
|
g¡}t d9d:d;d<d=d>d?d@dAdBdCdDg ¡} t dd5d6g| dE}t ||¡ d S )FNr r r r r r
|
r r r r éþÿÿÿéýÿÿÿr r r ÚgreenÚred)r r r r r ZhamÚeggsZporkgþI|î»ø?gçoB!²¿gHk :!4Àgn2dÝ?gC8fÙÀà?g£Wâ?gV¿Òùð,ï¿gN'Ùêrð?gÃGÄH"é¿gû( æ?)ÚcolorÚfoodÚtwoÚonerj rk F)rS rG r rM )éd r rQ r7 c S s
|
| j d S )Nr )r& )Úxr0 r0 r1 Ú<lambda>õ ó ztest_nth.<locals>.<lambda>)Z check_nameszFor a DataFrame©Úmatch)ÚsortTÚallz4/1/2014z 6/30/2014)ÚstartÚendÚfreqr5 r6 ©r r rF z2014/4/1z2014/4/4z 2014/4/29z 2014/4/30z2014/5/1z2014/5/6z 2014/5/29z 2014/5/30z2014/6/2z2014/6/5z 2014/6/27z 2014/6/30rb )r r' r( r r" r# r$ r& r r] r
|
r* ÚrandomÚrandintr Úapplyr r) ÚpytestÚraisesÚ
|
ValueErrorÚpdÚ
|
date_ranger ÚyearÚmonthÚ to_datetime)
|
r+ r. r/ r- rX Z expected2ÚvZbusiness_datesÚkeyZexpected_datesr0 r0 r1 Útest_nth¼ s² $
|
( ûûöÿí
|
"$
|
0
|
$$(( ôÿr c C s: | ddg¡}| d¡}| jddddg }t ||¡ d S )Nr r
|
r r r r )r r$ r& r" r# ©Z three_groupr, r/ r- r0 r0 r1 Útest_nth_multi_grouper/ s
|
r z#data, expected_first, expected_lastr z2012-02-01 14:00:00z
|
US/Central©Útzr )rD Útimer r
|
z2012-01-01 13:00:00zAmerica/New_Yorkz2012-03-01 12:00:00z Europe/Londonr r c C sÐ t | }|jddd ¡ }t |}dddg}t || || ¡ |jdddd ¡ }t ||ddg ¡ |jddd ¡ }t |}dddg}t || || ¡ |jdddd ¡ }t ||ddg ¡ d S )NrD FrF r r )r r r r" r# r% )rP Zexpected_firstZ expected_lastr+ r/ r- Úcolsr0 r0 r1 Útest_first_last_tz8 s 6
|
|
r zmethod, ts, alphaz
|
2013-01-01ú
|
US/Easternz
|
2013-01-02r6 c C s t td d¡}tdddg|tjddddd }t| d
|
¡| }ttj|d g|j d |t
|
d ddgdt ddgd
|
dd}t ||¡ d S )NÚabcÚcategoryr r Z20130101r r )Zperiodsr )ÚgroupÚcategory_stringÚ
|
datetimetzr r_ r7 z
|
2013-01-03r )r r r r9 )r rU Úastyper r r r: r Z Categoricalr8 r r r" r# )r3 ÚtsÚalphar r+ r/ r- r0 r0 r1 Ú test_first_last_tz_multi_column s$ ýÿ ÿüù r r> TFÚbooleanZInt64z
|
2020-01-01z
|
2020-02-01r )ÚunitÚfunctionÚminÚmaxc C s t ddg| d}| d¡}tddgdd}t| d|d}t d| i|d}t|d | }t ||¡ | d|i¡}t ||¡ d S ) Nr r r4 r5 r r6 )r r r9 ) r r r r r: r" r] Zaggr# ) r> r r+ r, r? Zexpected_seriesZexpected_frameZ result_seriesZ result_framer0 r0 r1 Ú+test_first_last_extension_array_keeps_dtype s
|
r c C s t dddddddddddg dddddddddddg dddddddddddg d} | dd g¡}| d
|
¡}| jd
|
d d d g }t ||¡ d S )Nr r rm rl ZdullZshiny)r r
|
r r r
|
r r r r rA r r0 r0 r1 Ú test_nth_multi_index_as_expected¹ sV õõõåÿ+
|
r zop, n, expected_rowsÚheadr Útailr rG c C sp t ddgddgddggddgd}|jd|d }|j| }|d k rR|| }|| }t|| |}t ||¡ d S )
|
Nr r r r r r r
|
r rF ©r r r& r: r" r# ) ÚopÚnÚ expected_rowsr rG r+ r. r- r/ r0 r0 r1 Útest_groupby_head_tailí s "
|
r¦ zop, n, expected_colsc C sv t dddgdddgdddgddd ggd
|
d d gd }|jdddgdd}|jd d
|f }t|| |}t ||¡ d S )Nr r r r r r r é rN r r
|
r r r ©Zaxisr¢ )r£ r¤ Z expected_colsr+ r. r- r/ r0 r0 r1 Ú test_groupby_head_tail_axis_1 s " ÿr© c C s t ddgddgddggddgd} | jd dg }| d¡}|jdd
|
}| d ¡}t || ¡ t ||¡ | d¡}|jdd
|
}| d ¡}t || ¡ t ||¡ | d¡}| d ¡}|jdd
|
}t ||¡ t || ¡ | d¡}| d ¡}|jdd
|
}t ||¡ t || ¡ d S ) Nr r r r r r r
|
r r ©r¤ )r r& r r r$ r" r# r¡ )r+ r- r. Zresult1Zresult2r0 r0 r1 Útest_group_selection_cache s, "
|
|
|
|
|
|
|
|
r« c C sp t dgdddgd} | d¡ d¡}| jd d
}t ||¡ | ddg¡ d¡}| jd d
}t ||¡ d S )Nr r5 r6 r_ ry rM rA ©r+ r/ r- r0 r0 r1 Útest_nth_empty= s r c C s t dddgdddgddtjgdddgdd d
|
ggd d d gd} | d ¡ d¡}| jddg }t ||¡ | d ¡jddd}| jddg }t ||¡ d S )Nr r6 rn r5 é2 r r_ éÈ r` é r r r
|
r r r r r r r ©r r' r( r r$ r& r" r# r¬ r0 r0 r1 Útest_nth_column_orderI s ,þ r² r r ru c C sf t tjdtjdtjgdddddgdd d
|
d d gd }| d¡jd| d}|jdd g }t ||¡ d S )Nr5 r6 r r r r r§ r r r r rN )r5 r6 r_ r r± ©r r+ r/ r- r0 r0 r1 Útest_nth_nan_in_grouperY s ýÿr´ c C sb t tjdtjdtjgdddddgd}| d¡d jd| d }|d jd
|
d g }t ||¡ d S ) Nr5 r6 r r r r r§ r4 r r r )r r' r( r r$ r& r" r] r³ r0 r0 r1 Ú test_nth_nan_in_grouper_seriesi s þÿrµ c C s t dddddgdt d¡g tdddddgd d
|
d } | d ¡ ¡ }t d t d¡g tdddg t dddg¡¡d}t dddgd d|_
|
t ||¡ d S )Nr ÚsecondÚthirdr ZNaTr5 r6 r_ r r7 )r r Ú
|
categoriesr r )r r¸ r ) r r' Z
|
datetime64r r r r r ÚCategoricalDtyper r r" r# r¬ r0 r0 r1 Ú,test_first_categorical_and_datetime_data_natx s ýÿÿþÿrº c
|
C s¸ t dddddgdddddgdddddgdd d
|
d d gd } | d di¡} | jddgd ¡ }t dddgtdd
|
d g t dd
|
d d g¡¡d}tjdddgddgd|_ t
|
||¡ d S )Nr r rn r¯ ZappleZorangeZmangoZjupiterZmercuryZmarsZvenus)r r
|
r r r r r r
|
)Zby)r r )r rn )r r¯ )r rn )Únames) r r r r r r r¹ r Ú from_tuplesr r" r# r¬ r0 r0 r1 Ú(test_first_multi_key_groupby_categorical s* üÿÿþÿ ÿr½ r$ c C sf t dg}t |||d|gdddddgdjdd}| dkrJt|| d}n t|| }t ||¡ d S )NÚyr r9 rR r$ r )r r r: r" r] )r3 r< r- rP r/ r0 r0 r1 Ú%test_groupby_last_first_nth_with_none¨ s
|
þý r¿ zarg, expected_rowsr r re r c C s: |j | }| |¡}| j| }t ||¡ t ||¡ d S ©N)r$ r& r" r# )Ú slice_test_dfÚslice_test_groupedÚargr¥ r/ Ú
|
equivalentr- r0 r0 r1 Ú
|
test_slice¹ s
|
|
|
|
rÅ c C s^ |j dddd
f }| ddtdd g¡}| jdddddddg }t ||¡ t ||¡ d S ) Nr r re r r r r r )r$ Úslicer& r" r# )rÁ r r/ rÄ r- r0 r0 r1 Útest_nth_indexedÍ s
|
rÇ c C s( t jtdd | d¡ W 5 Q R X d S )Nz Invalid indexrr g
ëQ¸ @)r} r~ Ú TypeErrorr$ ©r r0 r0 r1 Útest_invalid_argumentØ s rÊ c C s0 t jtdd | td d d¡ W 5 Q R X d S )Nz Invalid steprr r )r} r~ r r$ rÆ rÉ r0 r0 r1 Útest_negative_stepß s rË c C s8 | t ddg¡¡}| jdddddg }t ||¡ d S )Nr r r r r )r$ r' r; r& r" r# )rÁ r r/ r- r0 r0 r1 Ú test_np_intsæ s rÌ c C sh t dddgdddggddgdd d
|
gd } | j| jd d d d¡}| jd d
ddgf }t ||¡ d S )Nr r r r§ r Úzr¾ r r
|
r ry r r¨ r r )r r r& r$ r" r# r¬ r0 r0 r1 Ú!test_groupby_nth_with_column_axisî s þúrÎ z.start, stop, expected_values, expected_columnsÚABCDEZADrN ZABDZBCEZCErK Úcallc st t tdgtdgd}|jdddddgdd fdd fd dd
|
| | |}t |g|gd}t ||¡ d S ) Nr rÏ r r r r¨ c s t| |¡S rÀ )r$ rÆ ©rv Ústop©Úgbr0 r1 rp rq z2test_nth_slices_with_column_axis.<locals>.<lambda>c s j | |
S rÀ )r$ rÑ rÓ r0 r1 rp rq )rÐ r )r rV rU r r" r# )rv rÒ Zexpected_valuesZexpected_columnsr3 r+ r/ r- r0 rÓ r1 Ú test_nth_slices_with_column_axisý s
|
|
þý ýrÕ z<ignore:invalid value encountered in remainder:RuntimeWarningc C s® t ddgdtjgdtjgdtjggddgd} t ddggddgd}| ddg¡jdd }t ||¡ | ddg¡jdd }t ||¡ | ddg¡jd
|
d }t ||¡ d S ) Nr5 rÍ r6 r_ ÚXÚYr r rª r © r r' r( r r r" r# r¡ r$ ©r+ r- r/ r0 r0 r1 Útest_head_tail_dropna_true s ÿ rÚ c C s t ddgdtjgdtjggddgd} t ddgdtjgdtjggddgd}| jddgdd jd
|
d }t ||¡ | jddgdd jd
|
d }t ||¡ | jddgdd jd d }t ||¡ d S ) Nr5 rÍ r6 r_ rÖ r× r Fr r rª r rØ rÙ r0 r0 r1 Útest_head_tail_dropna_false, s && rÛ )9Únumpyr' r} Zpandasr r r r r r r Zpandas._testingZ_testingr" r2 ÚmarkZ parametrizer@ rB rI r( r\ rJ rZ r^ rd r r r r r; r Z to_timedeltar r r¦ r© r« r r² r´ rµ rº r½ r¿ rÆ rÅ rÇ rÊ rË rÌ rÎ rU rÕ ÚfilterwarningsrÚ rÛ r0 r0 r0 r1 Ú<module> s8 0
|
|
|
|
þþûþ s
|
ý
|
ý
|
ýõ
|
|
|
ýù
|
|
|
þú
|
|
þúîîþ1
|
þþ
|
üþ 4
|
|
øþ
|
|
øþ
|
|
|
|
$ üþ
|
øþ ÿ
|
|
