U b,D @s ddlmZddlmZmZddlZddlmZmZm Z ddl Z ddl m Z ddlmZddlmZddlZddlmZd d d d d Zd,d d dddddZdd dd ddddZdd ddddZd-dd dddd Zd.dd"d#d$d$d d dd%d& d'd(Zed)ed*d+ZdS)/) annotations)abc defaultdictN)Any DefaultDictIterableconvert_json_to_lines)Scalar) deprecate) DataFramestr)sreturncCs0|ddks|ddkr|S|dd}t|S)zJ Helper function that converts JSON lists to line delimited JSON. r[]r)rr=/tmp/pip-unpacked-wheel-ck39h295/pandas/io/json/_normalize.pyconvert_to_line_delimitss r.intz int | None)prefixseplevel max_levelc Csd}t|tr|g}d}g}|D]}t|}|D]\} } t| tsPt| } |dkr^| } n ||| } t| tr|dk r||kr|dkr6|| } | || <q6q6|| } |t| | ||d|q6| |q |r|dS|S)a A simplified json_normalize Converts a nested dict into a flat dict ("record"), unlike json_normalize, it does not attempt to extract a subset of the data. Parameters ---------- ds : dict or list of dicts prefix: the prefix, optional, default: "" sep : str, default '.' Nested records will generate names separated by sep, e.g., for sep='.', { 'foo' : { 'bar' : 0 } } -> foo.bar level: int, optional, default: 0 The number of levels in the json string. max_level: int, optional, default: None The max depth to normalize. .. versionadded:: 0.25.0 Returns ------- d - dict or list of dicts, matching `ds` Examples -------- >>> nested_to_record( ... dict(flat1=1, dict1=dict(c=1, d=2), nested=dict(e=dict(c=1, d=2), d=2)) ... ) {'flat1': 1, 'dict1.c': 1, 'dict1.d': 2, 'nested.e.c': 1, 'nested.e.d': 2, 'nested.d': 2} FTrNr) isinstancedictcopydeepcopyitemsr popupdatenested_to_recordappend) dsrrrrZ singletonZnew_dsdZnew_dkvZnewkeyrrrr%'s8.        r%rdict[str, Any])data key_stringnormalized_dict separatorrcCsnt|trb|D]L\}}|||}t||dt||krF|n|t|d||dqn|||<|S)a3 Main recursive function Designed for the most basic use case of pd.json_normalize(data) intended as a performance improvement, see #15621 Parameters ---------- data : Any Type dependent on types contained within nested Json key_string : str New key (with separator(s) in) for data normalized_dict : dict The new normalized/flattened Json dict separator : str, default '.' Nested records will generate names separated by sep, e.g., for sep='.', { 'foo' : { 'bar' : 0 } } -> foo.bar Nr,r-r.r/)rrr"_normalise_jsonlen)r,r-r.r/keyvalueZnew_keyrrrr1zs  r1)r,r/rcCs8dd|D}tdd|Ddi|d}||S)aw Order the top level keys and then recursively go to depth Parameters ---------- data : dict or list of dicts separator : str, default '.' Nested records will generate names separated by sep, e.g., for sep='.', { 'foo' : { 'bar' : 0 } } -> foo.bar Returns ------- dict or list of dicts, matching `normalised_json_object` cSs i|]\}}t|ts||qSrrr.0r)r*rrr s z+_normalise_json_ordered..cSs i|]\}}t|tr||qSrr5r6rrrr8s rr0)r"r1)r,r/Z top_dict_Z nested_dict_rrr_normalise_json_orderedsr9zdict | list[dict]zdict | list[dict] | Any)r'rrcs@i}t|trt|d}n t|tr<fdd|D}|S|S)a A optimized basic json_normalize Converts a nested dict into a flat dict ("record"), unlike json_normalize and nested_to_record it doesn't do anything clever. But for the most basic use cases it enhances performance. E.g. pd.json_normalize(data) Parameters ---------- ds : dict or list of dicts sep : str, default '.' Nested records will generate names separated by sep, e.g., for sep='.', { 'foo' : { 'bar' : 0 } } -> foo.bar Returns ------- frame : DataFrame d - dict or list of dicts, matching `normalised_json_object` Examples -------- >>> _simple_json_normalize( ... { ... "flat1": 1, ... "dict1": {"c": 1, "d": 2}, ... "nested": {"e": {"c": 1, "d": 2}, "d": 2}, ... } ... ) {'flat1': 1, 'dict1.c': 1, 'dict1.d': 2, 'nested.e.c': 1, 'nested.e.d': 2, 'nested.d': 2} )r,r/csg|]}t|dqS)r)_simple_json_normalize)r7rowr:rr sz*_simple_json_normalize..)rrr9list)r'rZnormalised_json_objectZnormalised_json_listrr:rr;s+  r;raisezstr | list | Nonez"str | list[str | list[str]] | Nonez str | Noner ) r, record_pathmeta meta_prefix record_prefixerrorsrrrc sddddddfdd ddd d fd d t|trD|sDtSt|trV|g}n$t|tjrvt|tsvt|}nt|d kr|d kr|d kr d krd krtt| dS|d krt dd|Drt | d}t|St|ts|g}|d krg}nt|ts|g}dd|Dg gt t fddDd  f dd ||iddt } d k r|j fddd} D]N\} } |d k r|| } | |krtd| dtj| td|| <q|S)!a Normalize semi-structured JSON data into a flat table. Parameters ---------- data : dict or list of dicts Unserialized JSON objects. record_path : str or list of str, default None Path in each object to list of records. If not passed, data will be assumed to be an array of records. meta : list of paths (str or list of str), default None Fields to use as metadata for each record in resulting table. meta_prefix : str, default None If True, prefix records with dotted (?) path, e.g. foo.bar.field if meta is ['foo', 'bar']. record_prefix : str, default None If True, prefix records with dotted (?) path, e.g. foo.bar.field if path to records is ['foo', 'bar']. errors : {'raise', 'ignore'}, default 'raise' Configures error handling. * 'ignore' : will ignore KeyError if keys listed in meta are not always present. * 'raise' : will raise KeyError if keys listed in meta are not always present. sep : str, default '.' Nested records will generate names separated by sep. e.g., for sep='.', {'foo': {'bar': 0}} -> foo.bar. max_level : int, default None Max number of levels(depth of dict) to normalize. if None, normalizes all levels. .. versionadded:: 0.25.0 Returns ------- frame : DataFrame Normalize semi-structured JSON data into a flat table. Examples -------- >>> data = [ ... {"id": 1, "name": {"first": "Coleen", "last": "Volk"}}, ... {"name": {"given": "Mark", "family": "Regner"}}, ... {"id": 2, "name": "Faye Raker"}, ... ] >>> pd.json_normalize(data) id name.first name.last name.given name.family name 0 1.0 Coleen Volk NaN NaN NaN 1 NaN NaN NaN Mark Regner NaN 2 2.0 NaN NaN NaN NaN Faye Raker >>> data = [ ... { ... "id": 1, ... "name": "Cole Volk", ... "fitness": {"height": 130, "weight": 60}, ... }, ... {"name": "Mark Reg", "fitness": {"height": 130, "weight": 60}}, ... { ... "id": 2, ... "name": "Faye Raker", ... "fitness": {"height": 130, "weight": 60}, ... }, ... ] >>> pd.json_normalize(data, max_level=0) id name fitness 0 1.0 Cole Volk {'height': 130, 'weight': 60} 1 NaN Mark Reg {'height': 130, 'weight': 60} 2 2.0 Faye Raker {'height': 130, 'weight': 60} Normalizes nested data up to level 1. >>> data = [ ... { ... "id": 1, ... "name": "Cole Volk", ... "fitness": {"height": 130, "weight": 60}, ... }, ... {"name": "Mark Reg", "fitness": {"height": 130, "weight": 60}}, ... { ... "id": 2, ... "name": "Faye Raker", ... "fitness": {"height": 130, "weight": 60}, ... }, ... ] >>> pd.json_normalize(data, max_level=1) id name fitness.height fitness.weight 0 1.0 Cole Volk 130 60 1 NaN Mark Reg 130 60 2 2.0 Faye Raker 130 60 >>> data = [ ... { ... "state": "Florida", ... "shortname": "FL", ... "info": {"governor": "Rick Scott"}, ... "counties": [ ... {"name": "Dade", "population": 12345}, ... {"name": "Broward", "population": 40000}, ... {"name": "Palm Beach", "population": 60000}, ... ], ... }, ... { ... "state": "Ohio", ... "shortname": "OH", ... "info": {"governor": "John Kasich"}, ... "counties": [ ... {"name": "Summit", "population": 1234}, ... {"name": "Cuyahoga", "population": 1337}, ... ], ... }, ... ] >>> result = pd.json_normalize( ... data, "counties", ["state", "shortname", ["info", "governor"]] ... ) >>> result name population state shortname info.governor 0 Dade 12345 Florida FL Rick Scott 1 Broward 40000 Florida FL Rick Scott 2 Palm Beach 60000 Florida FL Rick Scott 3 Summit 1234 Ohio OH John Kasich 4 Cuyahoga 1337 Ohio OH John Kasich >>> data = {"A": [1, 2]} >>> pd.json_normalize(data, "A", record_prefix="Prefix.") Prefix.0 0 1 1 2 Returns normalized data with columns prefixed with the given string. Fr+z list | strboolzScalar | Iterable)jsspecextract_recordrc s|}z:t|tr4|D]}|dkr(t|||}qn||}Wnhtk r}zJ|rhtd|d|n.dkr~tjWYStd|d|d|W5d}~XYnX|S)zInternal function to pull fieldNzKey zS not found. If specifying a record_path, all elements of data should have the path.ignorez) not found. To replace missing values of z% with np.nan, pass in errors='ignore')rr>KeyErrornpnan)rFrGrHresultfielde)rDrr _pull_fields.    z$_json_normalize.._pull_fieldr>)rFrGrcsF||dd}t|tsBt|r(g}nt|d|d|d|S)z Internal function to pull field for records, and similar to _pull_field, but require to return list. And will raise error if has non iterable value. T)rHz has non list value z for path z. Must be list or null.)rr>pdZisnull TypeError)rFrGrM)rPrr _pull_recordss  z&_json_normalize.._pull_recordsNr:css |]}dd|DVqdS)cSsg|]}t|tqSrr5)r7xrrrr=sz-_json_normalize...N)values)r7yrrr sz"_json_normalize..rrcSs g|]}t|tr|n|gqSr)rr>)r7mrrrr=sz#_json_normalize..csg|]}|qSr)join)r7valr:rrr=src st|tr|g}t|dkr|D]^}tD]*\}}|dt|kr.||d||<q.||d|dd||ddq n|D]}||d} fdd|D}t|tD]B\}}|dt|kr||}n|||d}||qĈ|qdS)Nrrrrcs(g|] }t|tr t|dn|qS)rX)rrr%)r7r)rrrrr=sz?_json_normalize.._recursive_extract..)rrr2zipr&extend) r,pathZ seen_metarobjr[r3ZrecsZmeta_val) _metarPrS_recursive_extractlengthsr meta_keys meta_valsrecordsrrrrcs(  (  z+_json_normalize.._recursive_extractr\cs |S)Nr)rT)rCrr z!_json_normalize..)columnszConflicting metadata name z, need distinguishing prefix )Zdtype)F)r)rr>r rrrr NotImplementedErrorr;anyr%rrenamer" ValueErrorrKarrayobjectrepeat) r,r@rArBrCrDrrrMr)r*r) rbrPrSrcrDrdrrerfrCrgrr_json_normalizesj           rrzpandas.io.json.json_normalizez1.0.0zpandas.json_normalize)rrrN)r)NNNNr?rN) __future__r collectionsrrr typingrrrZnumpyrKZpandas._libs.writersr Zpandas._typingr Zpandas.util._decoratorsr ZpandasrQr rr%r1r9r;rrZjson_normalizerrrrsF     S)7 ,