將 CSV 文件轉換為 json 遍歷列

Question

我有一個看起來像這樣的 csv 文件：

Question Answer
a           A 
b           NaN
c           NaN
d           B 
e           NaN
f           NaN

我想要的是創建一個字典（或 json 文件），例如：

[{'answer': 'A', 'questions':'a', 'b', 'c'}, {'answer': 'B', 'questions':'d', 'e', 'f'}]

實現它的最有效方法是什么？

Answer 1

我們可以在groupby之前檢查ffill

df.groupby(df.Answer.ffill()).Question.agg(list).reset_index().to_dict('r')
Out[17]: 
[{'Answer': 'A', 'Question': ['a', 'b', 'c']},
 {'Answer': 'B', 'Question': ['d', 'e', 'f']}]