Ñò
\ÐKc @ sø d Z d d k Z d d k l Z l Z l Z d d k Z d d k l
Z
l Z d d k i
Z
d d k l Z d d k l Z d d k l Z e i
i i Z d d d
d d d
d d d d d d g Z d „ Z d „ Z d „ Z d „ Z e d „ Z d d d „ Z d „ Z d „ Z d e d „ Z! e e d „ Z" d d „ Z# d e e e d „ Z$ e e d „ Z% d! „ Z& d" „ Z' d d d e e d# „ Z( d d$ „ Z) d e e e d% „ Z* d e e d& „ Z+ d' d( d) d e e d* „ Z, d' d( d) d d+ „ Z- d S(, sÆ
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.
iÿÿÿÿN( t chaint repeatt izip( t ndarrayt recarray( t MaskedArray( t
MaskedRecords( t _is_string_liket
append_fieldst drop_fieldst find_duplicatest get_fieldstructuret join_byt merge_arrayst rec_append_fieldst rec_drop_fieldst rec_joint recursive_fill_fieldst
rename_fieldst stack_arraysc C s~ | i } xn | i D]c } y | | } Wn t j
o
q n X| i i o t | | | ƒ q | | | t | ƒ *q W| S( s*
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
--------
>>> a = np.array([(1, 10.), (2, 20.)], dtype=[('A', int), ('B', float)])
>>> b = np.zeros((3,), dtype=a.dtype)
>>> recursive_fill_fields(a, b)
np.array([(1, 10.), (2, 20.), (0, 0.)], dtype=[('A', int), ('B', float)])
( t dtypet namest
ValueErrorR t len( t inputt outputt newdtypet fieldt current( ( s< /usr/lib64/python2.6/site-packages/numpy/lib/recfunctions.pyR # s
c C su g } | i } xR | D]J } | | } | i o# | i | t t | ƒ ƒ f ƒ q | i | ƒ q Wt | ƒ p d S( s»
Returns the field names of the input datatype as a tuple.
Parameters
----------
adtype : dtype
Input datatype
Examples
--------
>>> get_names(np.empty((1,), dtype=int)) is None
True
>>> get_names(np.empty((1,), dtype=[('A',int), ('B', float)]))
('A', 'B')
>>> adtype = np.dtype([('a', int), ('b', [('ba', int), ('bb', int)])])
>>> get_names(adtype)
('a', ('b', ('ba', 'bb')))
N( R t appendt tuplet get_namest None( t adtypet listnamesR t nameR ( ( s< /usr/lib64/python2.6/site-packages/numpy/lib/recfunctions.pyR I s
#c C si g } | i } xF | D]> } | i | ƒ | | } | i o | i t | ƒ ƒ q q Wt | ƒ p d S( sô
Returns the field names of the input datatype as a tuple. Nested structure
are flattend beforehand.
Parameters
----------
adtype : dtype
Input datatype
Examples
--------
>>> get_names_flat(np.empty((1,), dtype=int)) is None
True
>>> get_names_flat(np.empty((1,), dtype=[('A',int), ('B', float)]))
('A', 'B')
>>> adtype = np.dtype([('a', int), ('b', [('ba', int), ('bb', int)])])
>>> get_names_flat(adtype)
('a', 'b', 'ba', 'bb')
N( R R t extendt get_names_flatR R ( R! R" R R# R ( ( s< /usr/lib64/python2.6/site-packages/numpy/lib/recfunctions.pyR% g s
c C sŠ | i } | d j o | i Sg } xU | D]M } | i | \ } } | i o | i t | ƒ ƒ q+ | i | | f ƒ q+ Wt | ƒ Sd S( s
Flatten a structured data-type description.
Examples
--------
>>> ndtype = np.dtype([('a', '>> flatten_descr(ndtype)
(('a', dtype('int32')), ('ba', dtype('float64')), ('bb', dtype('int32')))
N( R R t descrt fieldsR$ t
flatten_descrR R ( t ndtypeR R& R t typt _( ( s< /usr/lib64/python2.6/site-packages/numpy/lib/recfunctions.pyR( … s
c C s¯ g } | o+ x | D] } | i t | i ƒ ƒ q Wnh xd | D]\ } | i } | i p d } t | ƒ d j o | i d | i f ƒ q? | i | i ƒ q? Wt i | ƒ i S( sî
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.
i t ( ( R$ R( R R R R R& t np( t seqarrayst flattenR t aR R ( ( s< /usr/lib64/python2.6/site-packages/numpy/lib/recfunctions.pyt zip_descrž s c C s | d j o
h } n | i } xÒ | D]Ê } | | } | i o? | o | g | | >> ndtype = np.dtype([('A', int),
... ('B', [('BA', int),
... ('BB', [('BBA', int), ('BBB', int)])])])
>>> get_fieldstructure(ndtype)
{'A': [], 'B': [], 'BA': ['B'], 'BB': ['B'],
'BBA': ['B', 'BB'], 'BBB': ['B', 'BB']}
N( R R t updateR t getR ( R! t lastnamet parentsR R# R t _[1]R+ t
lastparent( ( s< /usr/lib64/python2.6/site-packages/numpy/lib/recfunctions.pyR ¸ s$
4
c c sS xL | D]D } t | t i ƒ o& x( t t | ƒ ƒ D] } | Vq3 Wq | Vq Wd S( st
Returns an iterator of concatenated fields from a sequence of arrays,
collapsing any nested structure.
N( t
isinstanceR- t voidt _izip_fields_flatR ( t iterablet elementt f( ( s< /usr/lib64/python2.6/site-packages/numpy/lib/recfunctions.pyR: ç s
c c s§ x | D]˜ } t | d ƒ o1 t | t ƒ o xn t | ƒ D] } | Vq; Wq t | t i ƒ o9 t t | ƒ ƒ d j o x" t | ƒ D] } | Vq‡ Wq | Vq Wd S( sO
Returns an iterator of concatenated fields from a sequence of arrays.
t __iter__i N( t hasattrR8 t
basestringt _izip_fieldsR- R9 R R ( R; R< R= ( ( s< /usr/lib64/python2.6/site-packages/numpy/lib/recfunctions.pyRA ô s !
,
c
c s¹ | g t | ƒ d i d „ } t | ƒ } g } | D] } | t | | ƒ | ƒ q7 ~ } | o
t } n t } y, x% t | Œ D] } t | | ƒ ƒ Vqƒ WWn t j
o n Xd S( s+
Returns an iterator of concatenated items from a sequence of arrays.
Parameters
----------
seqarray : sequence of arrays
Sequence of arrays.
fill_value : {None, integer}
Value used to pad shorter iterables.
flatten : {True, False},
Whether to
i c s s | ƒ Vd S( s* Yields the fill_value or raises IndexErrorN( ( t counter( ( s< /usr/lib64/python2.6/site-packages/numpy/lib/recfunctions.pyt sentinel s N( R t popt
iterrepeatt iterchainR: RA t iterizipR t
IndexError(
R. t
fill_valueR/ RC t fillersR6 t itt iterst zipfunct tup( ( s< /usr/lib64/python2.6/site-packages/numpy/lib/recfunctions.pyt izip_records s 0
c C sl t | t ƒ p
t } n | o | o | i t ƒ } qh n* t i | ƒ } | o | i t ƒ } n | S( s€
Private function: return a recarray, a ndarray, a MaskedArray
or a MaskedRecords depending on the input parameters
( R8 R t Falset viewR t mat filledR ( R t usemaskt
asrecarray( ( s< /usr/lib64/python2.6/site-packages/numpy/lib/recfunctions.pyt _fix_output$ s
c C s~ | i i } | i | i | i } } } xN | p h i ƒ D]9 \ } } | | j o | | | <| | | | | >> 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', '>> merge_arrays((np.array([1, 2]), np.array([10., 20., 30.])),
... usemask=False)
array(data = [(1, 10.0) (2, 20.0) (-1, 30.0)],
dtype=[('f0', '>> merge_arrays((np.array([1, 2]).view([('a', int)]),
np.array([10., 20., 30.])),
usemask=False, asrecarray=True)
rec.array(data = [(1, 10.0) (2, 20.0) (-1, 30.0)],
dtype=[('a', int), ('f1', ' s R RI ( i ( R R8 R R R1 t TrueR& t ravelR RQ R R R RR t getdatat getmaskarrayt _check_fill_valuet maxt enumeratet zipR- R9 t onest itemt arrayRF RO R t fromitert listt _mask( R. RI R/ RT RU t seqdtypeR R6 R0 t seqdatat _[2]t seqmaskt _[3]t maxlengtht it mt fvalt fmskt nbmissingRW RX R ( ( s< /usr/lib64/python2.6/site-packages/numpy/lib/recfunctions.pyR
C s^ #
00-
# c s t | ƒ o
| g } n
t | ƒ } ‡ f d † ‰ ˆ | i | ƒ } | p d St i | i d | ƒ} t | | ƒ } t | d | d | ƒS( s2
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
Whether to return a recarray or a mrecarray (`asrecarray=True`) or
a plain ndarray or masked array with flexible dtype (`asrecarray=False`)
Examples
--------
>>> a = np.array([(1, (2, 3.0)), (4, (5, 6.0))],
dtype=[('a', int), ('b', [('ba', float), ('bb', int)])])
>>> drop_fields(a, 'a')
array([((2.0, 3),), ((5.0, 6),)],
dtype=[('b', [('ba', '>> drop_fields(a, 'ba')
array([(1, (3,)), (4, (6,))],
dtype=[('a', '>> drop_fields(a, ['ba', 'bb'])
array([(1,), (4,)],
dtype=[('a', '>> a = np.array([(1, (2, [3.0, 30.])), (4, (5, [6.0, 60.]))],
dtype=[('a', int),
('b', [('ba', float), ('bb', (float, 2))])])
>>> rename_fields(a, {'a':'A', 'bb':'BB'})
array([(1, (2.0, 3)), (4, (5.0, 6))],
dtype=[('A', ' i s; The dtypes argument must be None, a single dtype or a list.R RT RI R/ RU N( R8 R Rk R R R@ R R- Ri RP R_ Rf RQ R R? R
RD RR t
masked_allRd R& R RV ( R} R RW t dtypesRI RT RU t err_msgR6 R0 Ro R# t msgRq t nt dR ( ( s< /usr/lib64/python2.6/site-packages/numpy/lib/recfunctions.pyR s:
6I `
!c C s% t | | d | d | d t d t ƒS( sU
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
RW R„ RU RT ( R R_ RP ( R} R RW R„ ( ( s< /usr/lib64/python2.6/site-packages/numpy/lib/recfunctions.pyR T s c C sX t | t ƒ o | St | ƒ d j o | d Sg } | D] } | t i | ƒ i ƒ q< ~ } g } | D] } | t | ƒ ql ~ } g }
| D] } |
| i q“ ~
} g } | D] }
| |
i q· ~ } | d } | i } g } | D] } | | d qî ~ } x| d D] } x| i D]ø } | d p d } | | j o | i | ƒ | i | ƒ q#| i
| ƒ } | | } | oY t i | d ƒ t i | d ƒ j o. t | ƒ } | d | d >> x = np.array([1, 2,])
>>> 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 = stack_arrays((z,zz))
>>> 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', ' '%s's f%iRT RU N( R8 R R R- t
asanyarrayR` R R R& R t indexRk R t TypeErrort dictRR t concatenateRƒ t sumt cumsumt r_Rf R RV R] ( t arraysRZ RT RU t autoconvertR6 R0 R. Ro t nrecordsRq R) t _[4]Rˆ t fldnamest dtype_lt newdescrt _[5]R+ R t dtype_nR& R# t nameidxt
current_descrR t offsett seenR‡ Rs t j( ( s< /usr/lib64/python2.6/site-packages/numpy/lib/recfunctions.pyR w s\ % 0'$$
%
' *
c
C s t i | ƒ i ƒ } t | i ƒ } | } | o- x | | D] } | | } q< W| | } n | i ƒ } | | } | i ƒ } | d | d j }
| o | i } t |
| d >> ndtype = [('a', int)]
>>> a = ma.array([1, 1, 1, 2, 2, 3, 3],
... mask=[0, 0, 1, 0, 0, 0, 1]).view(ndtype)
>>> find_duplicates(a, ignoremask=True, return_index=True)
iÿÿÿÿi N(
R- R‰ R` R R t argsortRS t
recordmaskRP R (
R0 t keyt
ignoremaskt return_indexR' R} R= t sortidxt
sortedbaset
sorteddatat flagt
sortedmaskt
duplicates( ( s< /usr/lib64/python2.6/site-packages/numpy/lib/recfunctions.pyR
Ð s*
t innert 1t 2c - sÕ | d j o t d | ƒ ‚ n t ˆ t ƒ o
ˆ f ‰ n x\ ˆ D]T } | | i i j o t d | ƒ ‚ n | | i i j o t d | ƒ ‚ qE qE W| i ƒ } | i ƒ } t | ƒ t | ƒ }
} | i i | i i } }
t | g } | D] } | ˆ j o | | qø qø ~ ƒ } t | g } |
D] } | ˆ j o | | q3q3~ ƒ } t i | | f ƒ } | i
d ˆ ƒ } | | } t i t g | d | d j f ƒ } | d | d | d *| | } | | |
j } | | |
j |
} t | ƒ t | ƒ } } | d j o d \ } } nÒ | d j or | | } t i | | | |
j f ƒ } t i | | | |
j |
f ƒ } t | ƒ | t | ƒ | } } nS | d j oE | | } t i | | | |
j f ƒ } t | ƒ | d
} } n | | | | } } g } | i i
D] }! | t |! ƒ q%~ }" |" i ‡ f d † | i i
Dƒ ƒ t d „ |" Dƒ ƒ }# xÒ | i i
D]Ä }$ t |$ ƒ }$ |$ d
} | |# j o} |# i | ƒ }% |" |% }& | ˆ j o t |$ d |& d ƒ |& d
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ª t outert leftoutersW The 'jointype' argument should be in 'inner', 'outer' or 'leftouter' (got '%s' instead)s r1 does not have key field %ss r2 does not have key field %st orderi iÿÿÿÿi c 3 s4 x- | ]& } | d ˆ j o t | ƒ Vq q Wd S( i N( Rk ( R^ R+ ( R¡ ( s< /usr/lib64/python2.6/site-packages/numpy/lib/recfunctions.pys o s c s s x | ] } | d Vq Wd S( i N( ( R^ R+ ( ( s< /usr/lib64/python2.6/site-packages/numpy/lib/recfunctions.pys q s R RT RU ( Rª R R® ( i i ( R R® ( R R8 R@ R R R` R R RR R RŸ RP R- R& Rk R$ RŠ Rd t insertR R Rƒ t sortRŒ RV R] (- R¡ t r1t r2t jointypet r1postfixt r2postfixRZ RT RU R# t nb1t nb2t r1namest r2namesR6 R‡ t r1kRo t r2kt auxt idx_sortt flag_int idx_int idx_1t idx_2t r1cmnt r2cmnt r1spct r2spct idx_outt s1t s2Rq R+ R) R t descRš R R” t cmnR R= t selectedt kwargs( ( R¡ s< /usr/lib64/python2.6/site-packages/numpy/lib/recfunctions.pyR s 4
;;
&
#%
-#
'
c
C s@ t d | d | d | d | d t d t ƒ } t | | | | S( s¨
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¶ RZ RT RU ( RŒ RP R_ R ( R¡ R² R³ R´ Rµ R¶ RZ RÍ ( ( s< /usr/lib64/python2.6/site-packages/numpy/lib/recfunctions.pyR ¢ s
(. t __doc__t itertoolsR RF R RE R RG t numpyR- R R t numpy.maRR R t numpy.ma.mrecordsR t numpy.lib._iotoolsR t coreRc t __all__R R R% R( RP R1 R R R: RA R_ RO RV R] R
R R R R R R R
R R ( ( ( s< /usr/lib64/python2.6/site-packages/numpy/lib/recfunctions.pyt sV & /
! ^> &E # X5 œ