o ;Di@sddlmZmZmZddlmZddlmZmZddZ e e_ GdddeZ dd Z dd d Z e e_ Gd ddeZ ddZdddZee_GdddeZddZd S))absolute_importprint_functiondivision)next)Table asindicescOst||fi|S)a Replace missing values with non-missing values from the row above. E.g.:: >>> import petl as etl >>> table1 = [['foo', 'bar', 'baz'], ... [1, 'a', None], ... [1, None, .23], ... [1, 'b', None], ... [2, None, None], ... [2, None, .56], ... [2, 'c', None], ... [None, 'c', .72]] >>> table2 = etl.filldown(table1) >>> table2.lookall() +-----+-----+------+ | foo | bar | baz | +=====+=====+======+ | 1 | 'a' | None | +-----+-----+------+ | 1 | 'a' | 0.23 | +-----+-----+------+ | 1 | 'b' | 0.23 | +-----+-----+------+ | 2 | 'b' | 0.23 | +-----+-----+------+ | 2 | 'b' | 0.56 | +-----+-----+------+ | 2 | 'c' | 0.56 | +-----+-----+------+ | 2 | 'c' | 0.72 | +-----+-----+------+ >>> table3 = etl.filldown(table1, 'bar') >>> table3.lookall() +------+-----+------+ | foo | bar | baz | +======+=====+======+ | 1 | 'a' | None | +------+-----+------+ | 1 | 'a' | 0.23 | +------+-----+------+ | 1 | 'b' | None | +------+-----+------+ | 2 | 'b' | None | +------+-----+------+ | 2 | 'b' | 0.56 | +------+-----+------+ | 2 | 'c' | None | +------+-----+------+ | None | 'c' | 0.72 | +------+-----+------+ >>> table4 = etl.filldown(table1, 'bar', 'baz') >>> table4.lookall() +------+-----+------+ | foo | bar | baz | +======+=====+======+ | 1 | 'a' | None | +------+-----+------+ | 1 | 'a' | 0.23 | +------+-----+------+ | 1 | 'b' | 0.23 | +------+-----+------+ | 2 | 'b' | 0.23 | +------+-----+------+ | 2 | 'b' | 0.56 | +------+-----+------+ | 2 | 'c' | 0.56 | +------+-----+------+ | None | 'c' | 0.72 | +------+-----+------+ Use the `missing` keyword argument to control which value is treated as missing (`None` by default). ) FillDownView)tablefieldskwargsr M/var/www/Datamplify/venv/lib/python3.10/site-packages/petl/transform/fills.pyfilldown sNrc@eZdZdddZddZdS)rNcCs||_||_||_dSN)r r missing)selfr r rr r r __init__`s zFillDownView.__init__cCst|j|j|jSr) iterfilldownr r rrr r r __iter__eszFillDownView.__iter__r__name__ __module__ __qualname__rrr r r r r^s  rc cst|}zt|}Wn tyYdSwt|V|s|}t||}tt|}t|V|D]#}t|}|D]} || |krH|| || <q9|| || <q9t|Vq1dSr)iterr StopIterationtuplerlist) r fillfieldsrithdr fillindicesfillrowoutrowidxr r r ris*        rNcC t||dS)aY Replace missing values with preceding non-missing values. E.g.:: >>> import petl as etl >>> table1 = [['foo', 'bar', 'baz'], ... [1, 'a', None], ... [1, None, .23], ... [1, 'b', None], ... [2, None, None], ... [2, None, .56], ... [2, 'c', None], ... [None, 'c', .72]] >>> table2 = etl.fillright(table1) >>> table2.lookall() +------+-----+------+ | foo | bar | baz | +======+=====+======+ | 1 | 'a' | 'a' | +------+-----+------+ | 1 | 1 | 0.23 | +------+-----+------+ | 1 | 'b' | 'b' | +------+-----+------+ | 2 | 2 | 2 | +------+-----+------+ | 2 | 2 | 0.56 | +------+-----+------+ | 2 | 'c' | 'c' | +------+-----+------+ | None | 'c' | 0.72 | +------+-----+------+ Use the `missing` keyword argument to control which value is treated as missing (`None` by default). r) FillRightViewr rr r r fillright &r+c@r)r)NcC||_||_dSrr*rr rr r r r zFillRightView.__init__cCt|j|jSr) iterfillrightr rrr r r rzFillRightView.__iter__rrr r r r r)  r)ccst|}zt|}Wn tyYdSwt|V|D].}t|}t|D]\}}|dkrE|||krE||d|krE||d||<q't|VqdSNr)rrrrr enumerater rr r!r$r%i_r r r r1s    $ r1cCr')aX Replace missing values with following non-missing values. E.g.:: >>> import petl as etl >>> table1 = [['foo', 'bar', 'baz'], ... [1, 'a', None], ... [1, None, .23], ... [1, 'b', None], ... [2, None, None], ... [2, None, .56], ... [2, 'c', None], ... [None, 'c', .72]] >>> table2 = etl.fillleft(table1) >>> table2.lookall() +-----+------+------+ | foo | bar | baz | +=====+======+======+ | 1 | 'a' | None | +-----+------+------+ | 1 | 0.23 | 0.23 | +-----+------+------+ | 1 | 'b' | None | +-----+------+------+ | 2 | None | None | +-----+------+------+ | 2 | 0.56 | 0.56 | +-----+------+------+ | 2 | 'c' | None | +-----+------+------+ | 'c' | 'c' | 0.72 | +-----+------+------+ Use the `missing` keyword argument to control which value is treated as missing (`None` by default). r() FillLeftViewr*r r r fillleftr,r;c@r)r:NcCr-rr*r.r r r rr/zFillLeftView.__init__cCr0r) iterfillleftr rrr r r rr2zFillLeftView.__iter__rrr r r r r:r3r:ccst|}zt|}Wn tyYdSwt|V|D]2}tt|}t|D]\}}|dkrG|||krG||d|krG||d||<q)tt|VqdSr4)rrrrrreversedr6r7r r r r<s     $r<r) __future__rrr petl.compatrpetl.util.baserrrrrr+r)r1r;r:r<r r r r s Q ) )