假设我有一个如下所示的列表:
[1, 2, 2, 5, 8, 3, 3, 9, 0, 1]
现在我想对相同元素的索引进行分组,所以结果应该如下所示:
[[0, 9], [1, 2], [3], [4], [5, 6], [7], [8]]
如何高效地执行此操作?我尽量避免使用循环,因此使用 numpy/pandas 函数的任何实现都很棒。
最佳答案
使用 pandas GroupBy.apply
,这非常简单——使用您的数据对一系列索引进行分组。这里的一个好处是您可以保持索引的顺序。
data = [1, 2, 2, 5, 8, 3, 3, 9, 0, 1]
pd.Series(range(len(data))).groupby(data, sort=False).apply(list).tolist()
# [[0, 9], [1, 2], [3], [4], [5, 6], [7], [8]]
关于python - 对列表中相同元素的索引进行分组的有效方法,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/56641420/