B x\@szdZddlZddlZddlmZmZddlmZm Z m Z m Z ddlmZmZde jfddZdd d Zd d Zd dZdS)z pickle compat N) read_array write_array)PY3BytesIOcPickle pickle_compat) _get_handle_stringify_pathinferc Cs`t|}t|d|dd\}}|dkr*tj}z|tj||dWdx|D] }|qJWXdS)a Pickle (serialize) object to file. Parameters ---------- obj : any object Any python object. path : str File path where the pickled object will be stored. compression : {'infer', 'gzip', 'bz2', 'zip', 'xz', None}, default 'infer' A string representing the compression to use in the output file. By default, infers from the file extension in specified path. .. versionadded:: 0.20.0 protocol : int Int which indicates which protocol should be used by the pickler, default HIGHEST_PROTOCOL (see [1], paragraph 12.1.2). The possible values for this parameter depend on the version of Python. For Python 2.x, possible values are 0, 1, 2. For Python>=3.0, 3 is a valid value. For Python >= 3.4, 4 is a valid value. A negative value for the protocol parameter is equivalent to setting its value to HIGHEST_PROTOCOL. .. [1] https://docs.python.org/3/library/pickle.html .. versionadded:: 0.21.0 See Also -------- read_pickle : Load pickled pandas object (or any object) from file. DataFrame.to_hdf : Write DataFrame to an HDF5 file. DataFrame.to_sql : Write DataFrame to a SQL database. DataFrame.to_parquet : Write a DataFrame to the binary parquet format. Examples -------- >>> original_df = pd.DataFrame({"foo": range(5), "bar": range(5, 10)}) >>> original_df foo bar 0 0 5 1 1 6 2 2 7 3 3 8 4 4 9 >>> pd.to_pickle(original_df, "./dummy.pkl") >>> unpickled_df = pd.read_pickle("./dummy.pkl") >>> unpickled_df foo bar 0 0 5 1 1 6 2 2 7 3 3 8 4 4 9 >>> import os >>> os.remove("./dummy.pkl") wbF) compressionis_textr)protocolN)r rpklHIGHEST_PROTOCOLwritedumpsclose)objpathr rffh_fr/lib/python3.7/site-packages/pandas/io/pickle.py to_pickle s:  rcsXtfdddfdd }y|Stk rRtrL|ddSYnXdS) a@ Load pickled pandas object (or any object) from file. .. warning:: Loading pickled data received from untrusted sources can be unsafe. See `here `__. Parameters ---------- path : str File path where the pickled object will be loaded. compression : {'infer', 'gzip', 'bz2', 'zip', 'xz', None}, default 'infer' For on-the-fly decompression of on-disk data. If 'infer', then use gzip, bz2, xz or zip if path ends in '.gz', '.bz2', '.xz', or '.zip' respectively, and no decompression otherwise. Set to None for no decompression. .. versionadded:: 0.20.0 Returns ------- unpickled : same type as object stored in file See Also -------- DataFrame.to_pickle : Pickle (serialize) DataFrame object to file. Series.to_pickle : Pickle (serialize) Series object to file. read_hdf : Read HDF5 file into a DataFrame. read_sql : Read SQL query or database table into a DataFrame. read_parquet : Load a parquet object, returning a DataFrame. Examples -------- >>> original_df = pd.DataFrame({"foo": range(5), "bar": range(5, 10)}) >>> original_df foo bar 0 0 5 1 1 6 2 2 7 3 3 8 4 4 9 >>> pd.to_pickle(original_df, "./dummy.pkl") >>> unpickled_df = pd.read_pickle("./dummy.pkl") >>> unpickled_df foo bar 0 0 5 1 1 6 2 2 7 3 3 8 4 4 9 >>> import os >>> os.remove("./dummy.pkl") c s:tddd\}}z||Sx|D] }|q$WXdS)NrbF)r r )rr)funcrrr)r rrr read_wrappers  z!read_pickle..read_wrapperNc s|y0tjddtdtddSQRXWnFtk rvyfddStk rpfddSXYnXdS)NT)recordignorecSs t|S)N)rload)rrrrsz/read_pickle..try_read..cstj|ddS)NF)encodingcompat)pcr!)r)r#rrr"scstj|ddS)NT)r#r$)r%r!)r)r#rrr"s)warningscatch_warnings simplefilterWarning Exception)rr#)r)r#rtry_reads  zread_pickle..try_readlatin1)r#)N)r r*r)rr r+r)r rrr read_pickleSs9  r-cCs$|tj}t}t|||S)N)ZviewnpZndarrayrrgetvalue)arrZbufrrr _pickle_arrays  r1cCstt|}|S)N)rr)bytesr0rrr_unpickle_arrays r3)r )__doc__r&Znumpyr.Znumpy.lib.formatrrZ pandas.compatrrrrrr%Zpandas.io.commonrr rrr-r1r3rrrrsG j