Question

我认为我很难解释，但我将展示我的预期结果。我有以下数据框。

a = pd.DataFrame({"Character":["a","a","a","a","a","a","b","b","b","b","b","b"],
"value": [1,2,3,4,5,6,7,8,9,10,11,12]})
data.groupby("Sentence")

我可以根据句子对它进行分组，但是我不知道如何通过引用groupby来基于列将其提取。

预期结果：

[[1,2,3,4,5,6],[7,8,9,10,11,12]]

Answer 1

将GroupBy.apply与list一起用于Series，并为列表列表添加tolist：

data = pd.DataFrame({"Character":["a","a","a","a","a","a","b","b","b","b","b","b"],
"value": [1,2,3,4,5,6,7,8,9,10,11,12]})
L = data.groupby("Character")['value'].apply(list).tolist()

print (L)
[[1, 2, 3, 4, 5, 6], [7, 8, 9, 10, 11, 12]]

Answer 2

下面的代码给出了预期的输出

list(a.groupby("Character")["value"].apply(list))

使用groupby对A列进行分组，然后基于B列创建一个列表

2 个答案: