python - 在 pandas 中使用 to_json 时如何保留 DataFrame 索引名称？

Question

现在我有一个如下所示的 DataFrame df2:

       Kuwaiti  Non-Kuwaiti   Total
Age                                
0-4     164483       156459  320942
5-9     158377       136935  295312

当我这样做时:

df2.to_json()

我得到:

'{"Kuwaiti":{"0-4":164483,"5-9":158377},"Non-Kuwaiti":{"0-4":156459,"5-9":136935},"Total":{"0-4":320942,"5-9":295312}}'

如您所见，df2.index.name 未在任何地方保留。

如何保留索引名称？

Answer 1

不幸的是，我们可以传递给 df2.to_json() 的 orient 参数都不会保留索引名称。一个解决方案是:

1. 重新索引 DataFrame 以便您要保留的索引成为普通列:

   df2.reset_index(inplace=True)
   

   DataFrame df2 现在是:

       Age    Kuwaiti  Non-Kuwaiti   Total
   0   0-4     164483       156459  320942
   1   5-9     158377       136935  295312
   

2. 将其保存为 JSON(理想情况下使用不保留索引以节省空间的 orient，'split' 是最节省空间的)。

   df2.to_json('file.json', orient='split')
   

3. 加载并重新索引。

   df3 = pd.read_json('file.json', orient='split').set_index('Age')
   

   并且 df3.index.name 将如预期的那样:

          Kuwaiti  Non-Kuwaiti   Total
   Age
   0-4     164483       156459  320942
   5-9     158377       136935  295312