o EbW @s dZddlZddZddZdS)z Pythran implementation of columns grouping for finite difference Jacobian estimation. Used by ._numdiff.group_columns and based on the Cython version. Nc Cs|j}tj|tjd }d}tj|tjd}t|D][}||dkr#q|||<d}|||dd<t|jdD]3} || dkrCd}nq8d} t|D]} || dkr^|| | fdkr^d} nqJ| sk||| 7}||| <q8|rq|S|d7}q|S)NZdtyperTF)Tnponesintpemptyrangeshape) mnABgroups current_groupunioni all_groupedj intersectkr?/usr/lib/python3/dist-packages/scipy/optimize/_group_columns.py group_dense s:      rc Cs*tj|tjd }d}tj|tjd}t|D]{}||dkr q|||<d}|dt||||dD]} d||| <q6t|jdD]B} || dkrQd}nqFd} t|| || dD]} ||| dkrmd} nq_| st|| || dD]} d||| <q{||| <qF|r|S|d7}q|S)NrrTrF)rrrrr fillr ) r r indicesZindptrrrrrrrrrrrr group_sparse;s>     r)__doc__Znumpyrrrrrrrs 1