a 鴫j@s<ddlZddlZddlmZmZddlZddlmZm Z m Z m Z ddl m ZejddZejddZGdd d Zejd ggfd ged fd d ggfd dggfd ggfejdddgfeddd fejddddgfeddd fgddZejdgedfdggfgddZddZddZddZddZdd Zd!d"Zd#d$Zd%d&Z d'd(Z!d)d*Z"d+d,Z#d-d.Z$d/d0Z%d1d2Z&Gd3d4d4Z'd5d6Z(d7d8Z)d9d:Z*d;d<Z+d=d>Z,d?d@Z-ejdAgdBgdCgdDdEZ.dFdGZ/dHdIZ0dJdKZ1dLdMZ2dNdOZ3dPdQZ4dRdSZ5dTdUZ6dVdWZ7dS)XN) IndexingErrorPerformanceWarning) DataFrameIndex MultiIndexSeriescCstgdggdgdgdS)zsingle level MultiIndex)foobarbazZqux)rfirst)levelscodesnames)rrrh/var/www/html/laksyashrimp/venv/lib/python3.9/site-packages/pandas/tests/indexing/multiindex/test_loc.pysingle_level_multiindexsrcCsFddggdg}gdgdg}t||d}ttjdd|d S) Nrr rr r )rrrr r r )rr r rr r )rrr )r index)rrnprandom default_rngstandard_normal)rrrrrr%frame_random_data_integer_multi_indexs rc @sveZdZejdejdddgddZddZd d Z d d Z d dZ ddZ ddZ ddZddZejdddgddgfdggfddggfgddZdd Zd!d"Zd#d$Zd%d&Zd'd(Zejd)eeeeejeefejd*eeeeejeefd+d,Zd-d.Zejd/d0d1gd2d3Z ejd4ed5d6fd7ej!fgd8d9Z"d:d;Z#dd?Z%d5S)@TestMultiIndexLocz.ignore:Setting a value on a view:FutureWarninghas_refTFcCsv|}|r|dd}d|jd<|jddks0J|}ttd|_|rV|dd}d|jd<|jddksrJdS)N)r twoBr )r!r )loccopylistrangecolumns)self multiindex_dataframe_random_datarframeviewdfrrr&test_loc_setitem_frame_with_multiindex$s    z8TestMultiIndexLoc.test_loc_setitem_frame_with_multiindexc Cs*|}ddddddddddddd d d d d dd }t|}|||d }|jd dgd}d}tt,t|j||jddWdn1s0Y|j dd}|dusJ|j|}t j dgd|d}t j d gd|d} t j || gd dgd} tdgdi| d} t|| dS)NiiXM)rr r r g @@ )amountcolnum)r8r9r8r9)keys)r4r5r T)Zinplacer dtyperr7)r0r1r2r)rZastype set_indextmZassert_produces_warningrassert_frame_equalr%iloc sort_indexrarrayr from_arrays) r*Zany_real_numpy_dtyper<datar.keyZ return_valueresZcol_arrZyear_arrrexpectedrrrtest_loc_getitem_general7s& :   z*TestMultiIndexLoc.test_loc_getitem_generalcCspttjddgdgdggdgdgd}tjtdd |jdWdn1sb0YdS Nr r r r r r3r r3r3rNrNrOr5r)rz^2$match rrrrrpytestraisesKeyErrorr%r*r.rrr0test_loc_getitem_multiindex_missing_label_raisesTszBTestMultiIndexLoc.test_loc_getitem_multiindex_missing_label_raisescCs:|d}||jdd}|jddg}t||dS)NA13)r rO)r^r )reindexrr%r?assert_series_equal)r*/multiindex_year_month_day_dataframe_random_dataserrHresultrrr/test_loc_getitem_list_of_tuples_with_multiindex_szATestMultiIndexLoc.test_loc_getitem_list_of_tuples_with_multiindexcCstgdgdg}t|tdtjd}tddg}tgdtddggdgtjd}|j|}t|||jddg}t||tddgdd gd }|j|}t||tgtjd }tgt|j gggtjd tjd }|j|}t||dS)Nr r r r[r#C rrEr<r r rr r rr$rNrErr<r r)rEr<rrr<rr<) r from_productrr(rfloat64r%r?rar)r*rxyrHrdy1emptyrrrtest_loc_getitem_seriesgs.       z)TestMultiIndexLoc.test_loc_getitem_seriescCstgdgdg}t|tdtjd}tddg}tgdtddggdgtjd}|j|}t ||tg}tgt|j gggtjd d d }|j|}t ||t d}tgd gdtjd}|j|}t ||dS) Nrfrgrirjr r rkrlrmrprnr) rrorr(rrprCr%r?rarint64)r*rrqrrrHrdrtZscalarrrrtest_loc_getitem_arrays,       z(TestMultiIndexLoc.test_loc_getitem_arraycCsttjddgdgdggdgdgd}|jddgd}|jd }t |||jdddgfjddd }|jddd f}t |||jdgdgfdjddd }|jd jddd f}t |||jddg}|jd }t ||dS) Nr rK)irxj)r[r[r#)XrzYrRrr rxaxisry)rxrz rrrrrrA droplevelr%r?r@r*r.rHrdrrrtest_loc_multiindex_labelss"   "  z,TestMultiIndexLoc.test_loc_multiindex_labelscCsbttjddgdgdggdgdgd}|jdd gd}|jd }t ||dS) Nr rKrLrMrPrQrRrr r3r~rrrrtest_loc_multiindex_intss z*TestMultiIndexLoc.test_loc_multiindex_intscCspttjddgdgdggdgdgd}tjtdd |jdWdn1sb0YdSrJrUrYrrr(test_loc_multiindex_missing_label_raisessz:TestMultiIndexLoc.test_loc_multiindex_missing_label_raiseszkey, posr r3rr r cCspttjddgdgdggdgdgd}tjtdd |j|Wdn1sb0YdS) Nr rKrLrMrPrQrRz not in indexrSrU)r*rFposr.rrr&test_loc_multiindex_list_missing_labelsz8TestMultiIndexLoc.test_loc_multiindex_list_missing_labelcCsttdtddgddgddggd}tjtd d |jd Wdn1sV0Ytjtd d |jd Wdn1s0Ytjtdd |jdWdn1s0YdS)NrNabcdefrz^\('a', 'b'\)$rSrrz^\('a', 'd', 'g'\)$)rrgzToo many indexers)rrrry) rr(rrorVrWrXr%r)r*srrr(test_loc_multiindex_too_many_dims_raisess((z:TestMultiIndexLoc.test_loc_multiindex_too_many_dims_raisescCsddtdD}ddtdD}t||g}dtjddd }t||d }||}t ||tt d  d dtgd gdgd}|j ddgfddf}|j ddg}t ||dS)NcSsg|]}dt|qS) Attributestr.0rxrrr zFTestMultiIndexLoc.test_loc_multiindex_indexer_none..r cSsg|]}dt|qS)Valuerrrrrrrr g?r )rOr g?r)r5r r r r3rfr) r(rrorrrrrr?r@arangereshaper%)r* attributesZattribute_valuesrr.rdrHrrr test_loc_multiindex_indexer_nones  z2TestMultiIndexLoc.test_loc_multiindex_indexer_nonecCsttjdddttdgdg}|jddddf}|jdd ddf}t|||jdd ddf}t|||jddddf}t||ttjdddttdgdg}|j gd }|jd d d ddf}t||dS) Nrvr;r rrrrrrr3)rr$rNr5r_r ) rrrrror(r%r?rarA)r*rrHrdrrrtest_loc_multiindex_incompletes$     z0TestMultiIndexLoc.test_loc_multiindex_incompletecCs:|}ttjdt||d}|jD] }||q(dS)Nr r)rrrrrlenvalues)r*rZ single_levelrkrrrtest_get_loc_single_level s z+TestMultiIndexLoc.test_get_loc_single_levelcCstgdddgg}ttjdd||}|jddddf}|}t ||tgdddgg}ttjdd||}|jd d ddf}|j dd}t |||jd ddf}|j d d}ddg|_ t |||jddd f}|d }t ||dS) N)rr$rNrrr )rrrrNrOrrrOr) rrorrrrrr%r?r@rAr)r*rr.rdrHrrrtest_loc_getitem_int_slice)s"    z,TestMultiIndexLoc.test_loc_getitem_int_sliceindexer_type_1indexer_type_2c sddgd}gd}t||g}ttjt|dd|dgd}d d gd d gf}||f}tfd dt||D} |tus|turt j t dd|j | dfWdn1s0YdS|j | df} t gddt|d} t| | dS)NcSs,|tjkrt|S|tkr$t|S||SN)rndarrayrCslice) indexer_typer:rrrconvert_nested_indexerNs   zQTestMultiIndexLoc.test_loc_getitem_nested_indexer..convert_nested_indexerrrfrvr;ZDatarr)rOrr r c3s|]\}}||VqdSrr)rrrrrr aszDTestMultiIndexLoc.test_loc_getitem_nested_indexer..zas an indexer is not supportedrS)r r r3r namer)rrorrrrtuplezipsetrVrW TypeErrorr%rr?ra) r*rrrrrr.r:typesindexerrdrHrrrtest_loc_getitem_nested_indexerCs*  ,z1TestMultiIndexLoc.test_loc_getitem_nested_indexercCsHtddg}|ddg|d}d|jd<|ddg|d}t||dS)Nrr[rr[r r rrr)r from_tuplesr%r?Z assert_equal)r*Zframe_or_seriesmiobjrHrrr)test_multiindex_loc_one_dimensional_tuplers  z;TestMultiIndexLoc.test_multiindex_loc_one_dimensional_tuplerrrcCsPtddg}tddg|d}d|j|ddf<tddg|d}t||dS)Nrrr r rrrrrr%r?r@)r*rrrrHrrr-test_multiindex_one_dimensional_tuple_columnszs z?TestMultiIndexLoc.test_multiindex_one_dimensional_tuple_columnszindexer, exp_valueN?r r cCsptddg}tddgddgg|ddgd }d |j|d d gf<tgd dd||gg|gdd }t||dS)Nrr r3r r r r3rrrrrr)r r rrrrrrr)r*rZ exp_valuerr.rHrrr)test_multiindex_setitem_columns_enlargingsz;TestMultiIndexLoc.test_multiindex_setitem_columns_enlargingcCsttjdddtddgg}td|d}td|ddd}tj||gdd}|}|jd d }t ||t ||d }|jd d }t ||tj|| |j gdd}|jd d }t ||dS) NZ20110101r periodsrrr rr|z 2011-01-01z 2011-01-02rr )rropd date_rangerrconcatr&r%r?r@rr`r)r*ZmidxZser1ser2r.rHrdrrr"test_sorted_multiindex_after_unions   z4TestMultiIndexLoc.test_sorted_multiindex_after_unioncCsttttdtdtdgdgd}|jtjdddddf}tttdtdgdgd}t||dS)NabcdrZValrr) rrror'r%rZs_r?r@)r*r.rGrHrrrtest_loc_no_second_level_indexsz0TestMultiIndexLoc.test_loc_no_second_level_indexcCs^tgdgdgdgdd}tjtdd|jdWdn1sP0YdS) Nr)rrr)rhrx)ryrl)r)r r r r )r r3z(1, 4)rS)rr r3)rrVrWrXr%rYrrrtest_loc_multi_index_key_errorsz0TestMultiIndexLoc.test_loc_multi_index_key_error)&__name__ __module__ __qualname__rVmarkfilterwarnings parametrizer/rIrZrerurwrrrrrrrrrr'rrrrrrrrrrnanrrrrrrrrr#sF     ,   )   rz indexer, posr[r DErr bahcCstjgdgdgddgd}ttjddd|d }|j|}|jd kr|gkrtj t t |d |j |Wdq1s0Ynd|t dd d gfkrtj t dd |j |Wdq1s0Yn|j |}t||dS)Nrgrr r oner"r=rirvr;rrrSrrz'bah')rrorrrrBrAsizerVrWrXrr%rr?ra)rridxrcrHrdrrr7test_loc_getitem_duplicates_multiindex_missing_indexerss ** rcolumns_indexercCstgdddgf}ttjddtd|d}|jdd d }ttd| gdd}|j dd|f}t ||dS) Nralphabetar r rr rrr )levelr}) rrorrrrrr(rBr`r%r?r@)rZ multi_indexr.rHrdrrr4test_loc_getitem_duplicates_multiindex_empty_indexersrcCsLttjtjgddggtddgddgd}|jd}tj}||ksHJdS) Nmeanmedianfunctsr)rrfunctionrrR)rr)rrrrrrr%r.rdrHrrr=test_loc_getitem_duplicates_multiindex_non_scalar_type_objects  rcCspttdtdtjddtjdddddg}|jd}|jdddf}t ||dS)NrOr rrr)rr) rrrrrrr>r%r?ra)r.rHrdrrr!test_loc_getitem_tuple_plus_slices rcCs8|}|jd}|dd}|jd|_t||dS)Nr r)r%rrr?r@)rr.rdrHrrrtest_loc_getitem_ints   rcCs@|}tjtdd|jdWdn1s20YdS)Nz^3$rSr )rVrWrXr%)rr.rrr%test_loc_getitem_int_raises_exceptionsrcCsh|}tjtdd|jdWdn1s20Yd|jd<d}|jd}||ksdJdS)Nz^\('bar', 'three'\)$rS))r threer#r)rVrWrXr%rB)r+r.rHrdrrr test_loc_getitem_lowerdim_corners( rcCstdtdtdgdgd}d|d<|j|jdddf<tt |dt |jttj t dt gd d}| }td|jddd f<td|jddd f<t||dS) NstringabcdZMain)Zanotherrrrlabelsr3))r[1)r[2)r#rr#r )rr'rrorr%r?assert_numpy_array_equalrZasarrayrr(rr&rrAr@)r.rHrrr$test_loc_setitem_single_column_slice*s" rc Csddtjfddtjfddtjfddtjfdddd tjfdd tjfg}ttd tgd tj|gd dd}|jdjd}ttdttjg|sdndddtgd d}t ||dS)Nz Good ThingsrhRz Bad ThingsT) Okay ThingsNr#)rrrrr#r)rNr3)Zd1Zd2Zd3Zd4)u1u2u3r=rRrobjectrr )r<rr) rrrZonesrrrr%r?r@)Zusing_infer_stringZtupsr.rdrHrrrtest_loc_nan_multiindexAs2           r cCstjdddd}d}tj||fdd}t|dd }|jd }|j|d d fd f}t|s`Jt|snJ|jd }t|sJdS)N2013Q1Z2013Q4Q)freq)Wii )ZPeriodZCVRr=) OMSZOMKZRESZ DRIFT_INDZ OEVRIG_INDZFIN_INDZVARE_UDZLOEN_UDZFIN_UDr))rrrrrr)rZ period_rangerrorr%risnan)rrxrr.rdZaltrrrtest_loc_period_string_indexing_s  rcCstddg}tj||gddgd}tddgdd gd d gd d gg|d dgd}|j|d|jddkfd f}tdgd tj t dt dfgddgdd}t ||dS)Nz 2017-05-04z 2017-05-05ZIdx1ZIdx2r=r r r r3r rr$ZC1ZC2rErr)rr) r to_datetimerrorr%rZget_level_valuesrr Timestampr?ra)Zdt_idxZm_idxr.rdrHrrrtest_loc_datetime_mask_slicings""rcCsPtd}tdtjd|fgddgddd}|jdd|gf}t||dS)N2000r rrr=r)rr)rrrrrr%r?ra)datercrdrrr&test_loc_datetime_series_tuple_slicings rcCstddgddgddgddggtjgddd gd d d gd }tjddgdd gd }|j|ddf}tddgddgddggtjgddd gd d d gd}t||dS)Nrr rrrr )r)r rr r rrrr=authorZpricerrr)rrrr)rrrr%r?r@)r.rrdrHrrrtest_loc_with_mi_indexers rcCstjdddd}ttd|d}|}||d<|jddd }|jjd ksNJ|j|d|j |d}|j dg d}t |||d}|jjd ksJ|j |d}|j dg d}t ||dS) Nz 2016-01-01r z US/Pacific)rtzrr rT)append)Nr)rrrr(Zto_framer>rrget_locr%rArr?r@ra)Zdtircr.df2rdrHrrrrtest_loc_mi_with_level1_named_0s r#cCstgdgdgdgdgdgdgddd }|d d g}|jd td fddfd}|jd jtd ddf}t||dS)N)z20160525 13:30:00.023MSFT51.95r%)z20160525 13:30:00.048ZGOOGz720.50z720.93)z20160525 13:30:00.076AAPLz98.55z98.56)z20160525 13:30:00.131r&98.6198.62)20160525 13:30:00.135r$z51.92r%)r)r&r'r(ztime,ticker,bid,ask,rZtickertimer&z2016-05-25 13:30:00r) rsplitr>rBr%rrr?r@)r.r"rGrHrrrtest_getitem_str_slices  r-cCsltjgdddd}gd}gd}t|||g}ttd|tjd}|j|d d d f}|d kshJdS) N)z 20181101 1100z 20181101 1200z 20181102 1300z 20181102 1400datetimer)rr)r[r[ZW)r#rhrFr3rnrr[r#g) rZ PeriodIndexrrDrr(rrpr%)piZlev2Zlev3rrcrdrrr!test_3levels_leading_period_indexsr3c@s$eZdZddZddZddZdS)TestKeyErrorsWithMultiIndexcCshttdddgdd}|ddg}tjtdd |jd Wdn1sZ0YdS) Nr5r3r rgrr[r#rrSr r) rrrrr>rVrWrXr%)r*r.r"rrr!test_missing_keys_raises_keyerrorsz=TestKeyErrorsWithMultiIndex.test_missing_keys_raises_keyerrorcCsXtdtddggdd}tjtdd|jdWdn1sJ0YdS) Nrrr r rz\(0, 3\)rS)rr )rrrorVrWrXr%)r*rcrrr!test_missing_key_raises_keyerror2sz=TestKeyErrorsWithMultiIndex.test_missing_key_raises_keyerror2cCs(tjtgdtgdtgdggdd}ttjdd|d}d }tjt |d ,|j d d t dfddfWdn1s0Ytjt |d &|j d d t dfWdn1s0Ytjt d d "|j dddfWdn1s0YdS)Nrrrr)rrr3)rrrr)rr"rr=r r3r rz'\('b', '1', slice\(None, None, None\)\)rSrrz \('b', '1'\))rr)rrDrrCrrrrVrWrXr%rrZget_locs)r*rr.msgrrrtest_missing_key_combinations   :4z8TestKeyErrorsWithMultiIndex.test_missing_key_combinationN)rrrr6r7r<rrrrr4sr4cCs2|}|d}|d}|jdd}t||dS)Nr[)r^r )r%r?ra)rbr.rcrdrHrrrtest_getitem_loc_commutabilitys r=cCstgdddtjggddddg}|jdg}tdd gitjd gddgd d }t|||jd}tdd git d gdd d }t||dS)N)r r r rr)r r3r )r8ind1ind2r>r?r8r )rr r=rr) rrrr>r%rrr?r@rrrrrtest_loc_with_nans   rAcCs\tgdggddgd}tjtdd|jdWdn1sN0YdS)Nrrrrz\(2\.0, 2\.0, 3\.0\)rS)@rB@)rr>rVrWrXr%r.rrrtest_getitem_non_found_tuple)s rEcCstjddd}t|g}|dtdddks4J|dtdddksNJ|dddd}|dddd}||ksJ|dd}|dd}||ksJ||d}||d}t | d|dS)Nz 2001-01-01drz2001-01rr ) rrrrDr!rrepeatr r?rZnonzero)rrr%rHrrrtest_get_loc_datetime_index2s   rIcCstddgddgg}tddgddgdd gd d gg|d }dddgf}td dgddgg|j|ddf<tddgd dgdd gd d gg|d }t||dS)Nrrrr r r r3r rr$rNrrirOr6r5)rrorrrCr%r?r@)rr.rrHrrr,test_loc_setitem_indexer_differently_orderedHs $ $$rJcCstddgddgddgddgggd gd gd d gd }|jtdddgfddf}tddgddgddgddgggdgdgd d gd }t||dS)Nr r r r3r rr$rNr8r r r r rrr)rrrr)r r r r )rr%rr?r@rrrr5test_loc_getitem_index_differently_ordered_slice_noneSsrLr)r r r$rr r rNr$)r r r$rr rNcCstdgdtgddgd}|jtd|fddf}tdgdgdgdgdgd}t|||j|jj|ddddf}t||dS) Nr rN)rr)r r$r5r)r r )r rN)r r$rr)r r r r r r r r )r r r r$r$rr rN)r) rrrr%rr?r@risin)rr.rdrHrrr@test_loc_getitem_index_differently_ordered_slice_none_duplicatescs  rNcCstjdgdgdgggdd}tddgi|d}|dd g}|jddddf}t||tdg|d}|jddddf}tdgtdgd d d}t ||dS) Nrqrrzrr=rrrr rr@) rrDrrr%r?r@rrra)rr.rHrdrcrrr3test_loc_getitem_drops_levels_for_one_row_dataframeys rPcCsjtgdddgg}ttd|d}|jddddgf}ttdtgd d}t||dS) Nrgr r rrr#r[)r r3))r#r )r#r )r[r )r[r ) rrorrZzerosr%rr?r@)colsr.rdrHrrr$test_mi_columns_loc_list_label_orders rRcCsttddgdgdggdgdgd}dd g|j_d d g|j_tjt d d "|j ddgdfWdn1s0YdS)Nr5r:r8rK)OhiorSColorado)GreenRedrUrkey1key2statecolorz\[2\] not in indexrSrr rT) rrrrrrr)rVrWrXr%)r,rrr$test_mi_partial_indexing_list_raisess  r[cCsbttdtddgddggd}tjtdd|jd d gWdn1sT0YdS) Nr3r r rrrz\['not' 'found'\] not in indexrSnotfound)rr(rrorVrWrXr%)rrrr(test_mi_indexing_list_nonexistent_raisess"r^cCstgdgdgddgtddgddggd}d |jd <d |jd <d |jd <tgdgdgddtjtjtjgggdtddgddggd}t||dS)N)r r r r)r r3r$rNrr r r[r#rrrr )rr r[)rr_)rrBg@g@)rCr4g@g @)rCgrrrC)rrrr)rrror%rrr?r@)rdrHrrr'test_mi_add_cell_missing_row_non_uniques$    r`cCstdddtjdgdggddgdd }|jd }|dks>Jt|tjsNJ|jd gd fjd }|dkspJt|tjsJdS)Nrr rr r3rrr=r)rrrrr)rrrDr% isinstancerrvrA)r.rdrrr$test_loc_get_scalar_casting_to_floats    rbcCsftjddgddggddgd}t|tjd}|jd}ttjtjgtddgdd d }t ||dS) Nrrr[r#r rr=rnr@r) rrorrrpr%rrr?ra)rs2rdrHrrr)test_loc_empty_single_selector_with_namess   rdcCsftgdgdtdd}|ddg}tjtdd|jd Wdn1sX0YdS) N)rFrFre,rf)rOrOrrG!rrgr[r#z^1$rS)rFr )rr(r>rVrWrXr%rDrrr'test_loc_keyerror_rightmost_key_missingsricCs8tddg}tddg|d}|jd}|dks4JdS)Nr)r )r3r r r r)rrrr%)rrcrdrrr*test_multindex_series_loc_with_tuple_labels rj)8numpyrrVZ pandas.errorsrrZpandasrrrrrZpandas._testingZ_testingr?ZfixturerrrrrrZ IndexSlicerrrrrrrrr rrrrr#r-r3r4r=rArErIrJrLrNrPrRr[r^r`rbrdrirjrrrrsv             $