BUG: Fix json_normalize throwing TypeError (#21536) (#21540)

Author: vuminhle

doc/source/whatsnew/v0.23.2.txt

@@ -71,7 +71,7 @@ Bug Fixes
 **I/O**
 
 - Bug in :func:`read_csv` that caused it to incorrectly raise an error when ``nrows=0``, ``low_memory=True``, and ``index_col`` was not ``None`` (:issue:`21141`)
--
+- Bug in :func:`json_normalize` when formatting the ``record_prefix`` with integer columns (:issue:`21536`)
 -
 
 **Plotting**

pandas/io/json/normalize.py

@@ -170,6 +170,11 @@ def json_normalize(data, record_path=None, meta=None,
  3 Summit 1234 John Kasich Ohio OH
  4 Cuyahoga 1337 John Kasich Ohio OH
 
+ >>> data = {'A': [1, 2]}
+ >>> json_normalize(data, 'A', record_prefix='Prefix.')
+ Prefix.0
+ 0 1
+ 1 2
  """
  def _pull_field(js, spec):
  result = js
@@ -259,7 +264,8 @@ def _recursive_extract(data, path, seen_meta, level=0):
  result = DataFrame(records)
 
  if record_prefix is not None:
- result.rename(columns=lambda x: record_prefix + x, inplace=True)
+ result = result.rename(
+ columns=lambda x: "{p}{c}".format(p=record_prefix, c=x))
 
  # Data types, a problem
  for k, v in compat.iteritems(meta_vals):

pandas/tests/io/json/test_normalize.py

@@ -123,6 +123,12 @@ def test_simple_normalize_with_separator(self, deep_nested):
  'country', 'states_name']).sort_values()
  assert result.columns.sort_values().equals(expected)
 
+ def test_value_array_record_prefix(self):
+ # GH 21536
+ result = json_normalize({'A': [1, 2]}, 'A', record_prefix='Prefix.')
+ expected = DataFrame([[1], [2]], columns=['Prefix.0'])
+ tm.assert_frame_equal(result, expected)
+
  def test_more_deeply_nested(self, deep_nested):
 
  result = json_normalize(deep_nested, ['states', 'cities'],