在列上有一定条件后，如何在python中的groupby函数之后创建列表？

Question

我有一个如下所示的数据框：

Number  Time Response 
0       3    6
1       5    0
2       33   4
3       15   6
4       12   2

我正在根据响应进行分组，并获得时间列表

df.groupby('Response')['Time'].apply(list)

我不知道如何设置仅填充响应值“ 6”的“时间”的条件。

np.where无效。

预期输出：

The list of time for response 6: [3,15]

Answer 1

只需参考列表中的特定分组ID：

>>> df.groupby("Response")["Time"].apply(list)[6]
[3, 15]