U
|
ý°dßl ã @ sz d dl m Z mZ d dlZd dlZd dlmZ d dlZd dlm Z m
|
Z
|
m Z m Z m Z d dlmZ d dlmZ G dd dZdS )é ©ÚdatetimeÚ timedeltaN)ÚUnsortedIndexError)Ú DataFrameÚIndexÚ
|
MultiIndexÚSeriesÚ Timestamp)Ú_mklblc @ s
|
e Zd Zdd Zdd Zdd Zdd Zd d
|
Zd d Zd d Z dd Z
|
dd Z dd Z dd Z dd Zdd Zej dd ed d d ed d d fd ed d d ed d d fd ed d d ed d!d fd ed d d ed d!d fd ed d d"ed d d"fd ed d d"ed d d"fd ed d d"ed d!d"fd ed d d"ed d!d"fd#ed d d ed d d fd#ed$d d ed d d fd#ed d%d ed d!d fd#ed$d%d ed d!d fd#ed d d"ed d d"fd#ed$d d"ed d d"fd#ed d%d"ed d!d"fd#ed$d%d"ed d!d"fg¡d&d' Zd S )(ÚTestMultiIndexSlicersc C s t tddtddtddtddg¡}tt t| ¡ ¡|d }|jt d
|
d t d d d gfd d
f }|jdd |j
|
j D }t ||¡ |jdd |j
|
j D }|jt d
|
d t d t d d fd d
f }t ||¡ t jddddgddgd}t jddddgd d gd}ttjd d d dd¡||d!}|jd"d#jd$d#}|jt d t d fd d
f }t ||¡ |jt d t d ft d t d ff }t ||¡ |jd d
t d t d ff }t ||¡ |jt d d$gfd d
f }|jd"d%g }t ||¡ |jt d d$fd d
f }|jd"d%g }t ||¡ |jd d
t d d&gff }|jd d
d$d%gf }t ||¡ |jt d d$ft d d&gff }|jd"d%gd$d%gf }t ||¡ |jd' }td$dd(gd"dd)gd*td$dd%gdd+td,d&gd d+d!}t ||¡ |jt d d$dgfd d
f }|jd"d$d%g }t ||¡ tt t| ¡ ¡|d }|jd
|
d
d d
d d gf }|jd-d |j
|
j D }t ||¡ |jt d |jd d
df dkfd d
f }|jdd%g }t ||¡ d.}tjt|d/* |jt d t d0d1g¡fd d
f W 5 Q R X tjtd2d/ |jt d d$gf W 5 Q R X |j
|
jdks^t|jd$d"d3}|j
|
jd"ks~td4}tjt |d/$ |jt d t d,fd d
f W 5 Q R X |jt d |jd d
df dkfd d
f }t ||jd$d%gd d
f ¡ d S )5NÚAé ÚBé ÚCé ÚDé ©ÚindexÚA1ÚA3ÚC1ÚC3c S s0 g | ](\}}}}|d kr|dkr||||fqS ©©r ÚA2r ©r r © ©Ú.0ÚaÚbÚcÚdr r úbd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/indexing/multiindex/test_slice.pyÚ
|
<listcomp>" s
|
úüzITestMultiIndexSlicers.test_per_axis_per_level_getitem.<locals>.<listcomp>c S s0 g | ](\}}}}|d kr|dkr||||fqS )r )r ZC2r r r r r r& r' 0 s
|
úü©r é ©r r ©r é ©r r) ÚoneÚtwo©Únames©r" Úfoo©r" Úbar©r# r3 ©r# ÚbahÚlvl0Úlvl1é Úint64©Údtype©r Úcolumnsr ©Úaxisr) r, r3 ©r r" é é )r5 r3 ©Únamer5 c S s0 g | ](\}}}}|d kr|dkr||||fqS r r r r r r& r' w s
|
úüzLcannot index with a boolean indexer that is not the same length as the index©ÚmatchTFz\[1\] not in index)ÚlevelrB z_MultiIndex slicing requires the index to be lexsorted: slicing on levels \[1\], lexsort depth 0) r Ú from_productr r ÚnpÚarangeÚlenZto_numpyÚlocÚslicer ÚvaluesÚtmÚassert_frame_equalÚ from_tuplesÚreshapeÚ
|
sort_indexÚilocr r Úassert_series_equalÚpytestÚraisesÚ
|
ValueErrorÚarrayÚKeyErrorZ_lexsort_depthÚAssertionErrorr ) ÚselfÚixÚdfÚresultÚexpectedr r@ ÚsÚmsgr r r& Ú test_per_axis_per_level_getitem s¬ "ÿ&ùÿ ùÿ (
|
ÿ
|
þ ÿ &
|
ý ùÿ , ÿ. ÿ(,z5TestMultiIndexSlicers.test_per_axis_per_level_getitemc C s2 t ddddgddddgddddgddddgd dd d
|
g¡ ¡ }|jjrNtt ddgddgddgddgd dd d
|
g¡ ¡ }|jtd td dfd d
f }t ||¡ |j
|
ddd d }t ||¡ t ddddgddddgddddgddddgd dd d
|
g¡ ¡ }|jjrtt ddgddgddgddgd dd d
|
g¡ ¡ }|jtd td dfd d
f }|jjrxtt ||¡ ddddd dddddddddddddddddg}t |}t dg| |g¡}tdg| |d }| ¡ }|jtd td f }tdg|d |d d
d ¡ }t ||¡ d S ) Nr3 r" r) r r, r )r r r r r r r F)rJ Z
|
drop_levelr é r rE rD é
|
é é é é r; é é é i@ r i éþÿÿÿ)r Ú set_indexrV r Z is_uniquer^ rO rP rR rS ÚxsrN r Ú from_arraysr rX )r_ ra rc rb ÚintsÚnÚidxr r r& Ú"test_multiindex_slicers_non_unique¦ s
|
|
|
|
üÿøÿ "ÿÿ
|
|
|
|
üÿøÿ "ÿÿ ê"z8TestMultiIndexSlicers.test_multiindex_slicers_non_uniquec C sv dd t dD }ddg}tj||gddgd}ttjd d
|
d d d ¡|tdd}tj }|j
|
ddd gddgf }|j t t dt dt ddft ddf }t ||¡ |j |t dt d
|dd
ft ddf }t ||¡ |j t t dt ddft ddf }t ||¡ |j t ddt ddft ddf }t ||¡ |j |dd
df|d f }t ||¡ d S )Nc S s( g | ] }t d dddddt|d qS )iÜ r) rj )Údaysr )r! Úir r r& r' ý s zNTestMultiIndexSlicers.test_multiindex_slicers_datetimelike.<locals>.<listcomp>rg r) r ÚdateZ frequencyr0 é0 r< r= éÿÿÿÿr ÚABCDr? r z2012-01-01 12:12:12z2012-01-03 12:12:12r r )r r )Úranger rK r rL rM rU ÚlistÚpdÚ
|
IndexSlicerW rO rP r
|
rR rS )r_ ÚdatesÚfreqr ra rv rc rb r r r& Ú$test_multiindex_slicers_datetimelikeú sj ý ÿüúÿ ÿ
|
üúÿ ÿüúÿ ÿÿ ÿz:TestMultiIndexSlicers.test_multiindex_slicers_datetimelikec C s t dgd dgd dgd dddddgd d d
|
d d d d d
|
d d d ddd d dgdddddddddddddddgd}t |d ¡|d< | dd dg¡}| ¡ }|jtdd d
f }|jd d
}t ||¡ |jtdd d
f }|}t ||¡ |jtd tddfd d
f }|jdddd d dd!d"dg }t ||¡ |jtd td td#d$fd d
f }|jddd%d d!g }t ||¡ |jtdtdfd d
f }|jd ddd%d d&g }t ||¡ |jtd tdfd d
f }|}t ||¡ |jtd tddtd fd d
f }|jdddd d dd!d"g }t ||¡ |jtd td td'd$fd d
f }|jddd%d d!g }t ||¡ d S )(NZA0r r r ZB0ZB1ZB2r, z
|
2013-06-11z
|
2013-07-02z
|
2013-07-09z
|
2013-07-30z
|
2013-08-06z
|
2013-09-03z
|
2013-10-01é é# rl rD r é( rn r r) )r r ÚDATEZVALUESr r r r rh r rE rj rk Z20130702Z20130709rg ri Z20130701)
|
r r Ú to_datetimerq rV rO rP rW rR rS )r_ ra Zdf1rb rc r r r& Ú test_multiindex_slicers_edges7 sb ñ ìÿ & & &z3TestMultiIndexSlicers.test_multiindex_slicers_edgesc C sT t j}t tddtddtddtddg¡}tjddd d
|
gd d gd }ttjt |t | dd
|
t |t |f¡||d}|j t ddt d ddgfd d
f }|j dd |j jD }t ||¡ |j |dd
d d
ddgf d d
f }t ||¡ |j t d t d ddgfd d
f }|j dd |j jD }t ||¡ |j |d d
d d
ddgf d d
f }t ||¡ d}tjt|d |j ddt dff W 5 Q R X t |j dt d dff |j d jd d
d dgf ¡ |jd d }|j dt d dff |j t d t d ddgft d dff d |j d d d d
d d
ddgf< d S ) Nr r r r r r r2 r4 r6 r7 r9 r: r0 r< r= r? r r r r c S s0 g | ](\}}}}|d kr|dkr||||fqS r r r r r r& r' s
|
úüzNTestMultiIndexSlicers.test_per_axis_per_level_doc_examples.<locals>.<listcomp>c S s( g | ] \}}}}|d kr||||fqS ©r r r r r r& r' § s
|
úüz_MultiIndex slicing requires the index to be lexsorted: slicing on levels \[1\], lexsort depth 1rH r" r3 r r) rA iöÿÿÿ)r r r rK r rT r rL rM rN rU rO rP r rQ rR rS rY rZ r rW rV )r_ rv r r@ ra rb rc re r r r& Ú$test_per_axis_per_level_doc_examples sZ "ÿ
|
þÿû&ùÿ ( $ùÿ ( ÿ ÿ (z:TestMultiIndexSlicers.test_per_axis_per_level_doc_examplesc
|
C sæ t tddtddtddtddg¡}t jddd d
|
gd d gd }ttjt|t| dd t|t|f¡||d ¡ j dd}|j
|
dddd
d d
ddgf }|j
|
dd |j j D }t ||¡ |j
|
ddd d
d d
ddgf }|j
|
dd |j j D }t ||¡ |j
|
ddd d
d f }|j
|
d d
td d ff }t ||¡ |j
|
d dd d
d f }|j
|
d d
td d ff }t ||¡ d D ]N}d | d }tjt|d!( |j
|
|dd d
d d
ddgf W 5 Q R X qd S )"Nr r r r r r r2 r4 r6 r7 r9 r: r0 r< r= r? r) rA r r r r r c S s0 g | ](\}}}}|d kr|dkr||||fqS r r r r r r& r' ã s
|
úüzATestMultiIndexSlicers.test_loc_axis_arguments.<locals>.<listcomp>r c S s( g | ] \}}}}|d kr||||fqS r r r r r r& r' ò s
|
úür3 r@ )r| r r3 zNo axis named z for object type DataFramerH )r rK r rT r rL rM rN rU rV rO r rQ rR rS rP rY rZ r[ )r_ r r@ ra rb rc ry re r r r& Útest_loc_axis_argumentsÌ sR "ÿ
|
þÿû
|
øÿ "ùÿ "ùÿ z-TestMultiIndexSlicers.test_loc_axis_argumentsc C sj t t d¡ dd¡t dddgddd gg¡d
|
}|jd d dd
}|jd d
d d
f }t ||¡ d S )Né r, rD Úa1Úa2Úa3Úb1Úb2Úb3©r@ r) rA éýÿÿÿ)
|
r rL rM rU r rK rO rW rR rS ©r_ ra rb rc r r r& Ú?test_loc_axis_single_level_multi_col_indexing_multiindex_col_df s þzUTestMultiIndexSlicers.test_loc_axis_single_level_multi_col_indexing_multiindex_col_dfc C sr t t d¡ dd¡t dddgddd gg¡d
|
}|jd d d }|jd d
d d
f }ddd g|_t
|
||¡ d S ) Nr r, rD r r r r r r r r) rA ) r rL rM rU r rK rO rW r@ rR rS r r r r& Ú@test_loc_axis_single_level_single_col_indexing_multiindex_col_df s þ zVTestMultiIndexSlicers.test_loc_axis_single_level_single_col_indexing_multiindex_col_dfc C sX t t d¡ dd¡dddgd}|jddd }tt d dd
|
g¡dd }t ||¡ d S ) NrD r, r" r# r$ r r) rA r rg rF ) r rL rM rU rO r r\ rR rX r r r r& Ú*test_loc_ax_single_level_indexer_simple_df% s z@TestMultiIndexSlicers.test_loc_ax_single_level_indexer_simple_dfc C s t j}tjddddgddgd}tjdd d
|
d gd d gd}ttjddd dd¡||d}|jddjdd}| ¡ }d|j
|
t d t d fd d
f< | ¡ }d|j d d
d d
f< t ||¡ | ¡ }d|j
|
ddd d
d d
f< | ¡ }d|j d d
d d
f< t ||¡ | ¡ }d|j
|
t d t d ft d t d ff< | ¡ }d|j d d
d d
f< t ||¡ | ¡ }d|j
|
d d
t d t d ff< | ¡ }d|j d d
d d
f< t ||¡ | ¡ }d|j
|
t d dgfd d
f< | ¡ }d|j ddg< t ||¡ | ¡ }d|j
|
t d dfd d
f< | ¡ }d|j ddg< t ||¡ | ¡ }d|j
|
ddd d
df< | ¡ }d|j ddg< t ||¡ | ¡ }d|j
|
d d
t d dgff< | ¡ }d|j d d
ddgf< t ||¡ | ¡ }d|j
|
t d dft d dgff< | ¡ }d|j ddgddgf< t ||¡ | ¡ }d|j
|
|d d
df |d d
dgf f< | ¡ }d|j ddgddgf< t ||¡ | ¡ }d|j
|
d< | ¡ }d|j dd
dd
f< t ||¡ | ¡ }tjddgddggdd|j
|
t d dft d dgff< | ¡ }d|j ddgddgf< t ||¡ | ¡ }d}tjt|d : tjdgddggdd|j
|
t d dft d dgff< W 5 Q R X d }tjt|d 8 tjddddgdd|j
|
t d dft d dgff< W 5 Q R X | ¡ }|j
|
t d dft d dgff d |j
|
t d dft d dgff< | ¡ }|j ddgddgf d |j ddgddgf< t ||¡ | ¡ }|j
|
t d dft d dgff |j
|
t d dft d dgff 9 < | ¡ }|j ddgddgf |j ddgddgf 9 < t ||¡ |j
|
t d dft d dgff ¡ }d |j
|
d d
d f< | ¡ }|j
|
t d dft d dgff |9 < | ¡ }|j ddgddgf |j ddgddgf 9 < t ||¡ d S )!Nr( r* r+ r- r. r/ r0 r2 r4 r6 r7 r9 r: r; r< r= r r? r rA r) éd r, r3 rC r z)setting an array element with a sequence.rH z@Must have equal len keys and value when setting with an iterabler rh )r$ r8 )r r r rT r rL rM rU rV ÚcopyrO rP rW rR rS r\ rY rZ r[ ) r_ rv r r@ Zdf_origra rc re Úrhsr r r& Ú test_per_axis_per_level_setitem- sÞ
|
ÿ
|
þ ÿ & (
|
ÿ" ÿ,
|
ÿ,"ÿ , $ÿ0 $(0z5TestMultiIndexSlicers.test_per_axis_per_level_setitemc C sæ t t d¡t tdt d¡g¡}tj}t ||d d d
|d d d
¡ t ||dd d
|dd d
¡ t ||dd d
|dd d
¡ t ||d dd
|d dd
¡ t ||d dd
|d dd
¡ t ||dd d
|dd
|
d
¡ t ||dd d
|dd
|
d
¡ t ||dd d
|dd
|
d
¡ t ||dd d
|dd
|
d
¡ t ||d dd
|d d
¡ t ||d d d
|dd d
¡ t ||d d d
|d dd
¡ t ||dd d
|ddd
¡ d S )Né Úabcder r| r% é )r% ri r# r, )r# r )r$ r rh rD )Úer r; )
|
r rL rM r rK r r r rR Z!assert_indexing_slices_equivalent)r_ ZserÚSLCr r r& Ú0test_multiindex_label_slicing_with_negative_step¶ s, ÿ""""""""" "" ÿzFTestMultiIndexSlicers.test_multiindex_label_slicing_with_negative_stepc C sÊ ddddg}t |tdg¡}tttd|dgd}|jtjd d
d d
|
f d d
f }|jd }tttd d dgtd d d }t ||¡ |jd }tttd ddgtd d d }t ||¡ d S )Nr" r# r$ r% iô iÐ ZTestr? é éF éG )r@ r iú i# )
|
r rK r~ r r rO r r rR rS )r_ r rv ra Zdf_slicerb rc r r r& Ú!test_multiindex_slice_first_levelÐ s $
|
|
ÿz7TestMultiIndexSlicers.test_multiindex_slice_first_levelc C s |}|d }|dd
}| |jdd
¡}t ||¡ |d ¡ }d|dd
< d|jdd
< t |j|j¡ |dd
}| |jdd
¡}t ||¡ d S )Nr r r ) Zreindexr rR rX r rW Zassert_numpy_array_equalrQ rS )r_ Z/multiindex_year_month_day_dataframe_random_dataZymdrd rb rc Úexpr r r& Útest_int_series_slicingß s z-TestMultiIndexSlicers.test_int_series_slicingzdtype, loc, ilocÚintNr| r, r) r rp Ústrr% r# c C sÈ t dtdd| }t |gd ¡}td|dgd}tj}|j|d d
f }|j|| d d
f } |j||d d
f d d
f }
|
|j|d d
|f d d
f } t
|
| |¡ t
|
|
|
|¡ t
|
| |¡ d S )Nr r )r¬ r« r g ð?r r? ) r r~ r rs r r r rW rO rR rS ) r_ r> rO rW ÚlabelsÚmira r£ rc Zresult_get_locZresult_get_locs_level_0Zresult_get_locs_level_1r r r& Ú test_loc_slice_negative_stepsizeï s þý z6TestMultiIndexSlicers.test_loc_slice_negative_stepsize)Ú__name__Ú
|
__module__Ú __qualname__rf rw r r r r r r r r r¤ r¨ rª rY ÚmarkZ parametrizerP r¯ r r r r& r sH T=JKB
|
ìþr )r r ÚnumpyrL rY Z pandas.errorsr Zpandasr r r r r r
|
Zpandas._testingZ_testingrR Z pandas.tests.indexing.commonr r r r r r& Ú<module> s
|
