提取数据框的索引,按列值分组

时间:2018-04-22 00:56:31

标签: python pandas dataframe

我有一个正确编入索引的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]}

1 个答案:

答案 0 :(得分:2)

首先过滤你的df,我们使用groupbyto_dict

df[df.delta==1].reset_index().groupby('y').index.apply(list).to_dict()
Out[859]: {9: [5], 10: [1, 2, 3]}