σ šίΘ[c@`s dZddlmZmZmZmZddlmZddlm Z m Z ddl m Z ddl Z ddl mZmZddlZddljZd d lmZd gZd efd „ƒYZddd„Zd„Zd„Zd„Zd„Zd„ZdS(u§ High-level operations for numpy structured arrays. Some code and inspiration taken from numpy.lib.recfunctions.join_by(). Redistribution license restrictions apply. i(tabsolute_importtdivisiontprint_functiontunicode_literalsi(tsix(tziptrange(tchainN(t OrderedDicttCounteri(t _np_utilsuTableMergeErrortTableMergeErrorcB`seZRS((t__name__t __module__(((s5lib/python2.7/site-packages/astropy/table/np_utils.pyR su{col_name}_{table_name}c`s·tj‡fd†ƒ‰g}|dkr_gttˆƒƒD]}tj|dƒ^q=}nxΧtˆƒD]Ι\}}||}x°|jj D]’‰ˆ} ˆ|krƈ|kr#|j ˆƒq#n]t ˆƒ} | j |ƒt ‡fd†| Dƒƒr|jd|dˆƒ} n|j | ƒˆˆ| |+sic3`s!|]}ˆ|jjkVqdS(N(tdtypetnames(t.0tother(tname(s5lib/python2.7/site-packages/astropy/table/np_utils.pys @st table_nametcol_nameuhMerging column names resulted in duplicates: {0}. Change uniq_col_name or table_names args to fix this.c3`s|]}|ˆ|fVqdS(N((RR(t col_name_map(s5lib/python2.7/site-packages/astropy/table/np_utils.pys OsN(t collectionst defaultdictRRRRt text_typet enumerateRRtappendtlisttpoptanytformatR t iteritemsR R(Rt common_namest uniq_col_namet table_namest col_name_listtiitidxtarrayRtout_nametotherstcol_name_counttcounttrepeated_names((RRRs5lib/python2.7/site-packages/astropy/table/np_utils.pytget_col_name_maps0 5       4 c C`s4g}x'tj|ƒD]\}}gt||ƒD]"\}}|dk r2||^q2}g|D]}|dk ra|^qa}yt|ƒ} Wn2tk rΓ} tdj|d| jƒƒ‚nXtd„|Dƒƒ} t | ƒdkrtdj|ƒƒ‚n| j ƒ} |j t |ƒ| | fƒqW|S(u΄ 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. u.The '{0}' columns have incompatible types: {1}ics`s|]}|jdVqdS(iN(tshape(Rtcol((s5lib/python2.7/site-packages/astropy/table/np_utils.pys osiu&Key columns {0!r} have different shapeN( RR#RRt common_dtypeR R"t_incompat_typestsetRR Rtfix_column_name( RRt out_descrsR+tin_namestarrRtin_colsRRttmet uniq_shapesR1((s5lib/python2.7/site-packages/astropy/table/np_utils.pyt get_descrsTs8%   c`s*tjtjtjtjtjf‰t‡fd†|Dƒƒ}t|ƒdkr˜g|D]}|jj ^qY}t dj |ƒƒ}||_ |‚ng|D]}tj dd|jƒ^qŸ}x4|D],}|jjd krΚd|j|dsN(ttuple(R(tnp_types(R2s5lib/python2.7/site-packages/astropy/table/np_utils.pys siu#Columns have incompatible types {0}RuSuUu0i(uSuU(tnptbool_tobject_tnumbert charactertvoidR5RRRR R"R4temptytkindtitemsizeR*tstr(tcolst uniq_typesR2tincompat_typesR;tarrsR9t arr_common((RBs5lib/python2.7/site-packages/astropy/table/np_utils.pyR3ys$     + &cC`s‘d}t|tjƒs't|ƒ‚nxB|D]:}t|tjƒ sY|jjdkr.t|ƒ‚q.q.Wt |ƒdkrt dƒ‚ndS(Nu@`arrays` arg must be a sequence (e.g. list) of structured arraysiu,`arrays` arg must include at least one array( t isinstanceRtSequencet TypeErrorRCtndarrayRRRRt ValueError(RterrR*((s5lib/python2.7/site-packages/astropy/table/np_utils.pyt(_check_for_sequence_of_structured_arrays—s %cC`sR|dk rNyt|ƒ}WqNtk rJtjrDtdƒ‚n‚qNXn|S(uή 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. u<Column names must not contain Unicode characters on Python 2N(RRLtUnicodeEncodeErrorRtPY2RV(tval((s5lib/python2.7/site-packages/astropy/table/np_utils.pyR6£s     cC`s¬t|dƒ}tj|dtƒ}gt|ƒD]%}tj|d|fjƒƒ^q2}g}x!|D]}|j|jjƒqjWdj |ƒ}tj j |d|ƒS(uT Partial replacement for `~numpy.core.records.fromrecords` which includes a workaround for the bug with unicode arrays described at: https://github.com/astropy/astropy/issues/3052 This should not serve as a full replacement for the original function; this only does enough to fulfill the needs of the table module. iR.u,tformats( RRCR*tobjectRttolistRRRLtjointrect fromarrays(trec_listtnfieldstobjtit array_listR\((s5lib/python2.7/site-packages/astropy/table/np_utils.pytrecarray_fromrecords·s 8 ( t__doc__t __future__RRRRtexternRtextern.six.movesRRt itertoolsRRRR tnumpyRCtnumpy.matmatR t__all__RVR RR0R=R3RXR6Rg(((s5lib/python2.7/site-packages/astropy/table/np_utils.pyts$"    7 %