Ñò
\ÐKc
@ s÷ d d d d d d d d d d d
d d g
Z d
d k Z d
d k i i Z d
d k l Z l Z l Z l Z l
Z
l Z d
d k l
Z
d
d k Z d
d k Z d
d k i Z d
d k l Z e i Z d „ Z d „ Z d e f d „ ƒ YZ e d e ƒ Z e d e ƒ Z d e _ d e _ d e f d „ ƒ YZ d e f d „ ƒ YZ e ƒ Z! d e f d „ ƒ YZ" e" ƒ Z# d e f d „ ƒ YZ$ d e f d „ ƒ YZ% d e f d „ ƒ YZ& e& d! e ƒ Z' e& d! e ƒ Z( d" „ Z) d# d$ „ Z* d% „ Z+ d S(& t
unravel_indext mgridt ogridt r_t c_t s_t index_expt ix_t ndenumeratet ndindext
fill_diagonalt diag_indicest diag_indices_fromiÿÿÿÿN( t asarrayt
ScalarTypet arrayt alltruet cumprodt arange( t find_common_type( t diffc C sŽ | t i | ƒ d j p
| d j o t d ƒ ‚ n t i | ƒ } t i d g t | ƒ d d d … ƒ d d d … } t | | | ƒ S( s°
Convert a flat index to an index tuple for an array of given shape.
Parameters
----------
x : int
Flattened index.
dims : tuple of ints
Input shape, the shape of an array into which indexing is
required.
Returns
-------
idx : tuple of ints
Tuple of the same shape as `dims`, containing the unraveled index.
Notes
-----
In the Examples section, since ``arr.flat[x] == arr.max()`` it may be
easier to use flattened indexing than to re-map the index to a tuple.
Examples
--------
>>> arr = np.arange(20).reshape(5, 4)
>>> arr
array([[ 0, 1, 2, 3],
[ 4, 5, 6, 7],
[ 8, 9, 10, 11],
[12, 13, 14, 15],
[16, 17, 18, 19]])
>>> x = arr.argmax()
>>> x
19
>>> dims = arr.shape
>>> idx = np.unravel_index(x, dims)
>>> idx
(4, 3)
>>> arr[idx] == arr.max()
True
i i s4 Invalid index, must be 0 <= x <= number of elements.Niÿÿÿÿ( t _nxt prodt
ValueErrort
empty_likeR t listt tuple( t xt dimst idxt dim_prod( ( s< /usr/lib64/python2.6/site-packages/numpy/lib/index_tricks.pyR s
*' 6c G sÙ g } t | ƒ } d g | } x t | ƒ D]Ÿ } t i | | ƒ } | i d j o
t d ‚ n t | i i t i ƒ o | i
ƒ d } n t | ƒ | | <| i t | ƒ ƒ } | i
| ƒ d | | >> a = np.arange(10).reshape(2, 5)
>>> a
array([[0, 1, 2, 3, 4],
[5, 6, 7, 8, 9]])
>>> ixgrid = np.ix_([0,1], [2,4])
>>> ixgrid
(array([[0],
[1]]), array([[2, 4]]))
>>> ixgrid[0].shape, ixgrid[1].shape
((2, 1), (1, 2))
>>> a[ixgrid]
array([[2, 4],
[7, 9]])
i s! Cross index must be 1 dimensionali ( t lent rangeR R
t ndimR t
issubclasst dtypet typet bool_t nonzerot reshapeR t append( t argst outt ndt baseshapet kt new( ( s< /usr/lib64/python2.6/site-packages/numpy/lib/index_tricks.pyR S s ,
t nd_gridc B s5 e Z d Z e d „ Z d „ Z d „ Z d „ Z RS( s
Construct a multi-dimensional "meshgrid".
``grid = nd_grid()`` creates an instance which will return a mesh-grid
when indexed. The dimension and number of the output arrays are equal
to the number of indexing dimensions. If the step length is not a
complex number, then the stop is not inclusive.
However, if the step length is a **complex number** (e.g. 5j), then the
integer part of its magnitude is interpreted as specifying the
number of points to create between the start and stop values, where
the stop value **is inclusive**.
If instantiated with an argument of ``sparse=True``, the mesh-grid is
open (or not fleshed out) so that only one-dimension of each returned
argument is greater than 1.
Parameters
----------
sparse : bool, optional
Whether the grid is sparse or not. Default is False.
Notes
-----
Two instances of `nd_grid` are made available in the NumPy namespace,
`mgrid` and `ogrid`::
mgrid = nd_grid(sparse=False)
ogrid = nd_grid(sparse=True)
Users should use these pre-defined instances instead of using `nd_grid`
directly.
Examples
--------
>>> mgrid = np.lib.index_tricks.nd_grid()
>>> mgrid[0:5,0:5]
array([[[0, 0, 0, 0, 0],
[1, 1, 1, 1, 1],
[2, 2, 2, 2, 2],
[3, 3, 3, 3, 3],
[4, 4, 4, 4, 4]],
[[0, 1, 2, 3, 4],
[0, 1, 2, 3, 4],
[0, 1, 2, 3, 4],
[0, 1, 2, 3, 4],
[0, 1, 2, 3, 4]]])
>>> mgrid[-1:1:5j]
array([-1. , -0.5, 0. , 0.5, 1. ])
>>> ogrid = np.lib.index_tricks.nd_grid(sparse=True)
>>> ogrid[0:5,0:5]
[array([[0],
[1],
[2],
[3],
[4]]), array([[0, 1, 2, 3, 4]])]
c C s
| | _ d S( N( t sparse( t selfR0 ( ( s< /usr/lib64/python2.6/site-packages/numpy/lib/index_tricks.pyt __init__Ê s c C s yœg } t } xÿ t t | ƒ ƒ D]ë } | | i } | | i } | d j o
d } n | d j o
d } n t | t ƒ o# | i t t | ƒ ƒ ƒ t
} n* | i t i | | i
| | d ƒ ƒ t | t
ƒ p' t | t
ƒ p t | | i
t
ƒ o
t
} q" q" W| i o& t d „ | | f t | ƒ ƒ } n t i | | ƒ } xÎ t t | ƒ ƒ D]º } | | i } | | i } | d j o
d } n | d j o
d } n t | t ƒ oF t t | ƒ ƒ } | d j o# | | i
| t
| d ƒ } q
n | | | | | | ß s ( t intR R t stept startt Nonet
isinstancet complexR( t abst floatt matht ceilt stopR0 t mapR t indicest newaxist slicet
IndexErrort TypeErrorR ( R1 t keyt sizet typR- R6 R7 t nnt slobjR? t length( ( s< /usr/lib64/python2.6/site-packages/numpy/lib/index_tricks.pyt __getitem__Ì st
)
'
c C s t i | | ƒ S( N( R R ( R1 t it j( ( s< /usr/lib64/python2.6/site-packages/numpy/lib/index_tricks.pyt __getslice__ s c C s d S( Ni ( ( R1 ( ( s< /usr/lib64/python2.6/site-packages/numpy/lib/index_tricks.pyt __len__ s ( t __name__t
__module__t __doc__t FalseR2 RL RO RP ( ( ( s< /usr/lib64/python2.6/site-packages/numpy/lib/index_tricks.pyR/ Ž s
; 7 R0 t AxisConcatenatorc B sG e Z d Z d „ Z d e d d d „ Z d „ Z d „ Z d „ Z RS( sw
Translates slice objects to concatenation along an axis.
For detailed documentation on usage, see `r_`.
c C sl | i o= | i } t | ƒ } | d j o | i o
| i } qG n | i | _ | i | _ d | _ | S( Ni i ( t matrixR! t makematt colt Tt _axist axist _matrix( R1 t rest oldndim( ( s< /usr/lib64/python2.6/site-packages/numpy/lib/index_tricks.pyt _retval s
i i iÿÿÿÿc C sC | | _ | | _ | | _ | | _ d | _ | | _ | | _ d S( Ni ( RZ R\ R[ RV RX t trans1dt ndmin( R1 R[ RV Ra R` ( ( s< /usr/lib64/python2.6/site-packages/numpy/lib/index_tricks.pyR2 s c
C sÙ | i } | i } t | t ƒ o/ t i ƒ i } t i | | i | i
ƒ } | St | ƒ t j o
| f } n g } g } g } g } xÚt
t | ƒ ƒ D]Æ}
t } t | |
ƒ t j oø | |
i } | |
i }
| |
i } |
d j o
d }
n | d j o
d } n t | t ƒ o. t t | ƒ ƒ } t i |
| d | ƒ} n t i |
| | ƒ } | d j o? t | d t d | ƒ} | d j o | i d | ƒ } q³q&npt | |
t ƒ o(|
d j o
t d ‚ n | d } | d j o t | _ | d j | _ qœ n d
| j oˆ | i! d
ƒ } yb g } | d D] } | t | ƒ qI~ \ | _" } t | ƒ d j o t | d ƒ } n wœ Wq°t d
‚ q°Xn y t | |
ƒ | _" wœ Wq&t t# f j
o t d
‚ q&Xn4t | |
ƒ t$ j o= t | |
d | ƒ} | i% |
ƒ t } | i% | i& ƒ nà | |
} | d j oÈ t | d t d t ƒ} t | d t d t d | ƒ} | d j o~ | i' | j on | | i' } | d j o | | d 7} n t
| ƒ } | } | | | | | | | !} | i( | ƒ } n ~ n | i% | ƒ | o' t | t i) ƒ o | i% | i& ƒ qœ qœ Wt* | | ƒ } | d j o, x) | D] }
| |
i+ | ƒ | |
>> np.r_[np.array([1,2,3]), 0, 0, np.array([4,5,6])]
array([1, 2, 3, 0, 0, 4, 5, 6])
>>> np.r_[-1:1:6j, [0]*3, 5, 6]
array([-1. , -0.6, -0.2, 0.2, 0.6, 1. , 0. , 0. , 0. , 5. , 6. ])
String integers specify the axis to concatenate along or the minimum
number of dimensions to force entries into.
>>> np.r_['-1', a, a] # concatenate along last axis
array([[0, 1, 2, 0, 1, 2],
[3, 4, 5, 3, 4, 5]])
>>> np.r_['0,2', [1,2,3], [4,5,6]] # concatenate along first axis, dim>=2
array([[1, 2, 3],
[4, 5, 6]])
>>> np.r_['0,2,0', [1,2,3], [4,5,6]]
array([[1],
[2],
[3],
[4],
[5],
[6]])
>>> np.r_['1,2,0', [1,2,3], [4,5,6]]
array([[1, 4],
[2, 5],
[3, 6]])
Using 'r' or 'c' as a first string argument creates a matrix.
>>> np.r_['r',[1,2,3], [4,5,6]]
matrix([[1, 2, 3, 4, 5, 6]])
c C s t i | d ƒ d S( Ni ( RU R2 ( R1 ( ( s< /usr/lib64/python2.6/site-packages/numpy/lib/index_tricks.pyR2 ê s ( RQ RR RS R2 ( ( ( s< /usr/lib64/python2.6/site-packages/numpy/lib/index_tricks.pyR‰ s \t CClassc B s e Z d Z d „ Z RS( s"
Translates slice objects to concatenation along the second axis.
This is short-hand for ``np.r_['-1,2,0', index expression]``, which is
useful because of its common occurrence. In particular, arrays will be
stacked along their last axis after being upgraded to at least 2-D with
1's post-pended to the shape (column vectors made out of 1-D arrays).
For detailed documentation, see `r_`.
Examples
--------
>>> np.c_[np.array([[1,2,3]]), 0, 0, np.array([[4,5,6]])]
array([[1, 2, 3, 0, 0, 4, 5, 6]])
c C s t i | d d d d d ƒd S( NiÿÿÿÿRa i R` i ( RU R2 ( R1 ( ( s< /usr/lib64/python2.6/site-packages/numpy/lib/index_tricks.pyR2 s ( RQ RR RS R2 ( ( ( s< /usr/lib64/python2.6/site-packages/numpy/lib/index_tricks.pyRŠ ï s c B s) e Z d Z d „ Z d „ Z d „ Z RS( s¢
Multidimensional index iterator.
Return an iterator yielding pairs of array coordinates and values.
Parameters
----------
a : ndarray
Input array.
See Also
--------
ndindex, flatiter
Examples
--------
>>> a = np.array([[1, 2], [3, 4]])
>>> for index, x in np.ndenumerate(a):
... print index, x
(0, 0) 1
(0, 1) 2
(1, 0) 3
(1, 1) 4
c C s t | ƒ i | _ d S( N( R
t flatt iter( R1 t arr( ( s< /usr/lib64/python2.6/site-packages/numpy/lib/index_tricks.pyR2 s c C s | i i | i i ƒ f S( s
Standard iterator method, returns the index tuple and array value.
Returns
-------
coords : tuple of ints
The indices of the current iteration.
val : scalar
The array element of the current iteration.
( RŒ t coordst next( R1 ( ( s< /usr/lib64/python2.6/site-packages/numpy/lib/index_tricks.pyR " s c C s | S( N( ( R1 ( ( s< /usr/lib64/python2.6/site-packages/numpy/lib/index_tricks.pyt __iter__0 s ( RQ RR RS R2 R R ( ( ( s< /usr/lib64/python2.6/site-packages/numpy/lib/index_tricks.pyR s c B s; e Z d Z d „ Z d „ Z d „ Z d „ Z d „ Z RS( sT
An N-dimensional iterator object to index arrays.
Given the shape of an array, an `ndindex` instance iterates over
the N-dimensional index of the array. At each iteration a tuple
of indices is returned, the last dimension is iterated over first.
Parameters
----------
`*args` : ints
The size of each dimension of the array.
See Also
--------
ndenumerate, flatiter
Examples
--------
>>> for index in np.ndindex(3, 2, 1):
... print index
(0, 0, 0)
(0, 1, 0)
(1, 0, 0)
(1, 1, 0)
(2, 0, 0)
(2, 1, 0)
c G s¤ t | ƒ d j o" t | d t ƒ o | d } n t | ƒ | _ d g | i | _ d | _ | | _ d } x% t | i ƒ D] } | | | 9} q W| | _ d S( Ni i ( R R9 R R+ t indt indext maxvalsR t total( R1 R) t totR- ( ( s< /usr/lib64/python2.6/site-packages/numpy/lib/index_tricks.pyR2 R s ' c C sj | d j o d S| i | | i | d j o | i | c d 7>> np.s_[2::2]
slice(2, None, 2)
>>> np.index_exp[2::2]
(slice(2, None, 2),)
>>> np.array([0, 1, 2, 3, 4])[np.s_[2::2]]
array([2, 4])
c C s
| | _ d S( N( t maketuple( R1 R› ( ( s< /usr/lib64/python2.6/site-packages/numpy/lib/index_tricks.pyR2 ¿ s c C s3 | i o! t | ƒ t d ƒ j o | f S| Sd S( N( ( R› R$ ( R1 t item( ( s< /usr/lib64/python2.6/site-packages/numpy/lib/index_tricks.pyRL Â s #c C s | i S( N( t maxint( R1 ( ( s< /usr/lib64/python2.6/site-packages/numpy/lib/index_tricks.pyRP È s c C s+ | | i j o
d } n | | | d … S( N( R R8 ( R1 R7 R? ( ( s< /usr/lib64/python2.6/site-packages/numpy/lib/index_tricks.pyRO Ë s
( RQ RR RS Ri R R2 RL RP RO ( ( ( s< /usr/lib64/python2.6/site-packages/numpy/lib/index_tricks.pyRš “ s * R› c C s¨ | i d j o t d ƒ ‚ n | i d j o | i d d } nJ t t | i ƒ d j ƒ p t d ƒ ‚ n d t | i d ƒ i ƒ } | | i d d | … 2``, the diagonal is the list of
locations with indices ``a[i, i, ..., i]`` all identical. This function
modifies the input array in-place, it does not return a value.
Parameters
----------
a : array, at least 2-D.
Array whose diagonal is to be filled, it gets modified in-place.
val : scalar
Value to be written on the diagonal, its type must be compatible with
that of the array a.
See also
--------
diag_indices, diag_indices_from
Notes
-----
.. versionadded:: 1.4.0
This functionality can be obtained via `diag_indices`, but internally
this version uses a much faster implementation that never constructs the
indices and uses simple slicing.
Examples
--------
>>> a = zeros((3, 3), int)
>>> fill_diagonal(a, 5)
>>> a
array([[5, 0, 0],
[0, 5, 0],
[0, 0, 5]])
The same function can operate on a 4-D array:
>>> a = zeros((3, 3, 3, 3), int)
>>> fill_diagonal(a, 4)
We only show a few blocks for clarity:
>>> a[0, 0]
array([[4, 0, 0],
[0, 0, 0],
[0, 0, 0]])
>>> a[1, 1]
array([[0, 0, 0],
[0, 4, 0],
[0, 0, 0]])
>>> a[2, 2]
array([[0, 0, 0],
[0, 0, 0],
[0, 0, 4]])
i s array must be at least 2-di i s/ All dimensions of input must be of equal lengthiÿÿÿÿN( R! R t shapeR R R t sumR‹ ( t aR™ R6 ( ( s< /usr/lib64/python2.6/site-packages/numpy/lib/index_tricks.pyR
Ù s ; i c C s t | ƒ } | f | S( s+
Return the indices to access the main diagonal of an array.
This returns a tuple of indices that can be used to access the main
diagonal of an array `a` with ``a.ndim >= 2`` dimensions and shape
(n, n, ..., n). For ``a.ndim = 2`` this is the usual diagonal, for
``a.ndim > 2`` this is the set of indices to access ``a[i, i, ..., i]``
for ``i = [0..n-1]``.
Parameters
----------
n : int
The size, along each dimension, of the arrays for which the returned
indices can be used.
ndim : int, optional
The number of dimensions.
See also
--------
diag_indices_from
Notes
-----
.. versionadded:: 1.4.0
Examples
--------
Create a set of indices to access the diagonal of a (4, 4) array:
>>> di = np.diag_indices(4)
>>> di
(array([0, 1, 2, 3]), array([0, 1, 2, 3]))
>>> a = np.arange(16).reshape(4, 4)
>>> a
array([[ 0, 1, 2, 3],
[ 4, 5, 6, 7],
[ 8, 9, 10, 11],
[12, 13, 14, 15]])
>>> a[di] = 100
>>> a
array([[100, 1, 2, 3],
[ 4, 100, 6, 7],
[ 8, 9, 100, 11],
[ 12, 13, 14, 100]])
Now, we create indices to manipulate a 3-D array:
>>> d3 = np.diag_indices(2, 3)
>>> d3
(array([0, 1]), array([0, 1]), array([0, 1]))
And use it to set the diagonal of an array of zeros to 1:
>>> a = np.zeros((2, 2, 2), dtype=np.int)
>>> a[d3] = 1
>>> a
array([[[1, 0],
[0, 0]],
[[0, 0],
[0, 1]]])
( R ( t nR! R ( ( s< /usr/lib64/python2.6/site-packages/numpy/lib/index_tricks.pyR % s @ c C sc | i d j p t d ƒ ‚ n t t | i ƒ d j ƒ p t d ƒ ‚ n t | i d | i ƒ S( s
Return the indices to access the main diagonal of an n-dimensional array.
See `diag_indices` for full details.
Parameters
----------
arr : array, at least 2-D
See Also
--------
diag_indices
Notes
-----
.. versionadded:: 1.4.0
i s input array must be at least 2-di s/ All dimensions of input must be of equal length( R! R R R Rž R ( R ( ( s< /usr/lib64/python2.6/site-packages/numpy/lib/index_tricks.pyR i s
(, t __all__Ri t numpy.core.numerict coret numericR R
R R R R R t numpy.core.numerictypesR R= Ro t numpy.matrixlibt matrixlibRV R RW R R t objectR/ RT R Rr R R8 RS RU R‰ R RŠ R R R Rš R R R
R R ( ( ( s< /usr/lib64/python2.6/site-packages/numpy/lib/index_tricks.pyt sF . = ;{ ` /_= L D