a ߙfb@stdZddlZddlmZmZddlmZddlZdgZGddde Z ddd Z d d Z d d Z ddZddZdS)z High-level operations for numpy structured arrays. Some code and inspiration taken from numpy.lib.recfunctions.join_by(). Redistribution license restrictions apply. N) OrderedDictCounter)SequenceTableMergeErrorc@s eZdZdS)rN)__name__ __module__ __qualname__r r 5lib/python3.9/site-packages/astropy/table/np_utils.pyrs{col_name}_{table_name}c stfddg}|dur4ddttD}tD]\}}||}|jjD]p}|vrx|vr|n@t} | |t fdd| Dr|j |d}||||<qTq$z"get_col_name_map..NcSsg|]}t|dqS)str).0Ziir r r (rz$get_col_name_map..c3s|]}|jjvVqdSr )dtypenames)rother)namer r 9rz#get_col_name_map..) table_nameZcol_namecSsg|]\}}|dkr|qSrr )rrcountr r r rArzgMerging column names resulted in duplicates: {}. Change uniq_col_name or table_names args to fix this.c3s|]}||fVqdSr r rr) col_name_mapr r rHr) collections defaultdictranger enumeraterrappendlistpopanyformatritemsrr) rZ common_namesZ uniq_col_nameZ table_namesZ col_name_listidxarrayrout_nameZothersZcol_name_countZrepeated_namesr )rrrr get_col_name_maps2    r,c Csg}|D]\}}ddt||D}dd|D}z t|}Wn<ty~}z$td|d|j|WYd}~n d}~00tdd|D} t| d krtd | } | t ||| 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|]\}}|dur||qSr r )rarrrr r r rYrzget_descrs..cSsg|]}|dur|qSr r rr r r r\rz,The '{}' columns have incompatible types: {}rNcss|]}|jddVqdS)rN)shapercolr r r rhrzget_descrs..rz Key columns have different shape) r(zip common_dtyperr'_incompat_typessetr r%r#fix_column_name) rrZ out_descrsr+Zin_namesZin_colsrrtmeZ uniq_shapesr.r r r get_descrsMs$   r7cstjtjtjtjtjftfdd|D}t|dkr`dd|D}td|}||_ |dd|D}|D]}|j j dvrrd |j |d <qrt d d|D}|j jS) 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 c3s$|]tfddDVqdS)c3s|]}tjj|VqdSr ) issubclassrtype)rZnp_typer0r r rzrz)common_dtype...N)tuple)rZnp_typesr:r rzszcommon_dtype..rcSsg|] }|jjqSr )rrr/r r r r~rz common_dtype..z Columns have incompatible types cSsg|]}tjd|jdqS)r)r)npemptyrr/r r r rr)SU0rcSsg|] }|dqS)rr )rr-r r r rr)r=Zbool_Zobject_Znumber characterZvoidr4r rr3rkinditemsizer*r)ZcolsZ uniq_typesZincompat_typesr6Zarrsr-Z arr_commonr r<r r2rs   r2cCsXd}t|tst||D]$}t|tjr6|jjdurt|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   rIcCs.|dur*z t|}Wnty(Yn0|S)z Fixes column names so that they are compatible with Numpy on Python 2. Raises a ValueError exception if the column name contains Unicode characters, which can not reasonably be used as a column name. N)rUnicodeEncodeError)valr r r r5s   r5)r N)__doc__rrrcollections.abcrZnumpyr=__all__rGrr,r7r2rIr5r r r r s  8%