U
|
ý°dºF ã @ s¬ d dl Zd dlZd dlmZ d dlm mZ d dl Z
|
d dl m Z m Z m Z mZmZmZmZ d dlmZ dd ZG dd dZG dd d Zejd
|
d Zd d Zdd ZdS )é N)ÚSettingWithCopyError)Ú DataFrameÚ
|
MultiIndexÚSeriesÚ TimestampÚ
|
date_rangeÚisnaÚnotnac C s | |ks t d S ©N)ÚAssertionError)ÚaÚb© r údd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/indexing/multiindex/test_setitem.pyÚ assert_equal s r c @ sÀ e Zd ZedfddZdd Zdd Zdd Zej d
|
d Z
|
d d Z dd Z dd Z dd Zdd Zdd Zdd Zdd Zd d Zd d Zd d! Zd"d# Zd$d% Zd&d' Zd(d) Zd*d+ ZdS ),ÚTestMultiIndexSetItemNc C s. ||j |< |j | }|d kr |}||| d S r
|
)Úloc)ÚselfÚtargetÚindexersÚvalueÚ
|
compare_fnÚexpectedÚresultr r r Úcheck s
|
|
|
zTestMultiIndexSetItem.checkc C sæ ddddddddg}t jt d d
|
¡t d d ¡gd d gd}d\}}ttj||d}| j|||fdfd d td||d}| j|||fdfdd t||d}| j|||fdfdd td||d}| j|||fdft d¡dd d S )NÚAÚwÚlr ÚxÚXÚdZprofitr éd éP ÚtimeZfirm©Únames)r é ©ÚcolumnsÚindex)r r r iüÿÿé r& é )r r r r )r Ú from_productÚnpÚaranger Únanr Úarray)r Úcolsr) ÚtÚnÚdfr r r Útest_setitem_multiindex s. ÿý
|
üz-TestMultiIndexSetItem.test_setitem_multiindexc C s¤ t t d¡ dd¡d d¡td}d|d< |d d d
|
k}d d g}|d |j||f< t d d
|
d dgd d gd}t |j||f |¡ | j |||f|d tj|d d S )Né é z A,B,C,D,Eú,©r( Údtypeéc ÚFr r& r ÚBÚCg ÀX@é ©r) r( ©r r r r r )
|
r r- r. ÚreshapeÚsplitÚfloatr ÚtmÚassert_frame_equalr )r r4 Z row_selectionZ col_selectionÚoutputr r r Útest_setitem_multiindex2> s$ ÿûz.TestMultiIndexSetItem.test_setitem_multiindex2c C s t dddgtddddg¡}t dd gtd
|
d ddg¡}ttj d ¡||d }t dtdfdtdfg¡}t dtd
|
fdtd fg¡}ttj d¡||d }| j|||f|t j
|
d ttj d¡||d }| j||t d d d f|t j
|
d | ¡ }| j||j |jf|t j
|
|d d S )Nr r= r> z
|
2015-01-01z
|
2015-04-01ZMS)ÚfreqÚfooÚbarz
|
2016-01-01z
|
2016-02-01)é r? r@ z
|
2015-02-01)r& r& )r r r r )r& r? rA )r r, r r r- ÚrandomÚ from_tuplesr r rE rF ÚsliceÚcopyr) r( )r Úidxr1 r4 ZsubidxZsubcolsÚvalsrP r r r Útest_setitem_multiindex3Q sF ÿÿÿÿüü
|
ûz.TestMultiIndexSetItem.test_setitem_multiindex3c C sÎ t ddddddg¡t ddddddg¡t ddd¡g}tt j dd ¡|d
|
d d gd ¡ }|jdg d }| ¡ }|jdg d9 < t
|
|jdg |¡ d}t j t |d |jd d9 < W 5 Q R X d S )NrK ZbazZquxÚoneÚtwor é r* r+ r r= r> r@ r& zAcannot align on a multi-index with out specifying the join levels©Úmatch)r- r0 r. r rM ÚrandnÚ
|
sort_indexr rP rE rF ÚpytestÚraisesÚ TypeError)r ÚarraysÚdf_origr r4 Úmsgr r r Útest_multiindex_setitem~ s ý ÿ
|
z-TestMultiIndexSetItem.test_multiindex_setitemc
|
C sÚ t ddddddddi¡}tj|jdd d
|
gd |_| ¡ }|jddd g d9 < tj}| ¡ }|j |d d
d d
d f d d
f d9 < t
|
||¡ | ¡ }|j |d d
d d
d f df d9 < t
|
||¡ d S )NZpricer& r? r* r7 r ))ÚDEZCoalÚStock)rb ÚGasrc )rb ZElecZDemand)ÚFRrd rc )re ZSolarÚSupIm)re ZWindrf ZSitZComÚTyper$ r+ rc ) r Ú from_dictr rN r) rP ÚilocÚpdZ
|
IndexSlicer rE rF )r r_ r rQ r4 r r r Útest_multiindex_setitem2 s0 úÿÿ ÿ, (z.TestMultiIndexSetItem.test_multiindex_setitem2c C s~ t tjjdddd dd¡tddddgddd ggd
|
}tj|d < t d d g¡}||jd< t
|
|jd t |ddgd d¡ d S )Nr7 é
|
é ©Úsizer+ Úabcr? é rL r' r ç ç ð?)r? r )r) Úname) r r- rM ÚrandintrB Úlistr/ r0 r rE Úassert_series_equalr )r r4 Úarrr r r Útest_multiindex_assignmentµ s ý
|
|
z0TestMultiIndexSetItem.test_multiindex_assignmentc C s t ddg¡}tt jjdddd dd¡tdd d d
|
gd
|
dd ggt jd }|d jd d
j }||j
|
d< t |d
|
dgd dd}|j
|
d }t ||¡ |s¤t ||j ¡ |d |j
|
d< |j
|
d }|d }t ||¡ d|j
|
d< t dd
|
dgd dd}t |j
|
d |¡ d}tjt|d ddddg|j
|
d< W 5 Q R X tjt|d dg|j
|
d< W 5 Q R X dg|j
|
d d gf< |j
|
d dk ¡ std S )Nrr rs r7 rl rm rn r+ rp r? rq rL )r( r) r: Úcr& )r? rz Úint64)r) rt r: g à?Úfloat64z@Must have equal len keys and value when setting with an iterablerW r r* )r- r0 r rM ru rB rv r{ ri Úvaluesr r rE rw Zassert_numpy_array_equalr[ r\ Ú
|
ValueErrorÚallr )r Úusing_copy_on_writerx r4 ÚviewÚexpr r` r r r Ú'test_multiindex_assignment_single_dtypeÄ s8 ü
|
|
|
|
z=TestMultiIndexSetItem.test_multiindex_assignment_single_dtypec
|
C s® d}d}dd t tt |¡ ¡ D }|d d
}ttjjd||fdtj|d}| |¡
|
¡ }|j |d d
d }tj |d
|
< |D ]&\}}t |j d ¡} | |j|d
|
f< qd S ) Nr! rl c S s g | ] }d | qS )r r )Ú.0Únumr r r Ú
|
<listcomp>÷ s z>TestMultiIndexSetItem.test_groupby_example.<locals>.<listcomp>r7 rn )r: r( r? )ÚlevelZnew_colr )ÚmapÚstrr- r. Útolistr rM ru r{ Ú set_indexrZ Úgroupbyr/ Úshaper )
|
r ZNUM_ROWSZNUM_COLSZ col_namesZ
|
index_colsr4 Úgrprt Zdf2Únew_valsr r r Útest_groupby_exampleó s ý
|
z*TestMultiIndexSetItem.test_groupby_examplec C s¦ |}|d }t j|d< t|jdd
¡ s0tt|jd d
¡ sJtt|jdd
¡ sdtt j|d< t|jd sttj t
|
dd |d W 5 Q R X d S ) Nr )éÐ r+ é* éA )r r+ rl é1 Z49rW ) r- r/ r r} r r r ri r[ r\ ÚKeyError)r Ú/multiindex_year_month_day_dataframe_random_dataÚymdÚsr r r Útest_series_setitem s
|
|
z)TestMultiIndexSetItem.test_series_setitemc C sø |}|j ¡ }|j ¡ }||dk }| |dk¡}t ||¡ d||dk< d||dk< t |j|¡ d||dk< d||dk< t |j|¡ d||d d
dk < t |d d
|d d
dk d¡ t |j|¡ t j
|
t dd d||d < W 5 Q R X d S )Nr r7 r& éÿÿÿÿzboolean values onlyrW ) ÚTrP r} ÚwhererE rF Zassert_almost_equalr- Zputmaskr[ r\ r] )r Ú multiindex_dataframe_random_dataÚframer4 r} r r r r r Ú"test_frame_getitem_setitem_boolean s"
|
|
"z8TestMultiIndexSetItem.test_frame_getitem_setitem_booleanc C s ddgdddgg}dddddgdddddgg}t ||d d gd
|
}td ddd d dgi|d}|jd d
d f }t |d |¡ |j|jdd
d f }t |d dd
|¡ |jd d
d d
f }t ||¡ |}d|jd d
d f< d|d < t ||¡ d|jd d
d d
f< t ||¡ d S )NÚt1Út2r r rz r r* r& Úid)ÚcodesÚlevelsr% r r+ é rq ©r) rl )r r r rE rw r) rF )r r¤ r£ Zmidxr4 r r r r Ú%test_frame_getitem_setitem_multislice2 s z;TestMultiIndexSetItem.test_frame_getitem_setitem_multislicec C s t tj dd¡ddddgddddggd}| ¡ }|d |d< t |d |d ¡ | ¡ }|d j|d< t |d |d ¡ d S )Nrl r? r r r r* ©r( )r r- rM rY rP rE rF r} )r r4 Úcpr r r Ú test_frame_setitem_multi_columnI s ÿ z5TestMultiIndexSetItem.test_frame_setitem_multi_columnc C sÀ t dddg¡}tdddg|d}d|d < |d jd
|
k ¡ s@tdd dg|d< |d |d < |d }|d }|d }tj||d d tj||d d |jdks t|jdks®t|jdks¼td S )N©r Ú1©r Ú2©r= r¬ r* r+ r7 r@ rr r r r& F)Z check_names) r rN r r} r r rE rw rt )r r( r4 Z sliced_a1Z sliced_a2Z sliced_b1r r r Ú test_frame_setitem_multi_column2W s z6TestMultiIndexSetItem.test_frame_setitem_multi_column2c C sH |}|d d
}|j ddddgf }|j d dddg }t ||¡ d S )Nr7 )r r* rV r r= r> )r rE rw )r r r r4 r r r r r Ú#test_loc_getitem_tuple_plus_columnsn s
|
z9TestMultiIndexSetItem.test_loc_getitem_tuple_plus_columnsc C s t dddgddggddddddgddddddggd}ttj t|d¡|dddd gd
|
}t ||¡}|jdd
}| |j
|
dd
¡}t ||¡ d |jdd
< |jdd
d kj ¡ s¾td S ) Nr r* r& )r¤ r£ r? r r rz r r@ r¥ )r r r- rM rY ÚlenrE Úget_objr Úreindexr) r r} r r )r Zframe_or_seriesr) ÚobjÚresr r r r Ú'test_loc_getitem_setitem_slice_integersy s ÿ
|
ÿ z=TestMultiIndexSetItem.test_loc_getitem_setitem_slice_integersc C sd |}|j }|d }|| ¡ k|d< t |d || ¡ k¡ |jdgd}t |d || ¡ k¡ d S )N)rJ rU r¨ )r ZmedianrE rw r´ )r r r Zdftr Z reindexedr r r Útest_setitem_change_dtype s z/TestMultiIndexSetItem.test_setitem_change_dtypec C s~ |}|j dddg }d|j|< |j| jdk ¡ s6t| ¡ }|d }d||< |r`t ||¡ n|j|df dk ¡ sztd S )Nr* r? r7 r; r= éa )r) r r} r r rP rE rF )r r r r ZsubsetZframe_originalÚcolr r r Ú test_set_column_scalar_with_loc s
|
z5TestMultiIndexSetItem.test_set_column_scalar_with_locc C st t ddddgddddgddddggtd d
|
}| d d g¡}t d g¡}d|j|df< | d ¡d dk ¡ sptd S )Nr* r r ÚyÚYr& ÚzÚZÚABCDr¨ r r= ©r* r* Ú_r> ) r rv r r rN r Úxsr r )r r4 Úmir r r Ú test_nonunique_assignment_1750§ s ÿ z4TestMultiIndexSetItem.test_nonunique_assignment_1750c C sZ t dddg¡}tt d¡ d¡|td}|j ¡ }|d tj
|
¡|d< t |j|¡ d S )Nr« r¯ r r+ )r* r+ r9 r ) r rN r r- r. rB Úobjectr) rP Zastyper| rE Zassert_index_equal)r r1 r4 r) r r r Ú test_astype_assignment_with_dups³ s
|
|
z6TestMultiIndexSetItem.test_astype_assignment_with_dupsc C sj t jdddgddgd}tddd g|d
|
gd }tjd d dd|jd< td
|
d ddgi|d}t ||¡ d S )N)r rz )r r )r r Úl1Úl2r$ r r* r& Úe)Údatar) r( r; ée r{ )r: )r rÊ r! r¦ )r rN r r- r. r rE rF )r r) r4 r r r r Útest_setitem_nonmonotonic½ s ÿz/TestMultiIndexSetItem.test_setitem_nonmonotonic)Ú__name__Ú
|
__module__Ú __qualname__r r r5 rH rS ÚtdZ&skip_array_manager_not_yet_implementedra rk ry r r r r r§ rª r° r± r· r¸ r» rÅ rÇ rÍ r r r r r s, -
|
/
|
r c @ s e Zd Zdd Zdd ZdS )Ú"TestSetitemWithExpansionMultiIndexc C s ddddddgddddddgdd d
|
dddgg}t t| }t |¡}ttj d d ¡|d }| ¡ }| ¡ }dddd g|d< dddd g|d< t
|
||¡ d S )Nr ÚtopZroutine1Zroutine2Ú ZODZresult1Zresult2ZwxZwyr? rV r¨ r* r& r+ r )r rÔ rÔ ) ÚsortedÚzipr rN r r- rM rY rP rE rF )r r^ Ztuplesr) r4 r r r r r Ú#test_setitem_new_column_mixed_depthÉ s ý
|
zFTestSetitemWithExpansionMultiIndex.test_setitem_new_column_mixed_depthc C s\ t dddg¡}tddgddgdd gg|d
|
}tddd }||d < |d ¡ ¡ sXtd S ) N)Ú1aZ2a)rØ Z2b)rØ Z2cr* r& r+ r? r7 rV r¦ )rÁ )r* r& Únew)r rN r r r r r )r Zmixr4 r r r r Ú test_setitem_new_column_all_naÚ s
|
zATestSetitemWithExpansionMultiIndex.test_setitem_new_column_all_naN)rÎ rÏ rÐ r× rÚ r r r r rÒ È s rÒ c C sz | j }|rNtjtdd d|d jd d
< W 5 Q R X |d jdk ¡ svtn(d|d jd d
< |d jdk ¡ svtd S )Nz read-onlyrW r rJ )r r[ r\ r~ r} r r )r r r4 r r r Ú test_frame_setitem_view_directã s rÛ c C sZ | j }|r,t ¡ d|d d< W 5 Q R X n*d}tjt|d d|d d< W 5 Q R X d S ©Nr& rJ rT zAA value is trying to be set on a copy of a slice from a DataFramerW )r rE Ú raises_chained_assignment_errorr[ r\ r )r r r4 r` r r r Ú test_frame_setitem_copy_raisesô s
|
rÞ c C sv | j }|}| ¡ }|r8t ¡ d|d d< W 5 Q R X n*d}tjt|d d|d d< W 5 Q R X |}t ||¡ d S rÜ )r rP rE rÝ r[ r\ r rF )r r r r r4 r` r r r r Ú test_frame_setitem_copy_no_write s
|
rß ) Únumpyr- r[ Z pandas.errorsr Z pandas.util._test_decoratorsÚutilZ_test_decoratorsrÑ Zpandasrj r r r r r r r Zpandas._testingZ_testingrE r r rÒ Z skip_array_manager_invalid_testrÛ rÞ rß r r r r Ú<module> s $ 3
|
|
