小组回答&基于 python 中的问题的索引

Question

我有一个用于分析目的的数据框，我需要创建一个字典列表：

目标输出

[
{ 'is my anti hiv test conclusive or--Bla bla': [0, 1, 2] }, 
{'I have some hip pain 9 weeks--bla bla': [3, 4, 5, 6]} 
]

这里列表是答案索引，而不是实际的答案

是的，显而易见的方法是使用 groupby 但面临一些错误

我在转换之前尝试打印列出。实际上看起来不错，

你们能帮我弄清楚它的正确语法，以便我可以达到我的目标输出。

数据集链接 如果有人需要共享笔记本链接，请在评论中告诉我。

Answer 1

您需要实际选择要在列表中显示其值的列（“index”）：

df_ans = data.groupby(["question_text"])["index"].apply(list).to_dict()

而不是否则

df_ans = data.groupby(["question_text"]).apply(list).to_dict()

您将获得列的列表，如示例所示。这就是当您将 DataFrame 转换为列表时发生的情况，即 list(data) 给出的结果与 list(data.columns) 相同。