Файловый менеджер - Редактировать - /home/digitalm/venv/lib/python3.7/site-packages/pandas/tests/indexing/multiindex/__pycache__/test_slice.cpython-37.pyc
Назад
B �5�gQd � @ sj d dl Zd dlZd dlmZ d dlZd dlmZmZm Z m Z m Z d dl m Z d dlmZ G dd� d�ZdS )� N)�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� Zdd � Zd S ) �TestMultiIndexSlicersc C s t �tdd�tdd�tdd�tdd�g�}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 d Q R X tjtd2d/�� |jt d �d$gf W d Q R X |j jdk�s^t�|jd$d"d3�}|j jd"k�s~t�d4}tjt |d/��$ |jt d �t d,�fd d �f W d 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 sH g | ]@\}}}}|d ks(|dks(|dkr|dks8|dkr||||f�qS )r �A2r r r � )�.0�a�b�c�dr r �a/home/digitalm-up/venv/lib/python3.7/site-packages/pandas/tests/indexing/multiindex/test_slice.py� <listcomp> s zITestMultiIndexSlicers.test_per_axis_per_level_getitem.<locals>.<listcomp>c S sP g | ]H\}}}}|d ks(|dks(|dkr|dks@|dks@|dkr||||f�qS )r r r r ZC2r r )r r r r r r r r r - s )r � )r r )r � )r r �one�two)�names)r �foo)r �bar)r r% )r �bah�lvl0�lvl1� �int64)�dtype)r �columnsr )�axisr r! r% )r r � � )r& r% )�namer& c S sH g | ]@\}}}}|d ks(|dks(|dkr|dks8|dkr||||f�qS )r r r r r r )r r r r r r r r r u s zLcannot index with a boolean indexer that is not the same length as the index)�matchTFz\[1\] not in index)�levelr. 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 ) �selfZix�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��� }|jjrNt�t ddgddgddgddgd���dd d g��� }|jtd �td �dfd d �f }t� ||� |j ddd d �}t� ||� t ddddgddddgddddgddddgd���dd d g��� }|jj�rt�t ddgddgddgddgd���dd d g��� }|jtd �td �dfd d �f }|jj�rxt�t� ||� ddddd dddddddddddddddddg}t |�}t � dg| |g�}tdg| |d �}|�� }|jtd �td �f }tdg|d |d d � d ��� }t�||� d S ) Nr% r r r r! r )r r r r r r r F)r3 Z drop_levelr � r r0 r/ � � � � � r* � � � i@ )r i�� ���)r � set_indexr? r Z is_uniquerG r8 r9 r; r< Zxsr7 r Z from_arraysr rA )rH rI rK rJ �ints�n�idxr r r �"test_multiindex_slicers_non_unique� sl " " "z8TestMultiIndexSlicers.test_multiindex_slicers_non_uniquec s� dd l � � fdd�td�D �}ddg}tj||gddgd �}ttjd d d ��d d�|td�d�}t j }|j dddgddgf }|j t td�td��t dd�ft dd�f }t�||� |j |td�td�� |dd� ft dd�f }t�||� |j t td�td��dft dd�f }t�||� |j t dd�t dd�ft dd�f }t�||� |j |dd� df|d f }t�||� d S )Nr c s, g | ]$}� � d ddddd�� j|d� �qS )i� r rR )�days)�datetime� timedelta)r �i)r_ r r r � s zNTestMultiIndexSlicers.test_multiindex_slicers_datetimelike.<locals>.<listcomp>rO r r �dateZ frequency)r$ �0 r+ )r, ���r ZABCD)r r- z2012-01-01 12:12:12z2012-01-03 12:12:12r r )r r )r_ �ranger r4 r r5 r6 r> �list�pd� IndexSlicer@ r8 r9 r r; r< )rH �dates�freqr rI r\ rK rJ r )r_ r �$test_multiindex_slicers_datetimelike� sH 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�}|�� }|jtd�d d �f }|jd d � }t� ||� |jtd�d d �f }|}t� ||� |jtd �tdd�fd d �f }|jdddd d dd!d"dg }t� ||� |jtd �td �td#d$�fd d �f }|jddd%d d!g }t� ||� |jtd�td�fd d �f }|jd ddd%d d&g }t� ||� |jtd �td�fd d �f }|}t� ||� |jtd �tdd�td �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 �B0ZB1ZB2r! z 2013-06-11z 2013-07-02z 2013-07-09z 2013-07-30z 2013-08-06z 2013-09-03z 2013-10-01� �# rT r/ r �( rV r r )r r �DATEZVALUESrp r r r rP r r0 rR rS Z20130702Z20130709rO rQ Z20130701) r rg � to_datetimerY r? r8 r9 r@ r; r< )rH rI Zdf1rJ rK r r r � test_multiindex_slicers_edges; s\ ( & & &z3TestMultiIndexSlicers.test_multiindex_slicers_edgesc C sT t j}t�tdd�tdd�tdd�tdd�g�}tjddd d gd d gd �}ttjt |�t |� dd�� t |�t |�f�||d�}|j t dd�t 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 d�ff W d 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 )r r% )r r& )r r% )r r' r( r) )r$ r+ )r, )r r- r r r r c S sH g | ]@\}}}}|d ks(|dks(|dkr|dks8|dkr||||f�qS )r r r r r r )r r r r r r r r r � s zNTestMultiIndexSlicers.test_per_axis_per_level_doc_examples.<locals>.<listcomp>c S s0 g | ](\}}}}|d ks |dkr||||f�qS )r r r )r r r r r r r r r � s z_MultiIndex slicing requires the index to be lexsorted: slicing on levels \[1\], lexsort depth 1)r2 r r% r r )r. i����)rg rh r r4 r r= r r5 r6 r7 r> r8 r9 r r: r; r< rB rC r r@ r? )rH r\ r r- rI rJ rK rM r r r �$test_per_axis_per_level_doc_examples� sD & & ( $ ( 2 (z:TestMultiIndexSlicers.test_per_axis_per_level_doc_examplesc C s� t �tdd�tdd�tdd�tdd�g�}t jddd d gd d gd �}ttjt|�t|� dd��t|�t|�f�||d�� � j dd�}|j dd�dd�d d �ddgf }|j dd� |j j D � }t �||� |j dd�d d �d d �ddgf }|j dd� |j j D � }t �||� |j dd�d d �d f }|j d d �td �d ff }t �||� |j d d�d d �d f }|j d d �td �d ff }t �||� xVd D ]N}d |� d �}tjt|d!��( |j |d�d d �d d �ddgf W d Q R X �q�W d S )"Nr r r r r r )r r% )r r& )r r% )r r' r( r) )r$ r+ )r, )r r- r )r. r r r r r c S sH g | ]@\}}}}|d ks(|dks(|dkr|dks8|dkr||||f�qS )r r r r r r )r r r r r r r r r � s zATestMultiIndexSlicers.test_loc_axis_arguments.<locals>.<listcomp>r c S s0 g | ](\}}}}|d ks |dkr||||f�qS )r r r )r r r r r r r r r � s r% r- )rd r r% zNo axis named z for object type DataFrame)r2 )r r4 r r= r r5 r6 r7 r> r? r8 r r: r; r< r9 rB rC rD )rH r r- rI rJ rK ra rM r r r �test_loc_axis_arguments� s> & " " 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! r/ �a1�a2�a3�b1�b2�b3)r- r )r. ���) r r5 r6 r> r r4 r8 r@ r; r< )rH rI rJ rK 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 ) Nru r! r/ rv rw rx ry rz r{ )r- r )r. ) r r5 r6 r> r r4 r8 r@ r- r; r< )rH rI rJ rK 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�}|jdd�d }tt�d dd g�dd �}t�||� d S ) Nr/ r! r r r )r- r )r. r rO )r1 ) r r5 r6 r> r8 r rE r; rA )rH rI rJ rK 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�}|jdd�jdd�}|� � }d|j t d �t d �fd d �f< |� � }d|j d d �d d �f< t �||� |� � }d|j dd�d 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 dd�d 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 d Q R X d }tjt|d ��8 tjddddgdd�|j t d �dft d �dgff< W d 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 )!N)r r )r r )r r! )r r r" r# )r$ )r r% )r r& )r r% )r r' r( r) r* r+ )r, r )r r- r )r. r �d r! r% )r r r z)setting an array element with a sequence.)r2 z@Must have equal len keys and value when setting with an iterabler rP )r r' )rg rh r r= r r5 r6 r> r? �copyr8 r9 r@ r; r< rE rB rC rD ) rH r\ r r- Zdf_origrI rK rM �rhsr r r � test_per_axis_per_level_setitem6 s� & ( 2 :8@, $ 0 $(0z5TestMultiIndexSlicers.test_per_axis_per_level_setitemc s� t t�d�t�td�t�d�g��� tj}� fdd�}||d d d� |d d d� � ||dd d� |dd d� � ||d d d� |dd d� � ||d dd� |d d d� � ||d d d� |d d d� � ||dd d� |dd d� � ||d d d� |dd d� � ||dd d� |dd d� � ||d d d� |dd d� � ||d dd� |d d� � ||dd d� |dd d� � ||d dd� |d dd� � ||ddd� |ddd� � d S )N� Zabcder c s2 t �� j| � j| � t �� | � j| � d S )N)r; rA r8 r@ )Zl_slcZi_slc)rL r r �assert_slices_equivalent� s zhTestMultiIndexSlicers.test_multiindex_label_slicing_with_negative_step.<locals>.assert_slices_equivalentrd r � )r rQ r r! )r r )r r rP r/ )�er r* )r r5 r6 r r4 rf rg rh )rH ZSLCr� r )rL r �0test_multiindex_label_slicing_with_negative_step� s" " zFTestMultiIndexSlicers.test_multiindex_label_slicing_with_negative_stepc C s� ddddg}t �|t�d�g�}tttd��|dgd�}|jtj d d �d d �f d d �f }|jd }tttd d ��dgtd d �d �}t � ||� |jd }tttd d��dgtd d �d �}t � ||� d S )Nr r r r i� i� ZTest)r r- � �F �G )r- r i� i# ) r r4 r5 r6 r rf re r8 rg rh r; r< )rH rj r\ rI Zdf_slicerJ rK 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 r; rA r� r: Zassert_numpy_array_equalr< )rH Z/multiindex_year_month_day_dataframe_random_dataZymdrL rJ rK �expr r r �test_int_series_slicing� s z-TestMultiIndexSlicers.test_int_series_slicingc C s� t �ddgddgg�}tddgddgdd gd d gg|d �}|jdtd d d �fd d �f }tddgddggt �ddg�d �}t�||� d S )Nr r r r r r! r r rO r r0 )r rd )r r )r r )r r4 r r8 r9 r= r; r< )rH �mirI rJ rK r r r � test_loc_slice_negative_stepsize� s $ z6TestMultiIndexSlicers.test_loc_slice_negative_stepsizeN)�__name__� __module__� __qualname__rN r] rk rr rs rt r} r~ r r� r� r� r� r� r r r r r s UCJLC r )�numpyr5 rB Z pandas.errorsr Zpandasrg r r r r r Zpandas._testingZ_testingr; Z pandas.tests.indexing.commonr r r r r r �<module> s
| ver. 1.4 |
Github
|
.
| PHP 8.3.23 | Генерация страницы: 0.03 |
proxy
|
phpinfo
|
Настройка