Current File : //lib64/python3.9/site-packages/numpy/lib/__pycache__/npyio.cpython-39.opt-1.pyc
a
z[yc�y � @ s d dl Z d dlZd dlZd dlZd dlZd dlZd dlZd dlZd dlmZm Z
m Z d dl m
Z
d dlZddlmZ ddlmZ d dlmZ d dlmZmZ d d lmZ d d
lm Z m Z dd l m Z m Z m!Z!m"Z"m#Z#m$Z$m%Z%m&Z&m'Z'm(Z(m)Z) d d l*m+Z+m,Z,m-Z-m.Z.m/Z/m0Z0 g d
�Z1ej2ej3dd�Z3G dd� d�Z4dd� Z5G dd� de
�Z6e d�dYej7d�dd��Z8dZd d �Z9e3e9�d[d d ��Z:d d!� Z;e3e;�d"d#� �Z<d$d%� Z=e3e=�d&d'� �Z>d\d(d)�Z?d*d+� Z@eAd,�d-d.�ZBd/ZCd]dd0�d1d2�ZDd^d4d5�ZEd6d7� ZFd/ZCd8d9d:d;dd ddddejGd<d=� d>d?�ZHe e d�eId9ddd ddd d<df
ddd@�dAdB���ZJe3eD�eJ�ZKd_dCdD�ZLe3eL�d`dJdK��ZMe d�dadLdM��ZNdbdddN�dOdP�ZOe e d�eId9dd d dddddddH�PeQe jR��dQdddRdddddd<fd ddN�dSdT���ZSe3eO�eS�ZTdUdV� ZUdWdX� ZVdS )c� N)�
itemgetter�index� methodcaller)�Mapping� )�format)�
DataSource)� overrides)�packbits�
unpackbits)�_load_from_filelike)�set_array_function_like_doc�
set_module) � LineSplitter�
NameValidator�StringConverter�ConverterError�ConverterLockError�ConversionWarning�_is_string_like�has_nested_fields�
flatten_dtype�
easy_dtype� _decode_line)�asbytes�asstr� asunicode� os_fspath� os_PathLike�pickle)
�savetxt�loadtxt�
genfromtxt�
recfromtxt�
recfromcsv�load�save�savez�savez_compressedr
r � fromregexr �numpy)�modulec @ s( e Zd ZdZdd� Zdd� Zdd� ZdS ) �BagObjam
BagObj(obj)
Convert attribute look-ups to getitems on the object passed in.
Parameters
----------
obj : class instance
Object on which attribute look-up is performed.
Examples
--------
>>> from numpy.lib.npyio import BagObj as BO
>>> class BagDemo:
... def __getitem__(self, key): # An instance of BagObj(BagDemo)
... # will call this method when any
... # attribute look-up is required
... result = "Doesn't matter what you want, "
... return result + "you're gonna get this"
...
>>> demo_obj = BagDemo()
>>> bagobj = BO(demo_obj)
>>> bagobj.hello_there
"Doesn't matter what you want, you're gonna get this"
>>> bagobj.I_can_be_anything
"Doesn't matter what you want, you're gonna get this"
c C s t �|�| _d S �N)�weakref�proxy�_obj)�self�obj� r3 �5/usr/lib64/python3.9/site-packages/numpy/lib/npyio.py�__init__H s zBagObj.__init__c C s4 zt �| d�| W S ty. t|�d �Y n0 d S )Nr0 )�object�__getattribute__�KeyError�AttributeError)r1 �keyr3 r3 r4 r7 L s zBagObj.__getattribute__c C s t t�| d��� �S )z�
Enables dir(bagobj) to list the files in an NpzFile.
This also enables tab-completion in an interpreter or IPython.
r0 )�listr6 r7 �keys�r1 r3 r3 r4 �__dir__R s zBagObj.__dir__N)�__name__�
__module__� __qualname__�__doc__r5 r7 r> r3 r3 r3 r4 r, * s r, c O s: t | d�st| �} ddl}d|d< |j| g|�R i |��S )z�
Create a ZipFile.
Allows for Zip64, and the `file` argument can accept file, str, or
pathlib.Path objects. `args` and `kwargs` are passed to the zipfile.ZipFile
constructor.
�readr NTZ
allowZip64)�hasattrr �zipfile�ZipFile)�file�args�kwargsrE r3 r3 r4 �zipfile_factory[ s
rJ c @ sb e Zd ZdZdZdZdejd�dd�Zdd� Z d d
� Z
d d � Z d
d� Z dd� Z
dd� Zdd� ZdS )�NpzFilea�
NpzFile(fid)
A dictionary-like object with lazy-loading of files in the zipped
archive provided on construction.
`NpzFile` is used to load files in the NumPy ``.npz`` data archive
format. It assumes that files in the archive have a ``.npy`` extension,
other files are ignored.
The arrays and file strings are lazily loaded on either
getitem access using ``obj['key']`` or attribute lookup using
``obj.f.key``. A list of all files (without ``.npy`` extensions) can
be obtained with ``obj.files`` and the ZipFile object itself using
``obj.zip``.
Attributes
----------
files : list of str
List of all files in the archive with a ``.npy`` extension.
zip : ZipFile instance
The ZipFile object initialized with the zipped archive.
f : BagObj instance
An object on which attribute can be performed as an alternative
to getitem access on the `NpzFile` instance itself.
allow_pickle : bool, optional
Allow loading pickled data. Default: False
.. versionchanged:: 1.16.3
Made default False in response to CVE-2019-6446.
pickle_kwargs : dict, optional
Additional keyword arguments to pass on to pickle.load.
These are only useful when loading object arrays saved on
Python 2 when using Python 3.
max_header_size : int, optional
Maximum allowed size of the header. Large headers may not be safe
to load securely and thus require explicitly passing a larger value.
See :py:meth:`ast.literal_eval()` for details.
This option is ignored when `allow_pickle` is passed. In that case
the file is by definition trusted and the limit is unnecessary.
Parameters
----------
fid : file or str
The zipped archive to open. This is either a file-like object
or a string containing the path to the archive.
own_fid : bool, optional
Whether NpzFile should close the file handle.
Requires that `fid` is a file-like object.
Examples
--------
>>> from tempfile import TemporaryFile
>>> outfile = TemporaryFile()
>>> x = np.arange(10)
>>> y = np.sin(x)
>>> np.savez(outfile, x=x, y=y)
>>> _ = outfile.seek(0)
>>> npz = np.load(outfile)
>>> isinstance(npz, np.lib.npyio.NpzFile)
True
>>> sorted(npz.files)
['x', 'y']
>>> npz['x'] # getitem access
array([0, 1, 2, 3, 4, 5, 6, 7, 8, 9])
>>> npz.f.x # attribute lookup
array([0, 1, 2, 3, 4, 5, 6, 7, 8, 9])
NF��max_header_sizec C s� t |�}|�� | _g | _|| _|| _|| _| jD ]0}|�d�rT| j�|d d� � q0| j�|� q0|| _ t
| �| _ |r||| _ d S )N�.npy���)
rJ Znamelist�_files�files� allow_picklerM �
pickle_kwargs�endswith�append�zipr, �f�fid)r1 rX �own_fidrR rS rM �_zip�xr3 r3 r4 r5 � s
zNpzFile.__init__c C s | S r- r3 r= r3 r3 r4 � __enter__� s zNpzFile.__enter__c C s | � � d S r- ��close)r1 �exc_type� exc_value� tracebackr3 r3 r4 �__exit__� s zNpzFile.__exit__c C s> | j dur| j �� d| _ | jdur4| j�� d| _d| _dS )z"
Close the file.
N)rV r^ rX rW r= r3 r3 r4 r^ � s
z
NpzFile.closec C s | � � d S r- r] r= r3 r3 r4 �__del__� s zNpzFile.__del__c C s
t | j�S r- )�iterrQ r= r3 r3 r4 �__iter__� s zNpzFile.__iter__c C s
t | j�S r- )�lenrQ r= r3 r3 r4 �__len__� s zNpzFile.__len__c C s� d}|| j v rd}n|| jv r*d}|d7 }|r�| j�|�}|�ttj��}|�� |tjkr�| j�|�}tj || j
| j | j d�S | j�|�S n t
d| ��d S )NFTrN �rR rS rM z %s is not a file in the archive)rP rQ rV �openrC rf r � MAGIC_PREFIXr^ �
read_arrayrR rS rM r8 )r1 r: �member�bytes�magicr3 r3 r4 � __getitem__� s&
�zNpzFile.__getitem__)FFN)r? r@ rA rB rV rX r �_MAX_HEADER_SIZEr5 r\ rb r^ rc re rg ro r3 r3 r3 r4 rK j s H ��
rK FT�ASCIIrL c C s� |dvrt d��t||d�}t�� ��v}t| d�r<| }d} n|�tt| �d��}d} d}
d } tt j
�} |� | �}
|� t
| t|
�� d
� |
�|
�s�|
�| �r�|�� t|| |||d �}|W d � S |
t j
k�r"|�r |r�d
}t j| ||d�W d � S t j||||d�W d � S nl|�s0t d��z"tj|fi |��W W d � S t�y� } z t�d| �d��|�W Y d }~n
d }~0 0 W d � n1 �s�0 Y d S )a�
Load arrays or pickled objects from ``.npy``, ``.npz`` or pickled files.
.. warning:: Loading files that contain object arrays uses the ``pickle``
module, which is not secure against erroneous or maliciously
constructed data. Consider passing ``allow_pickle=False`` to
load data that is known not to contain object arrays for the
safer handling of untrusted sources.
Parameters
----------
file : file-like object, string, or pathlib.Path
The file to read. File-like objects must support the
``seek()`` and ``read()`` methods and must always
be opened in binary mode. Pickled files require that the
file-like object support the ``readline()`` method as well.
mmap_mode : {None, 'r+', 'r', 'w+', 'c'}, optional
If not None, then memory-map the file, using the given mode (see
`numpy.memmap` for a detailed description of the modes). A
memory-mapped array is kept on disk. However, it can be accessed
and sliced like any ndarray. Memory mapping is especially useful
for accessing small fragments of large files without reading the
entire file into memory.
allow_pickle : bool, optional
Allow loading pickled object arrays stored in npy files. Reasons for
disallowing pickles include security, as loading pickled data can
execute arbitrary code. If pickles are disallowed, loading object
arrays will fail. Default: False
.. versionchanged:: 1.16.3
Made default False in response to CVE-2019-6446.
fix_imports : bool, optional
Only useful when loading Python 2 generated pickled files on Python 3,
which includes npy/npz files containing object arrays. If `fix_imports`
is True, pickle will try to map the old Python 2 names to the new names
used in Python 3.
encoding : str, optional
What encoding to use when reading Python 2 strings. Only useful when
loading Python 2 generated pickled files in Python 3, which includes
npy/npz files containing object arrays. Values other than 'latin1',
'ASCII', and 'bytes' are not allowed, as they can corrupt numerical
data. Default: 'ASCII'
max_header_size : int, optional
Maximum allowed size of the header. Large headers may not be safe
to load securely and thus require explicitly passing a larger value.
See :py:meth:`ast.literal_eval()` for details.
This option is ignored when `allow_pickle` is passed. In that case
the file is by definition trusted and the limit is unnecessary.
Returns
-------
result : array, tuple, dict, etc.
Data stored in the file. For ``.npz`` files, the returned instance
of NpzFile class must be closed to avoid leaking file descriptors.
Raises
------
OSError
If the input file does not exist or cannot be read.
UnpicklingError
If ``allow_pickle=True``, but the file cannot be loaded as a pickle.
ValueError
The file contains an object array, but ``allow_pickle=False`` given.
See Also
--------
save, savez, savez_compressed, loadtxt
memmap : Create a memory-map to an array stored in a file on disk.
lib.format.open_memmap : Create or load a memory-mapped ``.npy`` file.
Notes
-----
- If the file contains pickle data, then whatever object is stored
in the pickle is returned.
- If the file is a ``.npy`` file, then a single array is returned.
- If the file is a ``.npz`` file, then a dictionary-like object is
returned, containing ``{filename: array}`` key-value pairs, one for
each file in the archive.
- If the file is a ``.npz`` file, the returned value supports the
context manager protocol in a similar fashion to the open function::
with load('foo.npz') as data:
a = data['a']
The underlying file descriptor is closed when exiting the 'with'
block.
Examples
--------
Store data to disk, and load it again:
>>> np.save('/tmp/123', np.array([[1, 2, 3], [4, 5, 6]]))
>>> np.load('/tmp/123.npy')
array([[1, 2, 3],
[4, 5, 6]])
Store compressed data to disk, and load it again:
>>> a=np.array([[1, 2, 3], [4, 5, 6]])
>>> b=np.array([1, 2])
>>> np.savez('/tmp/123.npz', a=a, b=b)
>>> data = np.load('/tmp/123.npz')
>>> data['a']
array([[1, 2, 3],
[4, 5, 6]])
>>> data['b']
array([1, 2])
>>> data.close()
Mem-map the stored array, and then access the second row
directly from disk:
>>> X = np.load('/tmp/123.npy', mmap_mode='r')
>>> X[1, :]
memmap([4, 5, 6])
)rq �latin1rm z.encoding must be 'ASCII', 'latin1', or 'bytes')�encoding� fix_importsrC F�rbTs PKs PKr )rY rR rS rM Nl )�moderM rh z@Cannot load file containing pickled data when allow_pickle=FalsezFailed to interpret file z as a pickle)�
ValueError�dict�
contextlib� ExitStackrD �
enter_contextri r rf r rj rC �seek�min�
startswith�pop_allrK Z open_memmaprk r r% � ExceptionZUnpicklingError)rG Z mmap_moderR rt rs rM rS �stackrX rY Z _ZIP_PREFIXZ _ZIP_SUFFIX�Nrn �ret�er3 r3 r4 r% sT y
� ��"
��r% c C s |fS r- r3 )rG �arrrR rt r3 r3 r4 �_save_dispatcher� s r� c C s� t | d�rt�| �}n$t| �} | �d�s0| d } t| d�}|�2}t�|�}tj |||t
|d�d� W d � n1 sv0 Y dS )a<
Save an array to a binary file in NumPy ``.npy`` format.
Parameters
----------
file : file, str, or pathlib.Path
File or filename to which the data is saved. If file is a file-object,
then the filename is unchanged. If file is a string or Path, a ``.npy``
extension will be appended to the filename if it does not already
have one.
arr : array_like
Array data to be saved.
allow_pickle : bool, optional
Allow saving object arrays using Python pickles. Reasons for disallowing
pickles include security (loading pickled data can execute arbitrary
code) and portability (pickled objects may not be loadable on different
Python installations, for example if the stored objects require libraries
that are not available, and not all pickled data is compatible between
Python 2 and Python 3).
Default: True
fix_imports : bool, optional
Only useful in forcing objects in object arrays on Python 3 to be
pickled in a Python 2 compatible way. If `fix_imports` is True, pickle
will try to map the new Python 3 names to the old module names used in
Python 2, so that the pickle data stream is readable with Python 2.
See Also
--------
savez : Save several arrays into a ``.npz`` archive
savetxt, load
Notes
-----
For a description of the ``.npy`` format, see :py:mod:`numpy.lib.format`.
Any data saved to the file is appended to the end of the file.
Examples
--------
>>> from tempfile import TemporaryFile
>>> outfile = TemporaryFile()
>>> x = np.arange(10)
>>> np.save(outfile, x)
>>> _ = outfile.seek(0) # Only needed here to simulate closing & reopening file
>>> np.load(outfile)
array([0, 1, 2, 3, 4, 5, 6, 7, 8, 9])
>>> with open('test.npy', 'wb') as f:
... np.save(f, np.array([1, 2]))
... np.save(f, np.array([1, 3]))
>>> with open('test.npy', 'rb') as f:
... a = np.load(f)
... b = np.load(f)
>>> print(a, b)
# [1 2] [1 3]
�writerN �wb)rt �rR rS N) rD ry � nullcontextr rT ri �np�
asanyarrayr � write_arrayrx )rG r� rR rt Zfile_ctxrX r3 r3 r4 r& � s =
�r& c o s |E d H |� � E d H d S r- ��values�rG rH �kwdsr3 r3 r4 �_savez_dispatcher s
r� c O s t | ||d� dS )a� Save several arrays into a single file in uncompressed ``.npz`` format.
Provide arrays as keyword arguments to store them under the
corresponding name in the output file: ``savez(fn, x=x, y=y)``.
If arrays are specified as positional arguments, i.e., ``savez(fn,
x, y)``, their names will be `arr_0`, `arr_1`, etc.
Parameters
----------
file : str or file
Either the filename (string) or an open file (file-like object)
where the data will be saved. If file is a string or a Path, the
``.npz`` extension will be appended to the filename if it is not
already there.
args : Arguments, optional
Arrays to save to the file. Please use keyword arguments (see
`kwds` below) to assign names to arrays. Arrays specified as
args will be named "arr_0", "arr_1", and so on.
kwds : Keyword arguments, optional
Arrays to save to the file. Each array will be saved to the
output file with its corresponding keyword name.
Returns
-------
None
See Also
--------
save : Save a single array to a binary file in NumPy format.
savetxt : Save an array to a file as plain text.
savez_compressed : Save several arrays into a compressed ``.npz`` archive
Notes
-----
The ``.npz`` file format is a zipped archive of files named after the
variables they contain. The archive is not compressed and each file
in the archive contains one variable in ``.npy`` format. For a
description of the ``.npy`` format, see :py:mod:`numpy.lib.format`.
When opening the saved ``.npz`` file with `load` a `NpzFile` object is
returned. This is a dictionary-like object which can be queried for
its list of arrays (with the ``.files`` attribute), and for the arrays
themselves.
Keys passed in `kwds` are used as filenames inside the ZIP archive.
Therefore, keys should be valid filenames; e.g., avoid keys that begin with
``/`` or contain ``.``.
When naming variables with keyword arguments, it is not possible to name a
variable ``file``, as this would cause the ``file`` argument to be defined
twice in the call to ``savez``.
Examples
--------
>>> from tempfile import TemporaryFile
>>> outfile = TemporaryFile()
>>> x = np.arange(10)
>>> y = np.sin(x)
Using `savez` with \*args, the arrays are saved with default names.
>>> np.savez(outfile, x, y)
>>> _ = outfile.seek(0) # Only needed here to simulate closing & reopening file
>>> npzfile = np.load(outfile)
>>> npzfile.files
['arr_0', 'arr_1']
>>> npzfile['arr_0']
array([0, 1, 2, 3, 4, 5, 6, 7, 8, 9])
Using `savez` with \**kwds, the arrays are saved with the keyword names.
>>> outfile = TemporaryFile()
>>> np.savez(outfile, x=x, y=y)
>>> _ = outfile.seek(0)
>>> npzfile = np.load(outfile)
>>> sorted(npzfile.files)
['x', 'y']
>>> npzfile['x']
array([0, 1, 2, 3, 4, 5, 6, 7, 8, 9])
FN��_savezr� r3 r3 r4 r' s Tr' c o s |E d H |� � E d H d S r- r� r� r3 r3 r4 � _savez_compressed_dispatcherj s
r� c O s t | ||d� dS )a�
Save several arrays into a single file in compressed ``.npz`` format.
Provide arrays as keyword arguments to store them under the
corresponding name in the output file: ``savez(fn, x=x, y=y)``.
If arrays are specified as positional arguments, i.e., ``savez(fn,
x, y)``, their names will be `arr_0`, `arr_1`, etc.
Parameters
----------
file : str or file
Either the filename (string) or an open file (file-like object)
where the data will be saved. If file is a string or a Path, the
``.npz`` extension will be appended to the filename if it is not
already there.
args : Arguments, optional
Arrays to save to the file. Please use keyword arguments (see
`kwds` below) to assign names to arrays. Arrays specified as
args will be named "arr_0", "arr_1", and so on.
kwds : Keyword arguments, optional
Arrays to save to the file. Each array will be saved to the
output file with its corresponding keyword name.
Returns
-------
None
See Also
--------
numpy.save : Save a single array to a binary file in NumPy format.
numpy.savetxt : Save an array to a file as plain text.
numpy.savez : Save several arrays into an uncompressed ``.npz`` file format
numpy.load : Load the files created by savez_compressed.
Notes
-----
The ``.npz`` file format is a zipped archive of files named after the
variables they contain. The archive is compressed with
``zipfile.ZIP_DEFLATED`` and each file in the archive contains one variable
in ``.npy`` format. For a description of the ``.npy`` format, see
:py:mod:`numpy.lib.format`.
When opening the saved ``.npz`` file with `load` a `NpzFile` object is
returned. This is a dictionary-like object which can be queried for
its list of arrays (with the ``.files`` attribute), and for the arrays
themselves.
Examples
--------
>>> test_array = np.random.rand(3, 2)
>>> test_vector = np.random.rand(4)
>>> np.savez_compressed('/tmp/123', a=test_array, b=test_vector)
>>> loaded = np.load('/tmp/123.npz')
>>> print(np.array_equal(test_array, loaded['a']))
True
>>> print(np.array_equal(test_vector, loaded['b']))
True
TNr� r� r3 r3 r4 r( o s ?r( c C s� dd l }t| d�s,t| �} | �d�s,| d } |}t|�D ]0\}} d| }
|
|�� v r`td|
��| ||
< q8|rv|j} n|j} t | d| d�} |�
� D ]\\}
} |
d }
t � | �} | j
|
dd d
��"}tj|| ||d � W d � q�1 s�0 Y q�| �� d S ) Nr r� z.npzzarr_%dz,Cannot use un-named variables and keyword %s�w)rv � compressionrN T)Z force_zip64r� )rE rD r rT � enumerater< rw � ZIP_DEFLATEDZ
ZIP_STOREDrJ �itemsr� r� ri r r� r^ )rG rH r� �compressrR rS rE Znamedict�i�valr: r� Zzipf�fnamerX r3 r3 r4 r� � s4
�
�&r� c C s | dvrt d| � ���dS )z�Just checks if the param ndmin is supported on
_ensure_ndmin_ndarray. It is intended to be used as
verification before running anything expensive.
e.g. loadtxt, genfromtxt
)r r � z Illegal value of ndmin keyword: N)rw ��ndminr3 r3 r4 �!_ensure_ndmin_ndarray_check_param� s r� r� c C sJ | j |krt�| �} | j |k rF|dkr2t�| �} n|dkrFt�| �j} | S )a This is a helper function of loadtxt and genfromtxt to ensure
proper minimum dimension as requested
ndim : int. Supported values 1, 2, 3
^^ whenever this changes, keep in sync with
_ensure_ndmin_ndarray_check_param
r r� )�ndimr� ZsqueezeZ
atleast_1d�
atleast_2d�T)�ar� r3 r3 r4 �_ensure_ndmin_ndarray� s
r� iP� )�likec C s | fS r- r3 ) r� �dtype�comments� delimiter�
converters�skiprows�usecols�unpackr� rs �max_rowsr� r3 r3 r4 �_loadtxt_dispatcher� s r� �argumentc C sL zt �| � W n" ty0 t|� d��d �Y n0 | dk rHt|� d���d S )Nz must be an integerr z must be nonnegative)�operatorr � TypeErrorrw )�value�namer3 r3 r4 �_check_nonneg_int s r� c c sB | D ]8}t |t�r |�|�}|D ]}|�|d�d }q |V qdS )a
Generator that consumes a line iterated iterable and strips out the
multiple (or multi-character) comments from lines.
This is a pre-processing step to achieve feature parity with loadtxt
(we assume that this feature is a nieche feature).
r r N)�
isinstancerm �decode�split)�iterabler� rs �line�cr3 r3 r4 �_preprocess_comments s
r� �,�#�"�jrm ) r� �comment�quote�imaginary_unitr� � skiplinesr� r� r� r� r� rs c sD d}
| dkrd} d}
| du r$t d��t�| �} d}| jdv rj| dks\| dks\| d ks\| d
krj| }t�t�} |dur�z t|�}W n t y� |g}Y n0 t| � |du r�d}n�d |v r�td ��t|�}d}t |�d
kr�d}n^t |�dk�rt
|d
t ��r:t |d
�dk�r:|d
}d}n ||v �r:t d|� d|� d���|du�rV|du�rVtd��t |�dk�rltd��t |� |du�r�t |� nd}t
�� }d}z�t
| tj��r�t�| �} t
| t ��r�tjjj| d| d�}| du �r�t|dd�} t
�|�}|}d}n | du �rt| dd�} t| �}W n< t �yX } z"tdt| �� d��|�W Y d}~n
d}~0 0 |��Z |du�r�|�r~t|�}d}t||| �}|du �r�t|||||||||| | ||
d�
� n�|�r�t|�}d}|d k�r�d}g }|d
k�r`|d
k �r�t}n
t t|�}t|||||||||| | ||
|d �}|� |� |�� d
}|d
k�rJ||8 }t |�|k �rܐq`�q�t |�dk�r�t |d �d
k�r�|d= t |�dk�r�|d
� ntj |d
d �� W d � n1 �s�0 Y t � | d �� � j!�r� j!d
d
k�rt"j#d | � d!�t$d"d#� |
�r<� j}|j%du�r4� fd$d%�|j%D �S � j&S n� S dS )&a�
Read a NumPy array from a text file.
Parameters
----------
fname : str or file object
The filename or the file to be read.
delimiter : str, optional
Field delimiter of the fields in line of the file.
Default is a comma, ','. If None any sequence of whitespace is
considered a delimiter.
comment : str or sequence of str or None, optional
Character that begins a comment. All text from the comment
character to the end of the line is ignored.
Multiple comments or multiple-character comment strings are supported,
but may be slower and `quote` must be empty if used.
Use None to disable all use of comments.
quote : str or None, optional
Character that is used to quote string fields. Default is '"'
(a double quote). Use None to disable quote support.
imaginary_unit : str, optional
Character that represent the imaginay unit `sqrt(-1)`.
Default is 'j'.
usecols : array_like, optional
A one-dimensional array of integer column numbers. These are the
columns from the file to be included in the array. If this value
is not given, all the columns are used.
skiplines : int, optional
Number of lines to skip before interpreting the data in the file.
max_rows : int, optional
Maximum number of rows of data to read. Default is to read the
entire file.
converters : dict or callable, optional
A function to parse all columns strings into the desired value, or
a dictionary mapping column number to a parser function.
E.g. if column 0 is a date string: ``converters = {0: datestr2num}``.
Converters can also be used to provide a default value for missing
data, e.g. ``converters = lambda s: float(s.strip() or 0)`` will
convert empty fields to 0.
Default: None
ndmin : int, optional
Minimum dimension of the array returned.
Allowed values are 0, 1 or 2. Default is 0.
unpack : bool, optional
If True, the returned array is transposed, so that arguments may be
unpacked using ``x, y, z = read(...)``. When used with a structured
data-type, arrays are returned for each field. Default is False.
dtype : numpy data type
A NumPy dtype instance, can be a structured dtype to map to the
columns of the file.
encoding : str, optional
Encoding used to decode the inputfile. The special value 'bytes'
(the default) enables backwards-compatible behavior for `converters`,
ensuring that inputs to the converter functions are encoded
bytes objects. The special value 'bytes' has no additional effect if
``converters=None``. If encoding is ``'bytes'`` or ``None``, the
default system encoding is used.
Returns
-------
ndarray
NumPy array.
Examples
--------
First we create a file for the example.
>>> s1 = '1.0,2.0,3.0\n4.0,5.0,6.0\n'
>>> with open('example1.csv', 'w') as f:
... f.write(s1)
>>> a1 = read_from_filename('example1.csv')
>>> a1
array([[1., 2., 3.],
[4., 5., 6.]])
The second example has columns with different data types, so a
one-dimensional array with a structured data type is returned.
The tab character is used as the field delimiter.
>>> s2 = '1.0\t10\talpha\n2.3\t25\tbeta\n4.5\t16\tgamma\n'
>>> with open('example2.tsv', 'w') as f:
... f.write(s2)
>>> a2 = read_from_filename('example2.tsv', delimiter='\t')
>>> a2
array([(1. , 10, b'alpha'), (2.3, 25, b'beta'), (4.5, 16, b'gamma')],
dtype=[('f0', '<f8'), ('f1', 'u1'), ('f2', 'S5')])
Frm NTza dtype must be provided.ZSUMZS0ZU0ZM8Zm8� zJcomments cannot be an empty string. Use comments=None to disable comments.r r zComment characters 'z ' cannot include the delimiter '�'z�when multiple comments or a multi-character comment is given, quotes are not supported. In this case quotechar must be set to None.z len(imaginary_unit) must be 1.����rt�rs rs rr zGfname must be a string, filehandle, list of strings,
or generator. Got � instead.) r� r� r� r� r� r� r� r� r� rs �filelike�byte_converters�S)
r� r� r� r� r� r� r� r� r� rs r� r� �c_byte_converters)Zaxisr� z#loadtxt: input contained no data: "r� � )�category�
stacklevelc s g | ] }� | �qS r3 r3 ��.0Zfield�r� r3 r4 �
<listcomp>2 � z_read.<locals>.<listcomp>)'r� r� r� �kindr6 r; r� rw �tuplerf r� �strr� ry r� �os�PathLike�fspath�lib� _datasourceri �getattr�closingrd �typer� r �_loadtxt_chunksizer} rU ZastypeZ concatenater� �shape�warnings�warn� UserWarning�namesr� )r� r� r� r� r� r� r� r� r� r� r� r� rs r� Z read_dtype_via_object_chunksr� Zfh_closing_ctxr� �fh�datar� r� �chunksZ
chunk_sizeZnext_arrr� �dtr3 r� r4 �_read! s \
����
� "
��
�
���
�
�
.
� r� )� quotecharr� c C s� | dur(t | ||||||||| |
| d� S t|t�r<|�d� |du rJtj}|}
|
durxt|
ttf�rj|
g}
dd� |
D �}
t|t�r�|�d�}t| ||
||||||| |
| d� }|S )a7$
Load data from a text file.
Each row in the text file must have the same number of values.
Parameters
----------
fname : file, str, pathlib.Path, list of str, generator
File, filename, list, or generator to read. If the filename
extension is ``.gz`` or ``.bz2``, the file is first decompressed. Note
that generators must return bytes or strings. The strings
in a list or produced by a generator are treated as lines.
dtype : data-type, optional
Data-type of the resulting array; default: float. If this is a
structured data-type, the resulting array will be 1-dimensional, and
each row will be interpreted as an element of the array. In this
case, the number of columns used must match the number of fields in
the data-type.
comments : str or sequence of str or None, optional
The characters or list of characters used to indicate the start of a
comment. None implies no comments. For backwards compatibility, byte
strings will be decoded as 'latin1'. The default is '#'.
delimiter : str, optional
The character used to separate the values. For backwards compatibility,
byte strings will be decoded as 'latin1'. The default is whitespace.
.. versionchanged:: 1.23.0
Only single character delimiters are supported. Newline characters
cannot be used as the delimiter.
converters : dict or callable, optional
A function to parse all columns strings into the desired value, or
a dictionary mapping column number to a parser function.
E.g. if column 0 is a date string: ``converters = {0: datestr2num}``.
Converters can also be used to provide a default value for missing
data, e.g. ``converters = lambda s: float(s.strip() or 0)`` will
convert empty fields to 0.
Default: None.
.. versionchanged:: 1.23.0
The ability to pass a single callable to be applied to all columns
was added.
skiprows : int, optional
Skip the first `skiprows` lines, including comments; default: 0.
usecols : int or sequence, optional
Which columns to read, with 0 being the first. For example,
``usecols = (1,4,5)`` will extract the 2nd, 5th and 6th columns.
The default, None, results in all columns being read.
.. versionchanged:: 1.11.0
When a single column has to be read it is possible to use
an integer instead of a tuple. E.g ``usecols = 3`` reads the
fourth column the same way as ``usecols = (3,)`` would.
unpack : bool, optional
If True, the returned array is transposed, so that arguments may be
unpacked using ``x, y, z = loadtxt(...)``. When used with a
structured data-type, arrays are returned for each field.
Default is False.
ndmin : int, optional
The returned array will have at least `ndmin` dimensions.
Otherwise mono-dimensional axes will be squeezed.
Legal values: 0 (default), 1 or 2.
.. versionadded:: 1.6.0
encoding : str, optional
Encoding used to decode the inputfile. Does not apply to input streams.
The special value 'bytes' enables backward compatibility workarounds
that ensures you receive byte arrays as results if possible and passes
'latin1' encoded strings to converters. Override this value to receive
unicode arrays and pass strings as input to converters. If set to None
the system default is used. The default value is 'bytes'.
.. versionadded:: 1.14.0
max_rows : int, optional
Read `max_rows` rows of content after `skiprows` lines. The default is
to read all the rows. Note that empty rows containing no data such as
empty lines and comment lines are not counted towards `max_rows`,
while such lines are counted in `skiprows`.
.. versionadded:: 1.16.0
.. versionchanged:: 1.23.0
Lines containing no data, including comment lines (e.g., lines
starting with '#' or as specified via `comments`) are not counted
towards `max_rows`.
quotechar : unicode character or None, optional
The character used to denote the start and end of a quoted item.
Occurrences of the delimiter or comment characters are ignored within
a quoted item. The default value is ``quotechar=None``, which means
quoting support is disabled.
If two consecutive instances of `quotechar` are found within a quoted
field, the first is treated as an escape character. See examples.
.. versionadded:: 1.23.0
${ARRAY_FUNCTION_LIKE}
.. versionadded:: 1.20.0
Returns
-------
out : ndarray
Data read from the text file.
See Also
--------
load, fromstring, fromregex
genfromtxt : Load data with missing values handled as specified.
scipy.io.loadmat : reads MATLAB data files
Notes
-----
This function aims to be a fast reader for simply formatted files. The
`genfromtxt` function provides more sophisticated handling of, e.g.,
lines with missing values.
.. versionadded:: 1.10.0
The strings produced by the Python float.hex method can be used as
input for floats.
Examples
--------
>>> from io import StringIO # StringIO behaves like a file object
>>> c = StringIO("0 1\n2 3")
>>> np.loadtxt(c)
array([[0., 1.],
[2., 3.]])
>>> d = StringIO("M 21 72\nF 35 58")
>>> np.loadtxt(d, dtype={'names': ('gender', 'age', 'weight'),
... 'formats': ('S1', 'i4', 'f4')})
array([(b'M', 21, 72.), (b'F', 35, 58.)],
dtype=[('gender', 'S1'), ('age', '<i4'), ('weight', '<f4')])
>>> c = StringIO("1,0,2\n3,0,4")
>>> x, y = np.loadtxt(c, delimiter=',', usecols=(0, 2), unpack=True)
>>> x
array([1., 3.])
>>> y
array([2., 4.])
The `converters` argument is used to specify functions to preprocess the
text prior to parsing. `converters` can be a dictionary that maps
preprocessing functions to each column:
>>> s = StringIO("1.618, 2.296\n3.141, 4.669\n")
>>> conv = {
... 0: lambda x: np.floor(float(x)), # conversion fn for column 0
... 1: lambda x: np.ceil(float(x)), # conversion fn for column 1
... }
>>> np.loadtxt(s, delimiter=",", converters=conv)
array([[1., 3.],
[3., 5.]])
`converters` can be a callable instead of a dictionary, in which case it
is applied to all columns:
>>> s = StringIO("0xDE 0xAD\n0xC0 0xDE")
>>> import functools
>>> conv = functools.partial(int, base=16)
>>> np.loadtxt(s, converters=conv)
array([[222., 173.],
[192., 222.]])
This example shows how `converters` can be used to convert a field
with a trailing minus sign into a negative number.
>>> s = StringIO('10.01 31.25-\n19.22 64.31\n17.57- 63.94')
>>> def conv(fld):
... return -float(fld[:-1]) if fld.endswith(b'-') else float(fld)
...
>>> np.loadtxt(s, converters=conv)
array([[ 10.01, -31.25],
[ 19.22, 64.31],
[-17.57, 63.94]])
Using a callable as the converter can be particularly useful for handling
values with different formatting, e.g. floats with underscores:
>>> s = StringIO("1 2.7 100_000")
>>> np.loadtxt(s, converters=float)
array([1.e+00, 2.7e+00, 1.e+05])
This idea can be extended to automatically handle values specified in
many different formats:
>>> def conv(val):
... try:
... return float(val)
... except ValueError:
... return float.fromhex(val)
>>> s = StringIO("1, 2.5, 3_000, 0b4, 0x1.4000000000000p+2")
>>> np.loadtxt(s, delimiter=",", converters=conv, encoding=None)
array([1.0e+00, 2.5e+00, 3.0e+03, 1.8e+02, 5.0e+00])
Note that with the default ``encoding="bytes"``, the inputs to the
converter function are latin-1 encoded byte strings. To deactivate the
implicit encoding prior to conversion, use ``encoding=None``
>>> s = StringIO('10.01 31.25-\n19.22 64.31\n17.57- 63.94')
>>> conv = lambda x: -float(x[:-1]) if x.endswith('-') else float(x)
>>> np.loadtxt(s, converters=conv, encoding=None)
array([[ 10.01, -31.25],
[ 19.22, 64.31],
[-17.57, 63.94]])
Support for quoted fields is enabled with the `quotechar` parameter.
Comment and delimiter characters are ignored when they appear within a
quoted item delineated by `quotechar`:
>>> s = StringIO('"alpha, #42", 10.0\n"beta, #64", 2.0\n')
>>> dtype = np.dtype([("label", "U12"), ("value", float)])
>>> np.loadtxt(s, dtype=dtype, delimiter=",", quotechar='"')
array([('alpha, #42', 10.), ('beta, #64', 2.)],
dtype=[('label', '<U12'), ('value', '<f8')])
Two consecutive quote characters within a quoted field are treated as a
single escaped character:
>>> s = StringIO('"Hello, my name is ""Monty""!"')
>>> np.loadtxt(s, dtype="U", delimiter=",", quotechar='"')
array('Hello, my name is "Monty"!', dtype='<U26')
N) r� r� r� r� r� r� r� r� rs r� r� rr c S s$ g | ] }t |t�r |�d �n|�qS )rr )r� rm r� )r� r[ r3 r3 r4 r� 5 s zloadtxt.<locals>.<listcomp>) r� r� r� r� r� r� r� r� rs r� r� )�_loadtxt_with_liker� rm r� r� �float64r� r� )r� r� r� r� r� r� r� r� r� rs r� r� r� r� r� r3 r3 r4 r! 9 s6 j�
�
�r! c C s |fS r- r3 ) r� �X�fmtr� �newline�header�footerr� rs r3 r3 r4 �_savetxt_dispatcherG s r� �%.18e� �
r� �# c C s@ t |t�rt|�}t|�}G dd� d�} d}
t | t�r>t| �} t| �rnt| d��� tj j
j| d|d�} d}
n"t | d�r�| | |p�d�} nt d ���z�t�
|�}|jd
ks�|jd kr�t d |j ��n@|jd
kr�|jjdu r�t�|�j}d
} n t|jj�} n
|jd
} t�|�}
t|�ttfv �rRt|�| k�r<tdt|� ��t|��t t|��}n�t |t��r�|� d�}t d| �}|d
k�r�|
�r�d||f g| }n
|g| }|�|�}n4|
�r�|d | k�r�|�n|
�s�|| k�r�|�n|}nt d|f ��t|�d
k�r"|� dd| �}| � || | � |
�r�|D ]P}g }|D ] }|� |j!� |� |j"� �q8|t|� | }| � |� dd�� �q,nj|D ]d}z|t|� | }W n> t#�y� } z$t#dt|j�|f �|�W Y d}~n
d}~0 0 | � |� �q�t|�d
k�r|� dd| �}| � || | � W |
�r<| �� n|
�r:| �� 0 dS )a�
Save an array to a text file.
Parameters
----------
fname : filename or file handle
If the filename ends in ``.gz``, the file is automatically saved in
compressed gzip format. `loadtxt` understands gzipped files
transparently.
X : 1D or 2D array_like
Data to be saved to a text file.
fmt : str or sequence of strs, optional
A single format (%10.5f), a sequence of formats, or a
multi-format string, e.g. 'Iteration %d -- %10.5f', in which
case `delimiter` is ignored. For complex `X`, the legal options
for `fmt` are:
* a single specifier, `fmt='%.4e'`, resulting in numbers formatted
like `' (%s+%sj)' % (fmt, fmt)`
* a full string specifying every real and imaginary part, e.g.
`' %.4e %+.4ej %.4e %+.4ej %.4e %+.4ej'` for 3 columns
* a list of specifiers, one per column - in this case, the real
and imaginary part must have separate specifiers,
e.g. `['%.3e + %.3ej', '(%.15e%+.15ej)']` for 2 columns
delimiter : str, optional
String or character separating columns.
newline : str, optional
String or character separating lines.
.. versionadded:: 1.5.0
header : str, optional
String that will be written at the beginning of the file.
.. versionadded:: 1.7.0
footer : str, optional
String that will be written at the end of the file.
.. versionadded:: 1.7.0
comments : str, optional
String that will be prepended to the ``header`` and ``footer`` strings,
to mark them as comments. Default: '# ', as expected by e.g.
``numpy.loadtxt``.
.. versionadded:: 1.7.0
encoding : {None, str}, optional
Encoding used to encode the outputfile. Does not apply to output
streams. If the encoding is something other than 'bytes' or 'latin1'
you will not be able to load the file in NumPy versions < 1.14. Default
is 'latin1'.
.. versionadded:: 1.14.0
See Also
--------
save : Save an array to a binary file in NumPy ``.npy`` format
savez : Save several arrays into an uncompressed ``.npz`` archive
savez_compressed : Save several arrays into a compressed ``.npz`` archive
Notes
-----
Further explanation of the `fmt` parameter
(``%[flag]width[.precision]specifier``):
flags:
``-`` : left justify
``+`` : Forces to precede result with + or -.
``0`` : Left pad the number with zeros instead of space (see width).
width:
Minimum number of characters to be printed. The value is not truncated
if it has more characters.
precision:
- For integer specifiers (eg. ``d,i,o,x``), the minimum number of
digits.
- For ``e, E`` and ``f`` specifiers, the number of digits to print
after the decimal point.
- For ``g`` and ``G``, the maximum number of significant digits.
- For ``s``, the maximum number of characters.
specifiers:
``c`` : character
``d`` or ``i`` : signed decimal integer
``e`` or ``E`` : scientific notation with ``e`` or ``E``.
``f`` : decimal floating point
``g,G`` : use the shorter of ``e,E`` or ``f``
``o`` : signed octal
``s`` : string of characters
``u`` : unsigned decimal integer
``x,X`` : unsigned hexadecimal integer
This explanation of ``fmt`` is not complete, for an exhaustive
specification see [1]_.
References
----------
.. [1] `Format Specification Mini-Language
<https://docs.python.org/library/string.html#format-specification-mini-language>`_,
Python Documentation.
Examples
--------
>>> x = y = z = np.arange(0.0,5.0,1.0)
>>> np.savetxt('test.out', x, delimiter=',') # X is an array
>>> np.savetxt('test.out', (x,y,z)) # x,y,z equal sized 1D arrays
>>> np.savetxt('test.out', x, fmt='%1.4e') # use exponential notation
c @ s@ e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dS )zsavetxt.<locals>.WriteWrapz0Convert to bytes on bytestream inputs.
c S s || _ || _| j| _d S r- )r� rs � first_write�do_write)r1 r� rs r3 r3 r4 r5 � s z#savetxt.<locals>.WriteWrap.__init__c S s | j �� d S r- )r� r^ r= r3 r3 r4 r^ � s z savetxt.<locals>.WriteWrap.closec S s | � |� d S r- )r� �r1 �vr3 r3 r4 r� � s z savetxt.<locals>.WriteWrap.writec S s0 t |t�r| j�|� n| j�|�| j�� d S r- )r� rm r� r� �encoders r� r3 r3 r4 � write_bytes� s
z&savetxt.<locals>.WriteWrap.write_bytesc S s | j �t|�� d S r- )r� r� r r� r3 r3 r4 � write_normal� s z'savetxt.<locals>.WriteWrap.write_normalc S s@ z| � |� | j | _W n$ ty: | �|� | j| _Y n0 d S r- )r r� r� r r� r3 r3 r4 r� � s
z&savetxt.<locals>.WriteWrap.first_writeN)
r? r@ rA rB r5 r^ r� r r r� r3 r3 r3 r4 � WriteWrap� s r FZwtr� Tr� rr z%fname must be a string or file handler r� z.Expected 1D or 2D array, got %dD array insteadr Nzfmt has wrong shape. %s�%z'fmt has wrong number of %% formats: %sz (%s+%sj)zinvalid fmt: %rr� z+-�-z?Mismatch between array dtype ('%s') and format specifier ('%s'))$r� rm r r r r ri r^ r� r� r� rD rw Zasarrayr� r� r� r� r� rf r� Z iscomplexobjr� r; r� r9 r� �join�map�count�replacer� rU �real�imagr� )r� r� r� r� r� r� r� r� rs r �own_fhr� ZncolZ iscomplex_Xr Z n_fmt_chars�error�rowZrow2�number�sr� r� r3 r3 r4 r M s� |
!
�
��
�r c C s. d}t | d�s0t�| �} tjjj| d|d�} d}z�t|tj�sHt�|�}|j du rZt
d��| � � }t|t �r�t|t
�r�t|�}n t|t
�r�t|t �r�t|�}t |d�s�t�|�}|�|�}|r�t|d t�s�t�||j d �}tj||d
�}||_ntj||d
�}|W |�r| �� S n|�r(| �� 0 dS ) aq
Construct an array from a text file, using regular expression parsing.
The returned array is always a structured array, and is constructed from
all matches of the regular expression in the file. Groups in the regular
expression are converted to fields of the structured array.
Parameters
----------
file : path or file
Filename or file object to read.
.. versionchanged:: 1.22.0
Now accepts `os.PathLike` implementations.
regexp : str or regexp
Regular expression used to parse the file.
Groups in the regular expression correspond to fields in the dtype.
dtype : dtype or list of dtypes
Dtype for the structured array; must be a structured datatype.
encoding : str, optional
Encoding used to decode the inputfile. Does not apply to input streams.
.. versionadded:: 1.14.0
Returns
-------
output : ndarray
The output array, containing the part of the content of `file` that
was matched by `regexp`. `output` is always a structured array.
Raises
------
TypeError
When `dtype` is not a valid dtype for a structured array.
See Also
--------
fromstring, loadtxt
Notes
-----
Dtypes for structured arrays can be specified in several forms, but all
forms specify at least the data type and field name. For details see
`basics.rec`.
Examples
--------
>>> from io import StringIO
>>> text = StringIO("1312 foo\n1534 bar\n444 qux")
>>> regexp = r"(\d+)\s+(...)" # match [digits, whitespace, anything]
>>> output = np.fromregex(text, regexp,
... [('num', np.int64), ('key', 'S3')])
>>> output
array([(1312, b'foo'), (1534, b'bar'), ( 444, b'qux')],
dtype=[('num', '<i8'), ('key', 'S3')])
>>> output['num']
array([1312, 1534, 444])
FrC r� r� TNz$dtype must be a structured datatype.�matchr �r� )rD r� r� r� r� r� ri r� r� r� r� rC rm r� r r �re�compile�findallr� �arrayr^ ) rG Zregexpr� rs r Zcontent�seqZnewdtype�outputr3 r3 r4 r) E s> >
� �r) )r� r� c C s |fS r- r3 )r� r� r� r� � skip_header� skip_footerr� �missing_values�filling_valuesr� r� � excludelist� deletechars�
replace_space� autostrip�case_sensitive�
defaultfmtr� �usemask�loose�
invalid_raiser� rs r� r� r3 r3 r4 �_genfromtxt_dispatcher� s r& �_zf%ic S sb |durBt | �||�
||||| �| | |
||�||||||||d�S t|� |durn|r^td��|dk rntd��|r�ddlm}m} |p�i }t|t�s�tdt |� ��|d kr�d}d
} nd } t| t
�r�t | �} t| t �r�t
jjj| d |d
�} t�| �} n| } t�| �} z t| �} W n< t�yL } z"tdt | �� d��| �W Y d} ~ n
d} ~ 0 0 | ��� t||||d�}!t| | ||
d�}"zvt�
�D ]�t| � �q�d}#|#�s�tt| �|�}$�d
u �r�|du�r�||$v �r�d�|$�|�dd� �}$|!|$�}#�q�W n. t �y d}$g }#t j d| dd� Y n0 �d
u �rL|#d � � }%|du�rL|%|v �rL|#d= | du�r�zdd� | �d�D �} W n< t �y� z t!| �} W n t�y� | g} Y n0 Y n0 t"| �p�|#�}&�d
u �r�|"dd� |#D ���d}$n2t#���r |"dd� ��d�D ���n��r|"����du�r.t$���| | ||
d���du�r@t!���| �r�t%| �D ]>\�}'t#|'��rp��&|'�| �< n|'dk �rN|'t"|#� | �< �qN�du�r�t"��|&k�rЈj'�t
�(�fd d�| D ���t!�j)��n*�du�rt"��|&k�r�fd d�| D ��n �du�r�du�rt!�j)��|�p"d }(t|(t*��r:|(�+d �}(d d� t|&�D �}t|(t��r&|(�,� D ]�\})}*t#|)��r�z��&|)�})W n t�y� Y �q`Y n0 | �r�z| �&|)�})W n t�y� Y n0 t|*t!t-f��r�d!d� |*D �}*n
t |*�g}*|)du �r|D ]}+|+�.|*� �q�n||) �.|*� �q`n�t|(t!t-f��rjt/|(|�D ]&\},}-t |,�},|,|-v�r@|-�0|,� �q@nJt|(t ��r�|(�d�}.|D ]}-|-�.|.� �q�n |D ]}-|-�.t |(�g� �q�|}/|/du �r�g }/dg|& }t|/t��rX|/�,� D ]p\})}*t#|)��r z��&|)�})W n t�y Y �q�Y n0 | �rJz| �&|)�})W n t�yH Y n0 |*||)< �q�nHt|/t!t-f��r�t"|/�}0|0|&k�r�|/|d|0�< n |/d|&� }n
|/g|& }�du �r�d"d� t/||�D �}nRt1�d
d#�}1t"|1�dk�r�t/|1||�}2d$d� |2D �}n t/||�}2�fd%d�|2D �}g }3|�,� D ]�\}4� t#|4��r`z��&|4�}4|4�W n t�y\ Y �q Y n0 n6| �r�z| �&|4��W n t�y� Y �q Y n0 n|4�t"|$��r�|#|4 }5nd}5� t*u �r�t2}6n"| �r�d&d'� }7t3j4|7� d(�}6n� }6|� j5|6d
|5|� |� d)� |3�0�|6f� �q |�5|3� g � � j0}8|�r6g }9|9j0}:g };|;j0}<t%t6�7|$g| ��D ]�\�}=|!|=��
t"�
�}>|>dk�rx�qR| �r�z�
fd*d�| D ��
W n. t8�y� |<��
d |>f� Y �qRY n0 n"|>|&k�r�|<��
d |>f� �qR|8t-�
�� |� r|:t-d+d� t/�
|�D ��� t"� �|k�rR � q.�qRW d � n1 � sD0 Y �du �
rt%|�D ]�\�}?�fd,d�� D �}@z|?�9|@� W n� t:�
y d-� }At;t<��� �}@t%|@�D ]V\}4},z|?�=|,� W n< t>tf�
y |Ad.7 }A|A|4d �
|,f; }At>|A��Y n0 � q�Y n0 � q`t"|;�}B|Bdk�
r�t"� �|B | �d/|& � |dk�
r|t"��
fd0d�|;D ��}C|;d|B|C � };||C8 }� fd1d�|;D �}At"|A��
r�|A�?dd2� d3�|A�}A|�
r�t|A��nt j |At@dd� |dk�
r�� d| � � |�
r�|9d| � }9|� r t!t/� fd4d�t%|�D �� �� n t!t/� fd5d�t%|�D �� �� � }D�du � r�d6d� |D �}Ed7d� t%|E�D �� | � r؈ � r�t j d8t
jAdd� � fd9d:��z�fd;d�|DD �}DW n tB� y� Y n0 � D ]�t
jC|E�< � q�|Edd� }Ft%|E�D ]<\�}Gt
�D|Gt
jE�� r�tF�fd<d=�|DD ��}H|G|Hf|F�< � q�du � r�d>d?� t/||E�D �}It"|I�dk� rh|I\}J|JtG }K}Ln2�fd@d�t%|F�D �}K|� rfdAd�t%|F�D �}Ln&t!t/�|F��}Kt!t/�tGgt"|F� ��}Lt
jH|D|KdB��|�r�t
jH|9|LdB�}M�n���
r �j)du�
r ��_)t"|1�dk�
r�dCdDd=� |1D �v �
rFtI���
r6tJdE��nt
jH|D�dB��n"t
jH|DdFd� |1D �dB�� � �K���|�r�t
jH|9t
�(dGd� |1D ��dB�}N|��}L|N�K|L�}Mn�|�rVd
}Og �t%dHd� |D ��D ]j\�}P�|v �r|O|P�j kM }Ot
�D|Pt
jE��r
|PtF�fdId=�|DD ��f}P��0d|Pf� n��0d�f� �
q�|O�sVt"��dk�rLt
�(���n
t
�(|P��t
�H|D���|�r��j)du�r�dJd� �j)D �}LntG}Lt
jH|9|LdB�}M�j(j)�|�r���r�t/�|�D ]B\}Q� � fdKd�� jLD �}|D ] }R|M|Q �|Q |RkO < �qq�|�r��K|��|M�_MtN�|dL��|�r^�du �r2�jOS t"��dk�rL��d S �fdMd��D �S �S )Na�
Load data from a text file, with missing values handled as specified.
Each line past the first `skip_header` lines is split at the `delimiter`
character, and characters following the `comments` character are discarded.
Parameters
----------
fname : file, str, pathlib.Path, list of str, generator
File, filename, list, or generator to read. If the filename
extension is ``.gz`` or ``.bz2``, the file is first decompressed. Note
that generators must return bytes or strings. The strings
in a list or produced by a generator are treated as lines.
dtype : dtype, optional
Data type of the resulting array.
If None, the dtypes will be determined by the contents of each
column, individually.
comments : str, optional
The character used to indicate the start of a comment.
All the characters occurring on a line after a comment are discarded.
delimiter : str, int, or sequence, optional
The string used to separate values. By default, any consecutive
whitespaces act as delimiter. An integer or sequence of integers
can also be provided as width(s) of each field.
skiprows : int, optional
`skiprows` was removed in numpy 1.10. Please use `skip_header` instead.
skip_header : int, optional
The number of lines to skip at the beginning of the file.
skip_footer : int, optional
The number of lines to skip at the end of the file.
converters : variable, optional
The set of functions that convert the data of a column to a value.
The converters can also be used to provide a default value
for missing data: ``converters = {3: lambda s: float(s or 0)}``.
missing : variable, optional
`missing` was removed in numpy 1.10. Please use `missing_values`
instead.
missing_values : variable, optional
The set of strings corresponding to missing data.
filling_values : variable, optional
The set of values to be used as default when the data are missing.
usecols : sequence, optional
Which columns to read, with 0 being the first. For example,
``usecols = (1, 4, 5)`` will extract the 2nd, 5th and 6th columns.
names : {None, True, str, sequence}, optional
If `names` is True, the field names are read from the first line after
the first `skip_header` lines. This line can optionally be preceded
by a comment delimiter. If `names` is a sequence or a single-string of
comma-separated names, the names will be used to define the field names
in a structured dtype. If `names` is None, the names of the dtype
fields will be used, if any.
excludelist : sequence, optional
A list of names to exclude. This list is appended to the default list
['return','file','print']. Excluded names are appended with an
underscore: for example, `file` would become `file_`.
deletechars : str, optional
A string combining invalid characters that must be deleted from the
names.
defaultfmt : str, optional
A format used to define default field names, such as "f%i" or "f_%02i".
autostrip : bool, optional
Whether to automatically strip white spaces from the variables.
replace_space : char, optional
Character(s) used in replacement of white spaces in the variable
names. By default, use a '_'.
case_sensitive : {True, False, 'upper', 'lower'}, optional
If True, field names are case sensitive.
If False or 'upper', field names are converted to upper case.
If 'lower', field names are converted to lower case.
unpack : bool, optional
If True, the returned array is transposed, so that arguments may be
unpacked using ``x, y, z = genfromtxt(...)``. When used with a
structured data-type, arrays are returned for each field.
Default is False.
usemask : bool, optional
If True, return a masked array.
If False, return a regular array.
loose : bool, optional
If True, do not raise errors for invalid values.
invalid_raise : bool, optional
If True, an exception is raised if an inconsistency is detected in the
number of columns.
If False, a warning is emitted and the offending lines are skipped.
max_rows : int, optional
The maximum number of rows to read. Must not be used with skip_footer
at the same time. If given, the value must be at least 1. Default is
to read the entire file.
.. versionadded:: 1.10.0
encoding : str, optional
Encoding used to decode the inputfile. Does not apply when `fname` is
a file object. The special value 'bytes' enables backward compatibility
workarounds that ensure that you receive byte arrays when possible
and passes latin1 encoded strings to converters. Override this value to
receive unicode arrays and pass strings as input to converters. If set
to None the system default is used. The default value is 'bytes'.
.. versionadded:: 1.14.0
ndmin : int, optional
Same parameter as `loadtxt`
.. versionadded:: 1.23.0
${ARRAY_FUNCTION_LIKE}
.. versionadded:: 1.20.0
Returns
-------
out : ndarray
Data read from the text file. If `usemask` is True, this is a
masked array.
See Also
--------
numpy.loadtxt : equivalent function when no data is missing.
Notes
-----
* When spaces are used as delimiters, or when no delimiter has been given
as input, there should not be any missing data between two fields.
* When the variables are named (either by a flexible dtype or with `names`),
there must not be any header in the file (else a ValueError
exception is raised).
* Individual values are not stripped of spaces by default.
When using a custom converter, make sure the function does remove spaces.
References
----------
.. [1] NumPy User Guide, section `I/O with NumPy
<https://docs.scipy.org/doc/numpy/user/basics.io.genfromtxt.html>`_.
Examples
--------
>>> from io import StringIO
>>> import numpy as np
Comma delimited file with mixed dtype
>>> s = StringIO(u"1,1.3,abcde")
>>> data = np.genfromtxt(s, dtype=[('myint','i8'),('myfloat','f8'),
... ('mystring','S5')], delimiter=",")
>>> data
array((1, 1.3, b'abcde'),
dtype=[('myint', '<i8'), ('myfloat', '<f8'), ('mystring', 'S5')])
Using dtype = None
>>> _ = s.seek(0) # needed for StringIO example only
>>> data = np.genfromtxt(s, dtype=None,
... names = ['myint','myfloat','mystring'], delimiter=",")
>>> data
array((1, 1.3, b'abcde'),
dtype=[('myint', '<i8'), ('myfloat', '<f8'), ('mystring', 'S5')])
Specifying dtype and names
>>> _ = s.seek(0)
>>> data = np.genfromtxt(s, dtype="i8,f8,S5",
... names=['myint','myfloat','mystring'], delimiter=",")
>>> data
array((1, 1.3, b'abcde'),
dtype=[('myint', '<i8'), ('myfloat', '<f8'), ('mystring', 'S5')])
An example with fixed-width columns
>>> s = StringIO(u"11.3abcde")
>>> data = np.genfromtxt(s, dtype=None, names=['intvar','fltvar','strvar'],
... delimiter=[1,3,5])
>>> data
array((1, 1.3, b'abcde'),
dtype=[('intvar', '<i8'), ('fltvar', '<f8'), ('strvar', 'S5')])
An example to show comments
>>> f = StringIO('''
... text,# of chars
... hello world,11
... numpy,5''')
>>> np.genfromtxt(f, dtype='S12,S12', delimiter=',')
array([(b'text', b''), (b'hello world', b'11'), (b'numpy', b'5')],
dtype=[('f0', 'S12'), ('f1', 'S12')])
N)r� r� r� r r r� r r r� r� r r r r r! r" r� r# r$ r% r� rs r� r� zPThe keywords 'skip_footer' and 'max_rows' can not be specified at the same time.r z 'max_rows' must be at least 1.r )� MaskedArray�make_mask_descrzNThe input argument 'converter' should be a valid dictionary (got '%s' instead)rm TFr� r� z\fname must be a string, a filehandle, a sequence of strings,
or an iterator of strings. Got r� )r� r� r rs )r r r! r r� z"genfromtxt: Empty input file: "%s"r� )r� c S s g | ] }|� � �qS r3 ��strip�r� r' r3 r3 r4 r� � r� z genfromtxt.<locals>.<listcomp>r� c S s g | ]}t |�� ��qS r3 )r� r+ r, r3 r3 r4 r� � r� c S s g | ] }|� � �qS r3 r* r, r3 r3 r4 r� � r� )r" r� r r r! r c s g | ] }� | �qS r3 r3 r, )�descrr3 r4 r� � r� c s g | ] }� | �qS r3 r3 r, )r� r3 r4 r� r� r3 rr c S s g | ]}t d g��qS �r� )r; r, r3 r3 r4 r� r� c S s g | ] }t |��qS r3 )r� r, r3 r3 r4 r� $ r� c S s g | ]\}}t d ||d��qS )N)r �default�r �r� �miss�fillr3 r3 r4 r� f s �)Z flatten_basec S s" g | ]\}}}t |d ||d��qS �T)�lockedr r/ r0 )r� r� r2 r3 r3 r3 r4 r� n s ��c s g | ]\}}t � d ||d��qS r4 r0 r1 r r3 r4 r� t s ��c S s" t | �tu r|| �S || �d��S �Nrr )r� rm r )r[ �convr3 r3 r4 �
tobytes_first� s z!genfromtxt.<locals>.tobytes_first�r7 )r5 �
testing_valuer/ r c s g | ] }� | �qS r3 r3 r, r� r3 r4 r� � r� c S s g | ]\}}|� � |v �qS r3 r* )r� r� �mr3 r3 r4 r� � s �c s g | ]}t � �|��qS r3 )r )r� Z_m�r� r3 r4 r� � r� z0Converter #%i is locked and cannot be upgraded: z"(occurred line #%i for value '%s')z- Line #%%i (got %%i columns instead of %i)c s g | ]}|d � � kr|�qS )r r3 r, )�nbrowsr r3 r4 r� � s �c s g | ]\}}� ||f �qS r3 r3 )r� r� �nb)�templater3 r4 r� � s �zSome errors were detected !r� c s, g | ]$\}� � fd d�t t|���D ��qS )c s g | ]}� � |��qS r3 )Z _loose_call�r� Z_rr9 r3 r4 r� r� �)genfromtxt.<locals>.<listcomp>.<listcomp>�r r �r� r� ��rowsr9 r4 r� s �c s, g | ]$\}� � fd d�t t|���D ��qS )c s g | ]}� � |��qS r3 )Z _strict_callr@ r9 r3 r4 r� r� rA rB rC rD r9 r4 r� s �c S s g | ]
}|j �qS r3 �r� �r� r7 r3 r3 r4 r�
r� c S s g | ]\}}|t jkr|�qS r3 )r� Zunicode_)r� r� r� r3 r3 r4 r� s
�z�Reading unicode strings without specifying the encoding argument is deprecated. Set the encoding, use None for the system default.c s, t | �}� D ]}|| �d�||< q t|�S r6 )r; r r� )Zrow_tupr r� )� strcolidxr3 r4 �encode_unicode_cols s z'genfromtxt.<locals>.encode_unicode_colsc s g | ] }� |��qS r3 r3 )r� �r)rI r3 r4 r� r� c 3 s | ]}t |� �V qd S r- �rf �r� r r< r3 r4 � <genexpr>+ r� z genfromtxt.<locals>.<genexpr>c S s h | ]\}}|j r|�qS r3 )Z_checked)r� r� Zc_typer3 r3 r4 � <setcomp>0 s �z genfromtxt.<locals>.<setcomp>c s g | ]\}}� | |f�qS r3 r3 �r� r� r� �r" r3 r4 r� 8 s �c s g | ]\}}� | t f�qS r3 ��boolrO rP r3 r4 r� ; s �r �Oc s s | ] }|j V qd S r- )�charr, r3 r3 r4 rM M r� z4Nested fields involving objects are not supported...c S s g | ] }d |f�qS r. r3 r, r3 r3 r4 r� T r� c S s g | ] }d t f�qS r. rQ )r� �tr3 r3 r4 r� Y r� c S s g | ]
}|j �qS r3 rF rG r3 r3 r4 r� c r� c 3 s | ]}t |� �V qd S r- rK rL r< r3 r4 rM h r� c S s g | ] }|t f�qS r3 rQ r, r3 r3 r4 r� x r� c s g | ]}|d kr� |��qS r. r3 r, r9 r3 r4 r� � s �r� c s g | ] }� | �qS r3 r3 r� )r r3 r4 r� � r� )P�_genfromtxt_with_liker� rw Znumpy.mar( r) r� rx r� r� r r r� r� r� r� ri ry r� r� rd r r �range�nextr r r� �
StopIterationr� r� r+ r9 r; rf r r r� r r- r� r� rm r� r� r� �extendrV rU r r � functools�partial�update� itertools�chain�
IndexErrorZ iterupgrader r r Zupgrader �insertr ZVisibleDeprecationWarning�UnicodeEncodeErrorZbytes_Z
issubdtype� character�maxrR r r �NotImplementedError�viewr Z_maskr� r� )Sr� r� r� r� r r r� r r r� r� r r r r r! r" r� r# r$ r% r� rs r� r� r( r) Zuser_convertersr� rX Zfid_ctxZfhdr� Z
split_lineZvalidate_namesZ first_valuesZ
first_lineZfvalZnbcols�currentZuser_missing_valuesr: r� r2 r� �entryZ
user_valueZuser_filling_values�nZ
dtype_flatZzipitZ uc_updater� r: Z user_convr8 Zappend_to_rowsZmasksZappend_to_masksZinvalidZappend_to_invalidr� ZnbvaluesZ converterZcurrent_column�errmsgZ nbinvalidZnbinvalid_skippedr� Z column_typesZsized_column_typesZcol_typeZn_chars�baseZ uniform_typeZddtypeZmdtypeZ
outputmaskZrowmasksZ
ishomogeneousZttyper� Zmvalr3 )r7 r" r- r� rI r� r� r= r rE r rH r? r� r4 r" � s D��
��
�����
�
�
� �
�
�
��
*
�
�� ��
�
�
�
�
�
�
r" c K sT |� dd� |�dd�}t| fi |��}|rDddlm} |�|�}n |�tj�}|S )a�
Load ASCII data from a file and return it in a record array.
If ``usemask=False`` a standard `recarray` is returned,
if ``usemask=True`` a MaskedRecords array is returned.
Parameters
----------
fname, kwargs : For a description of input parameters, see `genfromtxt`.
See Also
--------
numpy.genfromtxt : generic function
Notes
-----
By default, `dtype` is None, which means that the data-type of the output
array will be determined from the data.
r� Nr# Fr ��
MaskedRecords)�
setdefault�getr" �numpy.ma.mrecordsrm rf r� �recarray)r� rI r# r rm r3 r3 r4 r# � s r# c K sx |� dd� |� dd� |� dd� |� dd� t| fi |��}|�d d
�}|rhd d lm} |�|�}n |�tj�}|S )
a8
Load ASCII data stored in a comma-separated file.
The returned array is a record array (if ``usemask=False``, see
`recarray`) or a masked record array (if ``usemask=True``,
see `ma.mrecords.MaskedRecords`).
Parameters
----------
fname, kwargs : For a description of input parameters, see `genfromtxt`.
See Also
--------
numpy.genfromtxt : generic function to load ASCII data.
Notes
-----
By default, `dtype` is None, which means that the data-type of the output
array will be determined from the data.
r! �lowerr� Tr� r� r� Nr# Fr rl )rn r" ro rp rm rf r� rq )r� rI r r# rm r3 r3 r4 r$ � s r$ )NFTrq )NN)TT)TN)
NNNNNNNNNN)r� )NNNNNNN)r� r� r� r� r� r� N)N)NNNNNNNNNNNNNNNNNNNNNN)Wr� r r[ r^ r� r. ry r� r r Zopindexr �collections.abcr r* r� r� r r� r Z
numpy.corer Znumpy.core.multiarrayr
r Z numpy.core._multiarray_umathr Znumpy.core.overridesr
r Z_iotoolsr r r r r r r r r r r Z numpy.compatr r r r r r �__all__r\ Zarray_function_dispatchr, rJ rK rp r% r� r& r� r' r� r( r� r� �intr� r� r� r� r� r� r� �floatr! r� r� r r) r& r �sortedZdefaultdeletecharsr" rV r# r$ r3 r3 r3 r4 �<module> s� 4 �1 �� 8
J
V
A
%
��
� �� �� �
� x g �� �� c��
Mini Shell