使用 pandas 將 CSV 解析為 python 字典

Question

我將以下 DataFrame 解析為 Python：

df = pd.read_csv("my_file.csv")

結果：

   indexes X_values Y_values
0       IDX1     x1      y1
1       IDX1     x2      y2
2       IDX1     x3      y3
3       IDX1     x4      y4
6       IDX2     x1      y1
9       IDX2     x4      y4
10      IDX3     x1      y1
11      IDX3     x2      y2

我需要創建字典，每個索引作為鍵，x_values & y_values 作為嵌套字典的列表。 輸出應該是這樣的：

{"IDX1" : [{"x1": "y1"}, {"x2": "y2"}, {"x3": "y3"}, {"x4": "y4"}],
"IDX2": [{"x1": "y1"},{"x4": "y4"}],
"IDX3":[{"x1": "y1"}, {"x2": "y2"}]}

我正在嘗試使用 set_index() 方法解析它，但總是缺少一些東西。 你可以幫幫我嗎？

此外，以索引為鍵的嵌套字典的字典也是很好的解決方案。

Answer 1

我們可以做的

d = df[['X_values','Y_values']].apply(lambda x : {x[0]:x[1]},axis=1).groupby(df['indexes']).agg(list).to_dict()
Out[104]: 
{'IDX1': [{'x1': 'y1'}, {'x2': 'y2'}, {'x3': 'y3'}, {'x4': 'y4'}],
 'IDX2': [{'x1': 'y1'}, {'x4': 'y4'}],
 'IDX3': [{'x1': 'y1'}, {'x2': 'y2'}]}

Answer 2

你可以試試

out = (df.apply(lambda row: {row['X_values']: row['Y_values']}, axis=1)
       .groupby(df['indexes']).agg(list).to_dict())

print(out)

{'IDX1': [{'x1': 'y1'}, {'x2': 'y2'}, {'x3': 'y3'}, {'x4': 'y4'}], 'IDX2': [{'x1': 'y1'}, {'x4': 'y4'}], 'IDX3': [{'x1': 'y1'}, {'x2': 'y2'}]}

Answer 3

聽起來您需要使用to_dict()方法（ 文檔）。

您可能需要像您說的那樣進行索引。 我會嘗試：

df.set_index('indexes').T.to_dict('list')