U
|
ý°d; ã @ sÈ d dl Z d dlZd dlZd dlmZmZmZmZm Z m
|
Z
|
m Z m Z m Z mZmZmZmZmZ d dlmZ d dlmZ G dd dZG dd dZG dd d ZG d
|
d d ZG d d d Z G dd dZ dS )é N)Ú CategoricalÚCategoricalDtypeÚCategoricalIndexÚ DataFrameÚ
|
DateOffsetÚ DatetimeIndexÚIndexÚ
|
MultiIndexÚSeriesÚ TimestampÚconcatÚ
|
date_rangeÚ get_dummiesÚ period_range)Ú SparseArrayc @ s4 e Zd Zdd Zdd Zdd Zdd Zd d
|
Zd S ) Ú TestGetitemc C s^ t dddgddggddgddggd}td td
|
|d }tjtdd |d W 5 Q R X d S ) NZa_lotZonlyoneZ
|
notevenonei² Ú é r )ÚlevelsÚcodeséÿÿÿÿé )ÚindexÚcolumns©Úmatch)r r ÚrangeÚpytestÚraisesÚKeyError)ÚselfÚmiÚdf© r# ú_d:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/frame/indexing/test_getitem.pyÚ test_getitem_unused_level_raises s þz,TestGetitem.test_getitem_unused_level_raisesc C sv t ddd}ttj dd¡|d}||d }t ||jd d
df ¡ t| |d }t ||jd d
df ¡ d S )Nz1/1/2000é )Úperiodsé
|
©r r ) r r ÚnpÚrandomÚrandnÚtmÚassert_series_equalÚilocÚrepr)r Úrngr" Útsr# r# r$ Útest_getitem_periodindex& s z$TestGetitem.test_getitem_periodindexc C sZ t tdtdg}tddgddggdddg|d}t|}|t|j }t ||¡ d S )Nz
|
12-31-1999z
|
12-31-2000r r Úbool)Údtyper r )r r r r Úlistr r- Úassert_frame_equal)r ZcatsÚexpectedZdummiesÚresultr# r# r$ Ú1test_getitem_list_of_labels_categoricalindex_cols3 s
|
z=TestGetitem.test_getitem_list_of_labels_categoricalindex_colsc C sx t ddg}td|i}t|dd}|d }t ||¡ |jd d
df }t ||¡ |jd d
df }t ||¡ d S )Nr r ÚA©Úname)r r r
|
r- r. r/ Úloc)r Údatar" r8 r9 r# r# r$ Ú0test_getitem_sparse_column_return_type_and_dtype< s z<TestGetitem.test_getitem_sparse_column_return_type_and_dtypec C s< t ddggtddgddd}|j}|d }t ||¡ d S )Nr é r; ÚBÚstring©r5 r) )r r r; r- r. ©r r" r9 r8 r# r# r$ Útest_getitem_string_columnsK s z'TestGetitem.test_getitem_string_columnsN)Ú__name__Ú
|
__module__Ú __qualname__r% r3 r: r@ rF r# r# r# r$ r s
|
r c @ s e Zd Zdd Zdd Zdd Zejjde e
|
e e dd d
|
d gd d d dddgdej dddg¡dd Z dd Zdd Zdd Zd S ) ÚTestGetitemListLikec C sN t dgdgdgd}dddg|_tjtdd |dd dg W 5 Q R X d S )
|
Ng ð?g @ç @)ÚxÚyÚzrL rN z\['y'\] not in indexr rM )r r r r r ©r r" r# r# r$ Ú test_getitem_list_missing_keyT s z1TestGetitemListLike.test_getitem_list_missing_keyc C sd t tj dd¡tdd}d|j_|ddg }|jjdks>t|jd d
dd
f }t
|
||¡ d S )Né ZAABCr) ÚfoorB ÚCrA ) r r* r+ r, r6 r r= ÚAssertionErrorr/ r- r7 rE r# r# r$ Ú test_getitem_list_duplicates^ s z0TestGetitemListLike.test_getitem_list_duplicatesc C sT t dddgdddggdddgd }d
|
}tjtt |¡d |d g W 5 Q R X d S ) Nr rA r rQ r& é ÚaÚbr) z?"None of [Index(['baf'], dtype='object')] are in the [columns]"r Zbaf)r r r r ÚreÚescape)r r" Úmsgr# r# r$ Útest_getitem_dupe_colsi s "z*TestGetitemListLike.test_getitem_dupe_colsÚidx_typec C s t t| tt| S ©N)ÚdictÚzipr Úlen©Úkeysr# r# r$ Ú<lambda>v ó z TestGetitemListLike.<lambda>c C s t t| tt| ¡ S r^ )r_ r` r ra rc rb r# r# r$ rd w re r6 Úiterr Úsetr_ Ú dict_keys)Úidsr r rA c C sþ |dkr|d }}n(t tj dd¡tdddgdd d
|
}d }|jd |jd g}||}t||}t|tt frt
|
j t d d || W 5 Q R X d S || } |j d d
|f }
|
|jj|
|
j_t | |
|
¡ |||g }t
|
j tdd || W 5 Q R X d S )Nr Úfoodé r )rR Úbar)ZbazZqux)ÚpeekZaboo)ZsthZsth2r< r) )Zgoodrj r ú as an indexer is not supportedr z not in index)r r* r+ r, r r r6 Ú
|
isinstancerg r_ r r Ú TypeErrorr> Únamesr- r7 r ) r r] r Ú float_frameÚframeÚmissingrc ÚidxZ idx_checkr9 r8 r# r# r$ Útest_getitem_listlikeo s0 þþ z)TestGetitemListLike.test_getitem_listlikec C s` t dddgdddgd}dd d
|
D }|j| }t ddgddgdddgd }t ||¡ d S ) Nr rA r rQ r& rV ©rW rX c s s | ]
|
}|V qd S r^ r# ©Ú.0rL r# r# r$ Ú <genexpr>¤ s zBTestGetitemListLike.test_getitem_iloc_generator.<locals>.<genexpr>©r rA ©r )r r/ r- r7 ©r r" Zindexerr9 r8 r# r# r$ Útest_getitem_iloc_generator¡ s
|
|
z/TestGetitemListLike.test_getitem_iloc_generatorc C s\ t dddgdddgd}dd d
|
D }|j|df }tddgd ddgd }t ||¡ d S ) Nr rA r rQ r& rV rw c s s | ]
|
}|V qd S r^ r# rx r# r# r$ rz « s zRTestGetitemListLike.test_getitem_iloc_two_dimensional_generator.<locals>.<genexpr>r{ rX )r= r )r r/ r
|
r- r. r} r# r# r$ Ú+test_getitem_iloc_two_dimensional_generator© s
|
z?TestGetitemListLike.test_getitem_iloc_two_dimensional_generatorc C s t ttdtddtdddd}|jdd
}t d dd
|
gtdd dgd tddd d}t ||¡ t ttdtddtdd
|
ddd}|jdd
}t d dd
|
gtdddgd tdd
|
dd d}t ||¡ t ttdtddtdddd}|jdd
}t ||¡ d S )Nr( z
|
01-01-2022r )Údays)r' Úfreqr| z
|
2022-01-01z
|
2022-01-03r rA z
|
2022-01-02zdatetime64[ns])r5 r )r Úhoursz2022-01-01 00:00:00z2022-01-02 02:00:00z2022-01-03 04:00:00r )Úminutes) r r6 r r r r> r r- r7 rE r# r# r$ Ú!test_getitem_iloc_dateoffset_days° sL
|
þýþ
|
|
ÿþ
|
ýþ
|
þz5TestGetitemListLike.test_getitem_iloc_dateoffset_daysN)rG rH rI rP rU r\ r ÚmarkÚ parametrizer6 rf r rg rv r~ r r r# r# r# r$ rJ S s&
|
úö %rJ c @ s e Zd Zdd Zdd ZdS )ÚTestGetitemCallablec C s¨ |dd }|j d d
df }t ||¡ |dd }|j d d
ddgf }t ||j d d
ddgf ¡ |d d
}|dd }|jdd gd d
f }t ||¡ d S )
|
Nc S s dS )Nr; r# ©rL r# r# r$ rd Ý re z;TestGetitemCallable.test_getitem_callable.<locals>.<lambda>r; c S s ddgS )Nr; rB r# r r# r# r$ rd á re rB r c S s
|
dddgS )NTFr# r r# r# r$ rd æ re r rA )r> r- r. r7 r/ )r rr r9 r8 r" r# r# r$ Útest_getitem_callableÛ s z)TestGetitemCallable.test_getitem_callablec C s` t ddggddggd}t dgdggd}|d }t ||¡ |jd d
df }t ||¡ d S )Nr rA rW rX r) )r r- r7 r> )r r" r8 r9 r# r# r$ Ú%test_loc_multiindex_columns_one_levelê s z9TestGetitemCallable.test_loc_multiindex_columns_one_levelN)rG rH rI r r r# r# r# r$ r Ú s r c @ sâ e Zd Zdd Zej dddgddggdd gd
|
d ggejd gejd gejdgddggfddgddggdd gd
|
d ggejd gejd gdd
|
gd d ggff¡dd Z ej
|
dd Z dd Z dd Z dd Zd d Zd d Zd d! Zd"S )#ÚTestGetitemBooleanMaskc C sº t dtjddditddddddgtdddgdd d
|
d d }t dtjddditddddddgtdddgd d d
|
d d }||jdk }|jg }t ||¡ ||jdk }|jg }t ||¡ ||jdk }|jdddg }t ||¡ ||jdk }|jdddg }t ||¡ ||jdk }|jdg }t ||¡ ||jdk }|jg }t ||¡ d}t j
|
t |d ||jdk W 5 Q R X t j
|
t |d ||jdk W 5 Q R X d S )Nr; rV Úint64rD r rA r T)ZorderedrB )r5 r= r| FrW r rQ z7Unordered Categoricals can only compare equality or notr ) r r* Úaranger r r r/ r- r7 r r rp )r Zdf3Zdf4r9 r8 r[ r# r# r$ Ú(test_getitem_bool_mask_categorical_index÷ sV ÿýü
|
ÿýü
|
|
|
z?TestGetitemBooleanMask.test_getitem_bool_mask_categorical_indexzdata1,data2,expected_datar rA r rQ g à?rV é rk rK g @g @g @g @r& c
|
C sp t t |¡}t t |¡}t||gdd}||dk }dd t|D }t |jdddd} t || ¡ d S ) Nr )ZaxisrA c S s i | ]\}}|t |¡qS r# )r* Úarray)ry ÚiÚcolr# r# r$ Ú
|
<dictcomp>U s z`TestGetitemBooleanMask.test_getitem_bool_mask_duplicate_columns_mixed_dtypes.<locals>.<dictcomp>r )rA r r) )r r* r r Ú enumerateÚrenamer- r7 )
|
r Zdata1Zdata2Z expected_dataZdf1Údf2r" r9 Zexdictr8 r# r# r$ Ú5test_getitem_bool_mask_duplicate_columns_mixed_dtypes8 s zLTestGetitemBooleanMask.test_getitem_bool_mask_duplicate_columns_mixed_dtypesc C s, ddddg}t t d¡ dd¡|dd}|S ) Nr; rS ÚDé r rQ Úfloat64©r r5 )r r* r Úreshape)r Zdupsr" r# r# r$ Ú df_dup_colsY s z"TestGetitemBooleanMask.df_dup_colsc C s4 |}d}t jt|d ||jdk W 5 Q R X d S )Nz/cannot reindex on an axis with duplicate labelsr rV )r r Ú
|
ValueErrorr; )r r r" r[ r# r# r$ Ú;test_getitem_boolean_frame_unaligned_with_duplicate_columns_ s zRTestGetitemBooleanMask.test_getitem_boolean_frame_unaligned_with_duplicate_columnsc C sj t t d¡ dd¡ddddgdd }||jd
|
k }|j|_|}||jd
|
k }t ||¡ |j t | d S © Nr r rQ r; rB rS r r r rV )
|
r r* r r rS r r- r7 ÚdtypesÚstr©r r r" r8 r9 r# r# r$ Ú2test_getitem_boolean_series_with_duplicate_columnsh s
|
ÿ zITestGetitemBooleanMask.test_getitem_boolean_series_with_duplicate_columnsc C sf t t d¡ dd¡ddddgdd }||d
|
k }|j|_|}||d
|
k }t ||¡ |j t| d S r ) r r* r r r r- r7 r¡ r¢ r£ r# r# r$ Ú1test_getitem_boolean_frame_with_duplicate_columnsx s
|
ÿ zHTestGetitemBooleanMask.test_getitem_boolean_frame_with_duplicate_columnsc C s" t }||dk }t ||¡ d S )Nr )r r- r7 )r r" r r# r# r$ Ú%test_getitem_empty_frame_with_boolean s z<TestGetitemBooleanMask.test_getitem_empty_frame_with_booleanc C s| t dddgdddggdddgd }| ¡ }|d }d
|
|jd d
< |rJ|}n"t ddd
|
gddd
|
ggdddgd }t ||¡ d S ) Nr rA r rQ r& rV rW rX r) éd )r Úcopyr> r- r7 )r Zusing_copy_on_writer" Zdf_origÚviewr8 r# r# r$ Ú5test_getitem_returns_view_when_column_is_unique_in_df s ""zLTestGetitemBooleanMask.test_getitem_returns_view_when_column_is_unique_in_dfc C sV t ddddggtdgdddgd}|tdg }tdgtdgd }t ||¡ d S )
|
Nr rA r rQ ZKEYrB rS r) r< )r Ú frozensetr
|
r- r. rE r# r# r$ Ú'test_getitem_frozenset_unique_in_column s $z>TestGetitemBooleanMask.test_getitem_frozenset_unique_in_columnN)rG rH rI r r r
r r* Únanr Zfixturer r r¤ r¥ r¦ rª r¬ r# r# r# r$ r ö s, A ý ýúþ
|
|
r c @ s e Zd Zdd Zdd ZdS )ÚTestGetitemSlicec
|
C s t ddd¡}t|}|ddg \}}t j dd¡}|tk rL|d d
df }|||d }|||
}|jdd
|
} t || ¡ |j ||
}t || ¡ d S ) Ng $@g I@rA r& é é r r r| é )
|
r* r r r+ r, r r/ r- Z assert_equalr> )
|
r Zframe_or_seriesÚvaluesr ÚstartÚendr? Úobjr9 r8 r# r# r$ Útest_getitem_slice_float64¦ s z+TestGetitemSlice.test_getitem_slice_float64c C sF t dditddddgd}tjtdd |d
|
d
W 5 Q R X d S ) NrW r z11.01.2011 22:00z11.01.2011 23:00z12.01.2011 00:00z2011-01-13 00:00r| z,Value based partial slicing on non-monotonicr z
|
2011-01-01z
|
2011-11-01)r r r r r rO r# r# r$ Útest_getitem_datetime_slice¹ s üÿþ ÿz,TestGetitemSlice.test_getitem_datetime_sliceN)rG rH rI r¶ r· r# r# r# r$ r® ¥ s r® c @ s. e Zd Zej dddhddig¡dd ZdS )Ú TestGetitemDeprecatedIndexersÚkeyrW rX c C sH t ddgddggt ddg¡d}tjtdd || W 5 Q R X d S )
|
Nr rA r rQ )rW r )rX rA r) rn r )r r Ú from_tuplesr r rp )r r¹ r" r# r# r$ Ú$test_getitem_dict_and_set_deprecatedÍ s ÿzBTestGetitemDeprecatedIndexers.test_getitem_dict_and_set_deprecatedN)rG rH rI r r
r r» r# r# r# r$ r¸ Ì s r¸ ) rY Únumpyr* r Zpandasr r r r r r r r r
|
r r r r r Zpandas._testingZ_testingr- Zpandas.core.arraysr r rJ r r r® r¸ r# r# r# r$ Ú<module> s @ 9 0'
|
