Python Pandas DataFrame to_json方法用法及代碼示例

例子

輸出到文件

考慮以下 DataFrame ：

df = pd.DataFrame({"A":[2,3], "B":[4,5]}, index=["a","b"])
df



   A  B
a  2  4
b  3  5

我們可以通過傳入 path_or_buf 將 JSON 輸出到文件，而不是獲取 JSON 字符串，如下所示：

df.to_json(path_or_buf="my_json")

這將在與 Python 腳本相同的目錄中創建一個名為 my_json 的新文件：

{"A":{"a":2,"b":3},"B":{"a":4,"b":5}}

指定方向

考慮以下 DataFrame ：

df = pd.DataFrame({"A":[2,3], "B":[4,5]}, index=["a","b"])
df



   A  B
a  2  4
b  3  5

默認

默認情況下，orient="columns"：

df.to_json()



'{"A":{"a":2,"b":3},"B":{"a":4,"b":5}}'

分裂

df.to_json(orient="split")



'{"columns":["A","B"],"index":["a","b"],"data":[[2,4],[3,5]]}'

記錄

df.to_json(orient="records")



'[{"A":2,"B":4},{"A":3,"B":5}]'

index

df.to_json(orient="index")



'{"a":{"A":2,"B":4},"b":{"A":3,"B":5}}'

列

df.to_json(orient="columns")



'{"A":{"a":2,"b":3},"B":{"a":4,"b":5}}'

值

df.to_json(orient="values")



'[[2,4],[3,5]]'

表格

df.to_json(orient="table")



'{"schema":{"fields":[{"name":"index","type":"string"},
                      {"name":"A","type":"integer"},
                      {"name":"B","type":"integer"}],
            "primaryKey":["index"],
            "pandas_version":"0.20.0"}
  "data":[{"index":"a","A":2,"B":4},
          {"index":"b","A":3,"B":5}]}'

在這裏，為了清晰起見，我們對輸出進行了美化，但實際輸出全部在一行中。

指定date_format

考慮以下 DataFrame ：

df = pd.DataFrame({"A":["2020-12-25"], "B":["2020-12-20"]}, dtype="datetime64[ns]")
df



   A           B
0  2020-12-25  2020-12-20

默認情況下， date_format="epoch" (除非 orient="table" )：

df.to_json()    # date_format="epoch"



'{"A":{"0":1608854400000},"B":{"0":1608422400000}}'

這裏，大數字表示自 1970-01-01 以來經過的時間(以毫秒為單位)。

另一方麵，我們可以通過傳入 "iso" 將日期格式更改為 iso8601，如下所示：

df.to_json(date_format="iso")



'{"A":{"0":"2020-12-25T00:00:00.000Z"},"B":{"0":"2020-12-20T00:00:00.000Z"}}'

指定date_unit

考慮以下 DataFrame ：

df = pd.DataFrame({"A":["2020-12-25"], "B":["2020-12-20"]}, dtype="datetime64[ns]")
df



   A           B
0  2020-12-25  2020-12-20

默認情況下，當 date_format="epoch" 時，單位為毫秒：

df.to_json()    # date_format="epoch"



'{"A":{"0":1608854400000},"B":{"0":1608422400000}}'

這裏，大數字表示自 1970-01-01 以來經過的時間(以毫秒為單位)。

我們可以將單位更改為秒，如下所示：

df.to_json(date_unit="s")    # date_format="epoch"



'{"A":{"0":1608854400},"B":{"0":1608422400}}'

指定default_handler

考慮以下 DataFrame ：

df = pd.DataFrame({"A":[complex(3,4)], "B":[4]}, index=["a"])
df



   A                   B
a  3.000000+4.000000j  4

這裏，我們的df有一個複數。

默認情況下，當我們嘗試將 df 轉換為 JSON 字符串時，我們會得到以下結果：

df.to_json(orient="records")



'[{"A":{"imag":4.0},"B":4}]'

請注意複數如何錯誤地表示為 "imag" ，這是因為 JSON 不知道如何在內部解析複數。對於像這樣轉換不正確的情況，我們可以使用default_handler參數來控製返回的內容：

def my_handler(my_df):
    return pd.DataFrame({"A":[3], "B":[4]}, index=["a"])
df.to_json(orient="records", default_handler=my_handler)



'[{"A":[{"A":3,"B":4}],"B":4}]'

該處理程序將源 DataFrame 作為參數，並返回一個可序列化的對象，如Map、Series、DataFrame 等。現在，我們可以返回我們喜歡的另一個 JSON，而不是之前的格式錯誤的 JSON。

指定線路

考慮以下 DataFrame ：

df = pd.DataFrame({"A":[2,3], "B":[4,5]}, index=["a","b"])
df



   A  B
a  2  4
b  3  5

當 orient="records" 時，我們可以通過傳入 lines=True 使每個項目出現在新行中，如下所示：

my_json = df.to_json(orient="records", lines=True)
my_json



'{"A":2,"B":4}\n{"A":3,"B":5}'

當我們打印出字符串 my_json 時，我們會看到 \n 生效：

print(my_json)



{"A":2,"B":4}
{"A":3,"B":5}

指定縮進

我們可以通過傳入 indent 參數來添加空格縮進，如下所示：

my_json = df.to_json(orient="records", indent=3)
print(my_json)



[
   {
      "A":2,
      "B":4
   },
   {
      "A":3,
      "B":5
   }
]

這裏，第二行有 3 個空格，第三行有 6 個空格，依此類推。