python - 如何将包含列表的字典转换为列的不同类型的字典

Question

我有一个像这样的数据框:

import pandas as pd
    frame={'location': {(2, 'eng', 'US'): {"['sc']": 3, "['delhi']": 2, "['sonepat', 'delhi']": 1, "['new delhi']": 1}}}
    df=pd.DataFrame(frame)
    df.head()

输出

                            location
2       eng       US    {"['sc']": 3, "['delhi']": 2, "['sonepat', 'delhi']": 1, "['new delhi']": 1}

  

我想将 df['location'] 列内的类型更改为不列出，例如:

                            location
2       eng       US    {'sc': 3, 'delhi': 2, 'sonepat', 'delhi': 1, 'new delhi': 1}

Answer 1

你可以试试这个:

import pandas as pd


df = pd.DataFrame(
    {
        "location": {
            (2, "eng", "US"): {
                "['sc']": 3,
                "['delhi']": 2,
                "['sonepat', 'delhi']": 1,
                "['new delhi']": 1,
            }
        }
    }
)

df["location"] = df["location"].apply(
    lambda x: str(x).replace("['", "").replace("']", "")
)

print(df)
#Outputs
                                                   location
2 eng US  {"sc": 3, "delhi": 2, "sonepat', 'delhi": 1, "...