Skip to content

Commit dcbb22b

Browse files
author
Albert Villanova del Moral
committed
BUG: Fix read_json orient='table' without index (pandas-dev#25170)
1 parent e3b0950 commit dcbb22b

File tree

2 files changed

+17
-7
lines changed

2 files changed

+17
-7
lines changed

Diff for: pandas/io/json/table_schema.py

+8-7
Original file line numberDiff line numberDiff line change
@@ -314,12 +314,13 @@ def parse_table_schema(json, precise_float):
314314

315315
df = df.astype(dtypes)
316316

317-
df = df.set_index(table['schema']['primaryKey'])
318-
if len(df.index.names) == 1:
319-
if df.index.name == 'index':
320-
df.index.name = None
321-
else:
322-
df.index.names = [None if x.startswith('level_') else x for x in
323-
df.index.names]
317+
if 'primaryKey' in table['schema']:
318+
df = df.set_index(table['schema']['primaryKey'])
319+
if len(df.index.names) == 1:
320+
if df.index.name == 'index':
321+
df.index.name = None
322+
else:
323+
df.index.names = [None if x.startswith('level_') else x for x in
324+
df.index.names]
324325

325326
return df

Diff for: pandas/tests/io/json/test_pandas.py

+9
Original file line numberDiff line numberDiff line change
@@ -1262,3 +1262,12 @@ def test_index_false_error_to_json(self, orient):
12621262
"'orient' is 'split' or 'table'")
12631263
with pytest.raises(ValueError, match=msg):
12641264
df.to_json(orient=orient, index=False)
1265+
1266+
@pytest.mark.parametrize('orient', ['split', 'table'])
1267+
def test_index_false_from_json_to_json(self, orient):
1268+
# GH25170
1269+
# Test index=False in from_json to_json
1270+
df = DataFrame({'a': [1, 2], 'b': [3, 4]})
1271+
dfjson = df.to_json(orient=orient, index=False)
1272+
result = read_json(dfjson, orient=orient)
1273+
assert_frame_equal(result, df)

0 commit comments

Comments
 (0)