pandas中dataframe与dict相互转换

stone / 2024-10-10 / 原文

dataframe批量处理数据非常方便,但是在遍历时,需要使用json的records格式。而json格式又不方便在python中使用,所以,又需要将json转为list[dict]类型。

1. dataframe转dict,使用json的records格式

    import pandas as pd
    import numpy as np
    import json
    row_num=100
    dataframe_init=pd.DataFrame({'col1':range(row_num), 'col2':np.random.rand((row_num))})
    json_data=dataframe_init.to_json(orient='records')
    dict_data=json.loads(json_data)

可以先让dataframe转成json的records格式,方便后续使用。

2. json转dataframe,使用json的records格式

  dataframe_json=pd.read_json(json_data)

直接使用json的records格式来转换成dataframe。