U
|
ý°d ã @ s d dl Zd dlZd dlmZmZmZ d dlmZ ej
|
dd Z ej ddd dd fd d d
|
d fd d d d fg¡d d Zdd Zdd Zdd Zdd Zdd Zdd Zdd Zd d Zej d d d!d"gd d#d$ggd d%d&gd d'd&gd(d)gfd&d!d*gd&d#d+ggd d%d&gd,d-d.gd/d0gfd&d'd(gd d1d2ggd d%d&gd)d)d3gd4d5gfd&d6d1gd d!d"gd'd#d$ggd d%d&gd1d(d)gd2d7d8gfg¡d9d: Zd;d< ZdS )=é N)Ú DataFrameÚ
|
MultiIndexÚSeriesc C s< t j dd¡} t| dddgdddggdddgdddggdS ) z
|
Factory function to create simple 3 x 3 dataframe with
|
both columns and row MultiIndex using supplied data or
|
random data by default.
|
é é é é é é
|
é ©ÚcolumnsÚindex)ÚnpÚrandomÚrandnr )Údata© r úad:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/indexing/multiindex/test_iloc.pyÚsimple_multiindex_dataframe s ÿr zindexer, expectedc C s
|
| j d S )Nr ©Úiloc©Údfr r r Ú<lambda> ó r c C s" t | d dddgdddggddS ) Nr r r r r r
|
)r r ©r Úname©r ©Úarrr r r r r c C s
|
| j d S ©Nr r r r r r r " r c C s" t | d dddgdddggddS )Nr r r r r
|
)r r r r r r r r r # r c C s | j d d
df S r! r r r r r r & r c C s* t | d d
df dddgdddggddS )Nr r r r
|
r )r r
|
r r r r r r r ' r c C s* |}|j }| |}||}t ||¡ d S )N)ÚvaluesÚtmÚassert_series_equal)ZindexerÚexpectedr r r Úresultr r r Útest_iloc_returns_series s
|
r' c C s0 | }|j ddg }|jddd}t ||¡ d S )Nr é r F©Z
|
drop_level)r Úxsr# Úassert_frame_equal)r r r& r% r r r Útest_iloc_returns_dataframe3 s r, c C s, | }|j }|jd }|d }||ks(td S )N©r r )r" r ÚAssertionError)r r r r& r% r r r Útest_iloc_returns_scalar: s
|
|
r/ c C sh t ddddgddddgg } t | ¡}ttj dd¡|d}|jddg }|jdd d
|
}t
|
||¡ d S ) NÚaÚbÚxÚyr ©r r r Fr) ) Úzipr Ú from_tuplesr r r r r r* r# r+ )Útupr r r& r% r r r Ú test_iloc_getitem_multiple_itemsB s
|
r8 c C sb t j dd¡} t| dddgdddggddddgddd d ggd
|
}|jd }| d }||ks^td S ) Nr r ÚiÚjÚAÚBÚkÚXÚYr r- )r r r r r r. )r r r& r% r r r Útest_iloc_getitem_labelsL s ý
|
r@ c C s. | }|j d d
}|d d
}t ||¡ d S )Nr )r r# r+ ©Ú multiindex_dataframe_random_datar r& r% r r r Útest_frame_getitem_sliceY s rC c C sJ | }d|j d d
< |jd d
dk ¡ s,t|jdd
dk ¡ sFtd S )Nr r )r r" Úallr. )rB r r r r Útest_frame_setitem_slice` s rE c C sp t dddg¡} t ddddg¡}tt d¡ d ¡|| d
|
}|jd d
d f }|jd d
df }t ||¡ d S ) N)ÚOhioÚGreen)rF ZRed)ZColoradorG )r0 r( )r0 r )r1 r( )r1 r r )r r )r r r( )
|
r r6 r r ZarangeZreshaper Úlocr# r$ )r r r r& r% r r r Ú test_indexing_ambiguity_bug_1678h s ÿrI c st ddgddgddgddgd d
|
gg} t d d d d dg¡}t| }t| |d t fddtdD }t ||¡ d S )NZstr00Zstr01Zstr10Zstr11Zstr20Zsrt21Zstr30Zstr31Zstr40Zstr41)ÚCCr; )rJ r< )ÚBBr0 )rK r1 r4 c s$ g | ] fd dt dD qS )c s g | ]} j |f qS r r )Ú.0Úc)r Úrr r Ú
|
<listcomp> s z:test_iloc_integer_locations.<locals>.<listcomp>.<listcomp>r )Úrange)rL r )rN r rO s z/test_iloc_integer_locations.<locals>.<listcomp>é )r r6 r rP r# r+ )r r r% r& r r r Útest_iloc_integer_locationsv s û ÿ rR z!data, indexes, values, expected_kr é rQ é! r éÿÿÿÿr( r é r
|
i+ i éÈ i, éd ió i* r r iè é iú é é é c C sn t | dddgd}| ddg¡}|j ¡ }t||D ]\}}|j| |7 < q4||d< |j}t ||¡ d S )Nr9 r: r= )r r )r Z set_indexr= Úcopyr5 r r# r$ ) r Zindexesr" Z
|
expected_kr Zseriesr9 Úvr% r r r Ú'test_iloc_setitem_int_multiindex_series s
|
r_ c C s. | }|j d }| |jd ¡}t ||¡ d S r! )r r* r r# r$ rA r r r Útest_getitem_iloc§ s
|
r` )Únumpyr ZpytestZpandasr r r Zpandas._testingZ_testingr# Zfixturer ÚmarkZ parametrizer' r, r/ r8 r@ rC rE rI rR r_ r` r r r r Ú<module> sJ
|
þþþ÷þ
|
|
***4øþ
|
|
