我有一个正确编入索引的Pandas数据框:
delta y
0 0 10
1 1 10
2 1 10
3 1 10
4 0 9
5 1 9
我想提取以一列(其中delta = 1)为条件的索引,但是将它们分组为不同列(y)的值。 理想情况下,我想要一个以下形式的字典:
{10 : [ 1, 2, 3], 9 : [ 5]}
答案 0 :(得分:2)
首先过滤你的df,我们使用groupby
和to_dict
df[df.delta==1].reset_index().groupby('y').index.apply(list).to_dict()
Out[859]: {9: [5], 10: [1, 2, 3]}