U e{j@sldZddlZddlmZmZddlmZddlZdgZGddde Z ddd Z d d Z d d Z ddZdS)z High-level operations for numpy structured arrays. Some code and inspiration taken from numpy.lib.recfunctions.join_by(). Redistribution license restrictions apply. N)Counter OrderedDict)SequenceTableMergeErrorc@s eZdZdS)rN)__name__ __module__ __qualname__r r U/var/www/html/bwave/python/venv/lib/python3.8/site-packages/astropy/table/np_utils.pyrs{col_name}_{table_name}c stfddg}|dkr4ddttD}tD]\}}||}|jjD]p}|krx|kr|n@t} | |t fdd| Dr|j |d}||||<qTq%z"get_col_name_map..NcSsg|]}t|dqS)str).0iir r r )sz$get_col_name_map..c3s|]}|jjkVqdSr )dtypenames)rother)namer r :sz#get_col_name_map..) table_nameZcol_namecSsg|]\}}|dkr|qSrr )rrcountr r r rDszgMerging column names resulted in duplicates: {}. Change uniq_col_name or table_names args to fix this.c3s|]}||fVqdSr r rr) col_name_mapr r rNs) collections defaultdictranger enumeraterrappendlistpopanyformatritemsrr) rZ common_namesZ uniq_col_nameZ table_namesZ col_name_listidxarrayrout_nameZothersZcol_name_countZrepeated_namesr )rrrr get_col_name_maps<    r-c Csg}|D]\}}ddt||D}dd|D}z t|}Wn:tk r|}ztd|d|j|W5d}~XYnXdd|D} t| d krtd | } |dk rt|}| ||| fq |S) z Find the dtypes descrs resulting from merging the list of arrays' dtypes, using the column name mapping ``col_name_map``. Return a list of descrs for the output. cSs g|]\}}|dk r||qSr r )rarrrr r r r_szget_descrs..cSsg|]}|dk r|qSr r rr r r rbsz,The '{}' columns have incompatible types: {}rNcSsh|]}|jddqS)rN)shapercolr r r qszget_descrs..rz Key columns have different shape) r)zip common_dtyperr(_incompat_typesr r&rr$) rrZ out_descrsr,Zin_namesZin_colsrrtmeZ uniq_shapesr/r r r get_descrsSs.  r7cstjtjtjtjtjffdd|D}t|dkr\dd|D}td|}||_|dd|D}|D]}|j j dkrnd |j |d <qnt d d|D}|j j S) z Use numpy to find the common dtype for a list of structured ndarray columns. Only allow columns within the following fundamental numpy data types: np.bool_, np.object_, np.number, np.character, np.void cs"h|]tfddDqS)c3s|]}tjj|VqdSr ) issubclassrtype)rZnp_typer1r r rsz)common_dtype...)tuple)rZnp_typesr:r r2szcommon_dtype..rcSsg|] }|jjqSr )rrr0r r r rsz common_dtype..z Columns have incompatible types cSsg|]}tjd|jdqS)r)r)npemptyrr0r r r rs)SU0rcSsg|] }|dqS)rr )rr.r r r rs)r=Zbool_Zobject_number charactervoidr rr5rkinditemsizer+r)colsZ uniq_typesZincompat_typesr6Zarrsr.Z arr_commonr r<r r4}s   r4cCsXd}t|tst||D]$}t|tjr6|jjdkrt|qt|dkrTtddS)Nz@`arrays` arg must be a sequence (e.g. list) of structured arraysrz,`arrays` arg must include at least one array) isinstancer TypeErrorr=Zndarrayrrr ValueError)rerrr+r r r (_check_for_sequence_of_structured_arrayss   rL)r N)__doc__r rrcollections.abcrnumpyr=__all__rJrr-r7r4rLr r r r s  >*