Your IP : 216.73.216.213

Current Path : /opt/alt/python35/lib64/python3.5/site-packages/numpy/lib/__pycache__/
Upload File :
Current File : //opt/alt/python35/lib64/python3.5/site-packages/numpy/lib/__pycache__/recfunctions.cpython-35.pyc



(��^^��@s�dZddlmZmZmZddlZddlZddlZddl	j
Z
ddlmZmZddl	m
Z
ddlmZddlmZddlmZejdd	kr�dd
lmZej
jjZddd
dddddddddgZdd�Zdd�Zdd�Zdd�Zddd �Zddd!d�Zd"d#�Z d$d%�Z!dd&d'd(�Z"d&dd)d*�Z#dd+d,�Z$d=dddd.d�Z%d&dd/d�Z&d&dd0d1�Z'd2d�Z(d3d�Z)dd>d&dd4d�Z*dd5d�Z+dd&ddd6d�Z,dd&dd7d
�Z-d8d9d:dd&dd;d�Z.d8d9d:dd<d�Z/dS)?z�
Collection of utilities to manipulate structured arrays.

Most of these functions were initially implemented by John Hunter for
matplotlib.  They have been rewritten and extended for convenience.

�)�division�absolute_import�print_functionN)�ndarray�recarray)�MaskedArray)�
MaskedRecords)�_is_string_like)�
basestring�)�zip�
append_fields�drop_fields�find_duplicates�get_fieldstructure�join_by�merge_arrays�rec_append_fields�rec_drop_fields�rec_join�recursive_fill_fields�
rename_fields�stack_arrayscCs�|j}xq|jD]f}y||}Wntk
r>wYnX|jjr_t|||�q|||dt|��<qW|S)al
    Fills fields from output with fields from input,
    with support for nested structures.

    Parameters
    ----------
    input : ndarray
        Input array.
    output : ndarray
        Output array.

    Notes
    -----
    * `output` should be at least the same size as `input`

    Examples
    --------
    >>> from numpy.lib import recfunctions as rfn
    >>> a = np.array([(1, 10.), (2, 20.)], dtype=[('A', int), ('B', float)])
    >>> b = np.zeros((3,), dtype=a.dtype)
    >>> rfn.recursive_fill_fields(a, b)
    array([(1, 10.0), (2, 20.0), (0, 0.0)],
          dtype=[('A', '<i4'), ('B', '<f8')])

    N)�dtype�names�
ValueErrorr�len)�input�output�newdtype�field�current�r"�I/opt/alt/python35/lib64/python3.5/site-packages/numpy/lib/recfunctions.pyr"s	
cCsrg}|j}xP|D]H}||}|jrQ|j|tt|��f�q|j|�qWt|�pqdS)a�
    Returns the field names of the input datatype as a tuple.

    Parameters
    ----------
    adtype : dtype
        Input datatype

    Examples
    --------
    >>> from numpy.lib import recfunctions as rfn
    >>> rfn.get_names(np.empty((1,), dtype=int)) is None
    True
    >>> rfn.get_names(np.empty((1,), dtype=[('A',int), ('B', float)]))
    ('A', 'B')
    >>> adtype = np.dtype([('a', int), ('b', [('ba', int), ('bb', int)])])
    >>> rfn.get_names(adtype)
    ('a', ('b', ('ba', 'bb')))
    N)r�append�tuple�	get_names)�adtype�	listnamesr�namer!r"r"r#r&Is	

	"r&cCscg}|j}xA|D]9}|j|�||}|jr|jt|��qWt|�pbdS)a2
    Returns the field names of the input datatype as a tuple. Nested structure
    are flattend beforehand.

    Parameters
    ----------
    adtype : dtype
        Input datatype

    Examples
    --------
    >>> from numpy.lib import recfunctions as rfn
    >>> rfn.get_names_flat(np.empty((1,), dtype=int)) is None
    True
    >>> rfn.get_names_flat(np.empty((1,), dtype=[('A',int), ('B', float)]))
    ('A', 'B')
    >>> adtype = np.dtype([('a', int), ('b', [('ba', int), ('bb', int)])])
    >>> rfn.get_names_flat(adtype)
    ('a', 'b', 'ba', 'bb')
    N)rr$�extend�get_names_flatr%)r'r(rr)r!r"r"r#r+hs	


	r+cCs�|j}|dkr|jSg}xS|D]K}|j|\}}|jra|jt|��q)|j||f�q)Wt|�SdS)aD
    Flatten a structured data-type description.

    Examples
    --------
    >>> from numpy.lib import recfunctions as rfn
    >>> ndtype = np.dtype([('a', '<i4'), ('b', [('ba', '<f8'), ('bb', '<i4')])])
    >>> rfn.flatten_descr(ndtype)
    (('a', dtype('int32')), ('ba', dtype('float64')), ('bb', dtype('int32')))

    N)r�descr�fieldsr*�
flatten_descrr$r%)�ndtyperr,r �typ�_r"r"r#r.�s	
	r.FcCs�g}|r6x�|D]}|jt|j��qWndxa|D]Y}|j}|jpXf}t|�dkr�|jd|jf�q=|j|j�q=Wtj|�jS)z�
    Combine the dtype description of a series of arrays.

    Parameters
    ----------
    seqarrays : sequence of arrays
        Sequence of arrays
    flatten : {boolean}, optional
        Whether to collapse nested descriptions.
    ��)r*r.rrrr$r,�np)�	seqarrays�flattenr�ar!rr"r"r#�	zip_descr�s

	r8cCs�|dkri}|j}x�|D]�}||}|jrw|rQ|g||<n
g||<|jt|||��q"dd�|j|g�p�gD�}|r�|j|�n|r�|g}|p�g||<q"W|p�dS)ab
    Returns a dictionary with fields indexing lists of their parent fields.

    This function is used to simplify access to fields nested in other fields.

    Parameters
    ----------
    adtype : np.dtype
        Input datatype
    lastname : optional
        Last processed field name (used internally during recursion).
    parents : dictionary
        Dictionary of parent fields (used interbally during recursion).

    Examples
    --------
    >>> from numpy.lib import recfunctions as rfn
    >>> ndtype =  np.dtype([('A', int),
    ...                     ('B', [('BA', int),
    ...                            ('BB', [('BBA', int), ('BBB', int)])])])
    >>> rfn.get_fieldstructure(ndtype)
    ... # XXX: possible regression, order of BBA and BBB is swapped
    {'A': [], 'B': [], 'BA': ['B'], 'BB': ['B'], 'BBA': ['B', 'BB'], 'BBB': ['B', 'BB']}

    NcSsg|]}|�qSr"r")�.0r1r"r"r#�
<listcomp>�s	z&get_fieldstructure.<locals>.<listcomp>)r�updater�getr$)r'�lastname�parentsrr)r!Z
lastparentr"r"r#r�s"	

	
%	ccsQxJ|D]B}t|tj�rDx'tt|��D]}|Vq2Wq|VqWdS)zu
    Returns an iterator of concatenated fields from a sequence of arrays,
    collapsing any nested structure.

    N)�
isinstancer4�void�_izip_fields_flatr%)�iterable�element�fr"r"r#rA�s

rAccs�x�|D]�}t|d�rKt|t�rKxjt|�D]}|Vq9Wqt|tj�r�tt|��dkr�x!t|�D]}|Vq�Wq|VqWdS)zP
    Returns an iterator of concatenated fields from a sequence of arrays.

    �__iter__r2N)�hasattrr?r
�_izip_fieldsr4r@rr%)rBrCrDr"r"r#rG�s
*rGTccso|rt}nt}tjddkr4tj}n	tj}x+||d|�D]}t||��VqPWdS)a*
    Returns an iterator of concatenated items from a sequence of arrays.

    Parameters
    ----------
    seqarrays : sequence of arrays
        Sequence of arrays.
    fill_value : {None, integer}
        Value used to pad shorter iterables.
    flatten : {True, False},
        Whether to
    rr�	fillvalueN)rArG�sys�version_info�	itertools�zip_longest�izip_longestr%)r5�
fill_valuer6ZzipfuncrL�tupr"r"r#�izip_records	s		rPcCs[t|t�sd}|r3|rW|jt�}n$tj|�}|rW|jt�}|S)z�
    Private function: return a recarray, a ndarray, a MaskedArray
    or a MaskedRecords depending on the input parameters
    F)r?r�viewr�ma�filledr)r�usemask�
asrecarrayr"r"r#�_fix_output&srVcCsx|jj}|j|j|j}}}xH|p5ij�D]4\}}||kr<|||<|||||<q<W|S)zp
    Update the fill_value and masked data of `output`
    from the default given in a dictionary defaults.
    )rr�data�maskrN�items)r�defaultsrrWrXrN�k�vr"r"r#�
_fix_defaults7s
r]r2cCst|�dkr%tj|d�}t|ttjf�r�|j}|skt|fdd�|jkr�|j	�}|j
s�d|fg}|r�|r�t}q�t}n|r�t
}nt}|jd|d|�S|f}ndd	�|D�}td
d�|D��}t|�}t|d|�}	g}
g}|r�x1t||�D] \}}
||
}|j	�j�}tj|�j	�}|r$t||j�}t|ttjf�r0t|j�dkr�|j�d}d}q0tj|d|jdd�}tjdd|j�}nd
}d}|
jtj||g|��|jtj||g|��qPWtt|
d|��}tjtj|d|	d|�dtt|d|���}|r�|jt�}nx�t||�D]�\}}
||
}|j	�j�}|r�t||j�}t|ttjf�r�t|j�dkrp|j�d}q�tj|d|jdd�}nd
}|
jtj||g|��q�Wtjtt|
d|��d|	d|�}|r�|jt
�}|S)a�
    Merge arrays field by field.

    Parameters
    ----------
    seqarrays : sequence of ndarrays
        Sequence of arrays
    fill_value : {float}, optional
        Filling value used to pad missing data on the shorter arrays.
    flatten : {False, True}, optional
        Whether to collapse nested fields.
    usemask : {False, True}, optional
        Whether to return a masked array or not.
    asrecarray : {False, True}, optional
        Whether to return a recarray (MaskedRecords) or not.

    Examples
    --------
    >>> from numpy.lib import recfunctions as rfn
    >>> rfn.merge_arrays((np.array([1, 2]), np.array([10., 20., 30.])))
    masked_array(data = [(1, 10.0) (2, 20.0) (--, 30.0)],
                 mask = [(False, False) (False, False) (True, False)],
           fill_value = (999999, 1e+20),
                dtype = [('f0', '<i4'), ('f1', '<f8')])

    >>> rfn.merge_arrays((np.array([1, 2]), np.array([10., 20., 30.])),
    ...              usemask=False)
    array([(1, 10.0), (2, 20.0), (-1, 30.0)],
          dtype=[('f0', '<i4'), ('f1', '<f8')])
    >>> rfn.merge_arrays((np.array([1, 2]).view([('a', int)]),
    ...               np.array([10., 20., 30.])),
    ...              usemask=False, asrecarray=True)
    rec.array([(1, 10.0), (2, 20.0), (-1, 30.0)],
              dtype=[('a', '<i4'), ('f1', '<f8')])

    Notes
    -----
    * Without a mask, the missing value will be filled with something,
    * depending on what its corresponding type:
            -1      for integers
            -1.0    for floating point numbers
            '-'     for characters
            '-1'    for strings
            True    for boolean values
    * XXX: I just obtained these values empirically
    r2rr6Tr3r�typecSsg|]}tj|��qSr")r4�
asanyarray)r9�_mr"r"r#r:�s	z merge_arrays.<locals>.<listcomp>css|]}|jVqdS)N)�size)r9r7r"r"r#�	<genexpr>�szmerge_arrays.<locals>.<genexpr>ZndminN�countrX)r2)rr4r_r?rr@rr8r,�ravelrrrrrQr%�maxrZ	__array__rRZgetmaskarray�_check_fill_value�item�arrayZonesr$rK�chainrPZfromiter�list)r5rNr6rTrUZseqdtypeZseqtype�sizesZ	maxlengthrZseqdataZseqmaskr7�nZ	nbmissingrWrXZfvalZfmskrr"r"r#rEs|1					
	 $!
!$cs�t|�r|g}nt|�}�fdd���|j|�}|sRdStj|jd|�}t||�}t|d|d|�S)av
    Return a new array with fields in `drop_names` dropped.

    Nested fields are supported.

    Parameters
    ----------
    base : array
        Input array
    drop_names : string or sequence
        String or sequence of strings corresponding to the names of the
        fields to drop.
    usemask : {False, True}, optional
        Whether to return a masked array or not.
    asrecarray : string or sequence, optional
        Whether to return a recarray or a mrecarray (`asrecarray=True`) or
        a plain ndarray or masked array with flexible dtype. The default
        is False.

    Examples
    --------
    >>> from numpy.lib import recfunctions as rfn
    >>> a = np.array([(1, (2, 3.0)), (4, (5, 6.0))],
    ...   dtype=[('a', int), ('b', [('ba', float), ('bb', int)])])
    >>> rfn.drop_fields(a, 'a')
    array([((2.0, 3),), ((5.0, 6),)],
          dtype=[('b', [('ba', '<f8'), ('bb', '<i4')])])
    >>> rfn.drop_fields(a, 'ba')
    array([(1, (3,)), (4, (6,))],
          dtype=[('a', '<i4'), ('b', [('bb', '<i4')])])
    >>> rfn.drop_fields(a, ['ba', 'bb'])
    array([(1,), (4,)],
          dtype=[('a', '<i4')])
    cs�|j}g}xn|D]f}||}||kr5q|jri�||�}|r||j||f�q|j||f�qW|S)N)rr$)r/�
drop_namesrrr)r!r,)�_drop_descrr"r#rn�s	

	z drop_fields.<locals>._drop_descrNrrTrU)r	�setrr4�empty�shaperrV)�basermrTrUrrr")rnr#r�s#csV�fdd�|D�}tj�jd|�}t�|�}t|d|d|�S)a�
    Return a new array keeping only the fields in `keep_names`,
    and preserving the order of those fields.

    Parameters
    ----------
    base : array
        Input array
    keep_names : string or sequence
        String or sequence of strings corresponding to the names of the
        fields to keep. Order of the names will be preserved.
    usemask : {False, True}, optional
        Whether to return a masked array or not.
    asrecarray : string or sequence, optional
        Whether to return a recarray or a mrecarray (`asrecarray=True`) or
        a plain ndarray or masked array with flexible dtype. The default
        is False.
    cs#g|]}|�j|f�qSr")r)r9rl)rrr"r#r:!s	z _keep_fields.<locals>.<listcomp>rrTrU)r4rprqrrV)rrZ
keep_namesrTrUrrr")rrr#�_keep_fieldssrscCst||dddd�S)zK
    Returns a new numpy.recarray with fields in `drop_names` dropped.
    rTFrUT)r)rrrmr"r"r#r'scs1�fdd���|j|�}|j|�S)a�
    Rename the fields from a flexible-datatype ndarray or recarray.

    Nested fields are supported.

    Parameters
    ----------
    base : ndarray
        Input array whose fields must be modified.
    namemapper : dictionary
        Dictionary mapping old field names to their new version.

    Examples
    --------
    >>> from numpy.lib import recfunctions as rfn
    >>> a = np.array([(1, (2, [3.0, 30.])), (4, (5, [6.0, 60.]))],
    ...   dtype=[('a', int),('b', [('ba', float), ('bb', (float, 2))])])
    >>> rfn.rename_fields(a, {'a':'A', 'bb':'BB'})
    array([(1, (2.0, [3.0, 30.0])), (4, (5.0, [6.0, 60.0]))],
          dtype=[('A', '<i4'), ('b', [('ba', '<f8'), ('BB', '<f8', 2)])])

    csug}xh|jD]]}|j||�}||}|jrZ|j|�||�f�q|j||f�qW|S)N)rr<r$)r/�
namemapperrr)Znewnamer!)�_recursive_rename_fieldsr"r#ruEs
	z/rename_fields.<locals>._recursive_rename_fields)rrQ)rrrtrr")rur#r.sc	Cs�t|ttf�rBt|�t|�krcd}t|��n!t|t�rc|g}|g}|dkr�dd�|D�}dd�t||�D�}n�t|ttf�s�|g}t|�t|�krt|�dkr�|t|�}nd}t|��dd�t|||�D�}t|d	|d
|�}t|�dkrxt|ddd	|d
|�}n|j�}t	j
tt|�t|��d
|jj
|jj
�}t||�}t||�}t|d	|d|�S)a

    Add new fields to an existing array.

    The names of the fields are given with the `names` arguments,
    the corresponding values with the `data` arguments.
    If a single field is appended, `names`, `data` and `dtypes` do not have
    to be lists but just values.

    Parameters
    ----------
    base : array
        Input array to extend.
    names : string, sequence
        String or sequence of strings corresponding to the names
        of the new fields.
    data : array or sequence of arrays
        Array or sequence of arrays storing the fields to add to the base.
    dtypes : sequence of datatypes, optional
        Datatype or sequence of datatypes.
        If None, the datatypes are estimated from the `data`.
    fill_value : {float}, optional
        Filling value used to pad missing data on the shorter arrays.
    usemask : {False, True}, optional
        Whether to return a masked array or not.
    asrecarray : {False, True}, optional
        Whether to return a recarray (MaskedRecords) or not.

    z7The number of arrays does not match the number of namesNcSs+g|]!}tj|dddd��qS)�copyF�subokT)r4rh)r9r7r"r"r#r:}s	z!append_fields.<locals>.<listcomp>cSs1g|]'\}}|j||jfg��qSr")rQr)r9r)r7r"r"r#r:~s	r2z5The dtypes argument must be None, a dtype, or a list.c
SsLg|]B\}}}tj|ddddd|�j||fg��qS)rvFrwTr)r4rhrQ)r9r7rl�dr"r"r#r:�s	rTrNr6TrrU)r?r%rjrrr
rr�poprR�
masked_allrerr,rrV)	rrrrW�dtypesrNrTrU�msgrr"r"r#r
Us:				!cCs%t||d|d|dddd�S)aM
    Add new fields to an existing array.

    The names of the fields are given with the `names` arguments,
    the corresponding values with the `data` arguments.
    If a single field is appended, `names`, `data` and `dtypes` do not have
    to be lists but just values.

    Parameters
    ----------
    base : array
        Input array to extend.
    names : string, sequence
        String or sequence of strings corresponding to the names
        of the new fields.
    data : array or sequence of arrays
        Array or sequence of arrays storing the fields to add to the base.
    dtypes : sequence of datatypes, optional
        Datatype or sequence of datatypes.
        If None, the datatypes are estimated from the `data`.

    See Also
    --------
    append_fields

    Returns
    -------
    appended_array : np.recarray
    rWr{rUTrTF)r
)rrrrWr{r"r"r#r�scCs�t|t�r|St|�dkr-|dSdd�|D�}dd�|D�}dd�|D�}dd�|D�}|d}	|	j}
dd�|
D�}x|dd	�D]�}x�|jD]�}
|
dp�d
}||kr�|
j|
�|j|�q�|j|�}|
|}|rqtj|
d�tj|d�kr�t|�}|
d|d<t	|�|
|<q�|
d|dkr�t
dt|
�||
df��q�Wq�Wt|
�dkr�tj
|�}n�tjtj|�f|
�}tjtjd|f�}g}x�t|||d	d�|dd	��D]�\}}}}|jj}|d	kr�||dt|�||�<q?x?|D]7}||||||�<||kr�|j|�q�Wq?Wtt||�d
|d|�S)a�
    Superposes arrays fields by fields

    Parameters
    ----------
    arrays : array or sequence
        Sequence of input arrays.
    defaults : dictionary, optional
        Dictionary mapping field names to the corresponding default values.
    usemask : {True, False}, optional
        Whether to return a MaskedArray (or MaskedRecords is
        `asrecarray==True`) or a ndarray.
    asrecarray : {False, True}, optional
        Whether to return a recarray (or MaskedRecords if `usemask==True`)
        or just a flexible-type ndarray.
    autoconvert : {False, True}, optional
        Whether automatically cast the type of the field to the maximum.

    Examples
    --------
    >>> from numpy.lib import recfunctions as rfn
    >>> x = np.array([1, 2,])
    >>> rfn.stack_arrays(x) is x
    True
    >>> z = np.array([('A', 1), ('B', 2)], dtype=[('A', '|S3'), ('B', float)])
    >>> zz = np.array([('a', 10., 100.), ('b', 20., 200.), ('c', 30., 300.)],
    ...   dtype=[('A', '|S3'), ('B', float), ('C', float)])
    >>> test = rfn.stack_arrays((z,zz))
    >>> test
    masked_array(data = [('A', 1.0, --) ('B', 2.0, --) ('a', 10.0, 100.0) ('b', 20.0, 200.0)
     ('c', 30.0, 300.0)],
                 mask = [(False, False, True) (False, False, True) (False, False, False)
     (False, False, False) (False, False, False)],
           fill_value = ('N/A', 1e+20, 1e+20),
                dtype = [('A', '|S3'), ('B', '<f8'), ('C', '<f8')])

    r2rcSs%g|]}tj|�j��qSr")r4r_rd)r9r7r"r"r#r:�s	z stack_arrays.<locals>.<listcomp>cSsg|]}t|��qSr")r)r9r7r"r"r#r:�s	cSsg|]}|j�qSr")r)r9r7r"r"r#r:�s	cSsg|]}|j�qSr")r)r9rxr"r"r#r:�s	cSsg|]}|d�qS)rr")r9r1r"r"r#r:�s	Nr3zIncompatible type '%s' <> '%s'zf%irTrU���r}r}r})r?rrr,r$�indexr4rrjr%�	TypeError�dictrR�concatenaterz�sumZcumsumZr_rrrVr])ZarraysrZrTrUZautoconvertr5Znrecordsr/ZfldnamesZdtype_lZnewdescrrZdtype_nr,r)�nameidxZ
current_descrr�offset�seenr7rl�i�jr"r"r#r�sT'
	

&&<!
c
Cs(tj|�j�}t|j�}|}|rYx||D]}||}q;W||}|j�}||}|j�}	|	dd�|	dd�k}
|r�|j}d|
|dd�<tjdg|
f�}
|
dd�|
dd�|
dd�<|||
}|r |||
fS|SdS)aa
    Find the duplicates in a structured array along a given key

    Parameters
    ----------
    a : array-like
        Input array
    key : {string, None}, optional
        Name of the fields along which to check the duplicates.
        If None, the search is performed by records
    ignoremask : {True, False}, optional
        Whether masked data should be discarded or considered as duplicates.
    return_index : {False, True}, optional
        Whether to return the indices of the duplicated values.

    Examples
    --------
    >>> from numpy.lib import recfunctions as rfn
    >>> ndtype = [('a', int)]
    >>> a = np.ma.array([1, 1, 1, 2, 2, 3, 3],
    ...         mask=[0, 0, 1, 0, 0, 0, 1]).view(ndtype)
    >>> rfn.find_duplicates(a, ignoremask=True, return_index=True)
    ... # XXX: judging by the output, the ignoremask flag has no effect
    Nr2Fr}r}r})	r4r_rdrr�argsortrSZ
recordmaskr�)
r7�keyZ
ignoremaskZreturn_indexr-rrrDZsortidxZ
sortedbaseZ
sorteddata�flagZ
sortedmask�
duplicatesr"r"r#rs(

 	(�inner�1�2c	)sY|dkrtd|��t�t�r4�f�tt���t��kr�t�fdd�t��D��}	td|	��xR�D]J}
|
|jjkr�td|
��|
|jjkr�td	|
��q�W|j	�}|j	�}t|�}|jj|jj}}
tj
t|�t|
��j��ri|pI|rid
}|d7}t|���fdd
�|D�}t||�}t||�}t
j||f�}|jd��}||}t
jdg|dd�|dd�kf�}|dd�|dd�|dd�<||}|||k}|||k|}t|�t|�}}|dkr�d\}}n�|dkr
||}tj||||kf�}tj||||k|f�}t|�|t|�|}}nM|dkrW||}tj||||kf�}t|�|d}}||||}}dd
�|jjD�}|j�fdd�|jjD��tdd�|D��} x�|jjD]�}!t|!�}!|!d}
|
| kro|j|!�}"||"}#|
�kr<t|!d|#d�|#d <q�|#d|7<|!d|7<|j|"d|!�q�| j|!d�|j|!�q�Wdd
�|D�}t||�}$t
j|$||fd|�}%|%jj} x�|D]�}&||&}'|&| ks!|&|
kr+|r+|&�kr+|&|7}&|%|&}#|'d|�|#d|�<|d!kr�|'|d�|#|$|$|�<q�Wx�|
D]�}&||&}'|&| ks�|&|kr�|r�|&�kr�|&|7}&|%|&}#|'d|�|#d|�<|dkr�|r�|'|d�|#|d�<q�W|%jd��td|d|�}(tt|%|�|(�S)"a<
    Join arrays `r1` and `r2` on key `key`.

    The key should be either a string or a sequence of string corresponding
    to the fields used to join the array.  An exception is raised if the
    `key` field cannot be found in the two input arrays.  Neither `r1` nor
    `r2` should have any duplicates along `key`: the presence of duplicates
    will make the output quite unreliable. Note that duplicates are not
    looked for by the algorithm.

    Parameters
    ----------
    key : {string, sequence}
        A string or a sequence of strings corresponding to the fields used
        for comparison.
    r1, r2 : arrays
        Structured arrays.
    jointype : {'inner', 'outer', 'leftouter'}, optional
        If 'inner', returns the elements common to both r1 and r2.
        If 'outer', returns the common elements as well as the elements of
        r1 not in r2 and the elements of not in r2.
        If 'leftouter', returns the common elements and the elements of r1
        not in r2.
    r1postfix : string, optional
        String appended to the names of the fields of r1 that are present
        in r2 but absent of the key.
    r2postfix : string, optional
        String appended to the names of the fields of r2 that are present
        in r1 but absent of the key.
    defaults : {dictionary}, optional
        Dictionary mapping field names to the corresponding default values.
    usemask : {True, False}, optional
        Whether to return a MaskedArray (or MaskedRecords is
        `asrecarray==True`) or a ndarray.
    asrecarray : {False, True}, optional
        Whether to return a recarray (or MaskedRecords if `usemask==True`)
        or just a flexible-type ndarray.

    Notes
    -----
    * The output is sorted along the key.
    * A temporary array is formed by dropping the fields not in the key for
      the two arrays and concatenating the result. This array is then
      sorted, and the common entries selected. The output is constructed by
      filling the fields with the selected entries. Matching is not
      preserved if there are some duplicates...

    r��outer�	leftouterzWThe 'jointype' argument should be in 'inner', 'outer' or 'leftouter' (got '%s' instead)c3s5|]+\}}|�|dd�kr|VqdS)r2Nr")r9rl�x)r�r"r#rb�szjoin_by.<locals>.<genexpr>zduplicate join key %rzr1 does not have key field %rzr2 does not have key field %rz8r1 and r2 contain common names, r1postfix and r2postfix zcan't be emptycs"g|]}|�kr|�qSr"r")r9rl)r�r"r#r:�s	zjoin_by.<locals>.<listcomp>�orderFr2NrcSsg|]}t|��qSr")rj)r9r1r"r"r#r:�s	c3s+|]!}|d�krt|�VqdS)rN)rj)r9r1)r�r"r#rb�scss|]}|dVqdS)rNr")r9r1r"r"r#rb�scSsg|]}t|��qSr")r%)r9r1r"r"r#r:�s	rrTrU)r�r�r�r}r}r})rrr}r})r�r�)rr?r
rro�next�	enumeraterrrd�intersection�
differencersrRr�r�r4r,r*rjr~re�insertr$rz�sortr�rVr]))r��r1�r2�jointype�	r1postfix�	r2postfixrZrTrU�dupr)Znb1Zr1namesZr2namesr|�key1Zr1kZr2kZauxZidx_sortZflag_inZidx_inZidx_1Zidx_2Zr1cmnZr2cmnZr1spcZr2spcZidx_out�s1�s2r/r�descr�r!ZcmnrrD�selected�kwargsr")r�r#rLs�3
	%
'


2(
#$&

 

+

"

+

c
Cs@td|d|d|d|dddd�}t||||�S)	z�
    Join arrays `r1` and `r2` on keys.
    Alternative to join_by, that always returns a np.recarray.

    See Also
    --------
    join_by : equivalent function
    r�r�r�rZrTFrUT)r�r)r�r�r�r�r�r�rZr�r"r"r#r�s
r}r})0�__doc__�
__future__rrrrIrK�numpyr4Znumpy.marRrrrZnumpy.ma.mrecordsrZnumpy.lib._iotoolsr	�numpy.compatr
rJZfuture_builtinsr�corerf�__all__rr&r+r.r8rrArGrPrVr]rrrsrrr
rrrrrr"r"r"r#�<module>sX			'/�@'D"	Y6	�