我正在尝试将一个整数列表拆分成相等整数的索引的子列表。因此,假设我有一个列表:
original_list = [1,2,1,4,4,4,3,4,4,1,4,3,3]期望的输出将是:
indexes : [[0,2,9], [1], [6,11,12], [3,4,5,7,8,10]]
# corresponds to sublists: [[1,1,1] [2], [3,3,3], [4,4,4,4,4,4]]我不知道该怎么做,因为大多数解决方案都要求您首先对原始列表进行排序,但在我的例子中,这会弄乱索引。Itertools或np.arrays由于这个原因对我没有帮助,因为它们只对顺序相等的元素进行分组。
有没有人知道这个问题的解决方案?我很想听你说!
发布于 2018-12-04 23:37:16
您可以将collections.defaultdict用于一次性解决方案。然后,如果需要的话,可以使用sorted来按值对索引进行排序。
original_list = [1,2,1,4,4,4,3,4,4,1,4,3,3]
from collections import defaultdict
from operator import itemgetter
dd = defaultdict(list)
for idx, value in enumerate(original_list):
dd[value].append(idx)
keys, values = zip(*sorted(dd.items(), key=itemgetter(0)))
print(keys, values, sep='\n')
(1, 2, 3, 4)
([0, 2, 9], [1], [6, 11, 12], [3, 4, 5, 7, 8, 10])为了便于比较,dd的值在Python 3.6+中是按插入顺序排列的(在3.7+中是正式的,在3.6中作为CPython实现的详细信息):
print(list(dd.values()))
[[0, 2, 9], [1], [3, 4, 5, 7, 8, 10], [6, 11, 12]]https://stackoverflow.com/questions/53616179
复制相似问题