python - 从 Pandas 系列中，创建以唯一元素作为键、其索引作为值的字典

Question

目标是从 pandas 列(系列)创建一个字典，其中键是列的唯一元素，值是元素出现的行索引。我目前有完成此任务的代码，但我想知道是否有一种更简单且不那么麻烦的方法来做到这一点:

df = pd.DataFrame(np.random.randint(0,100,size=(1000, 4)), columns=list('ABCD'))
idx = df['A'].reset_index().groupby('A')['index'].apply(tuple).to_dict()

Answer 1

这是 GroupBy 对象的 groups 属性。它返回一个字典，该字典具有唯一值作为原始 DataFrame 的键和 Index 对象。

df.groupby('A').groups

<小时/>

{0: Int64Index([61, 466, 505, 619, 697, 811, 872], dtype='int64'),
 1: Int64Index([125, 254, 278, 330, 390, 396, 670, 732, 748, 849, 871, 880, 882,
                908, 943], dtype='int64'),
 2: Int64Index([77, 283, 401, 543, 544, 693, 816], dtype='int64'),
 ...}

或者如果您确实需要元组:

{k: tuple(v) for k,v in df.groupby('A').groups.items()}