获取列表中重复元素的索引范围
获取列表中重复元素的索引范围
提问于 2018-04-19 08:02:45
假设我在python中有一个list a = [-1,-1,-1,1,1,1,2,2,2,-1,-1,-1,1,1,1]，我想要的是，如果python中有内置的函数，我们在其中传递一个列表，它将返回哪一个元素在什么索引范围内，例如
>>> index_range(a)
{-1 :'0-2,9-11', 1:'3-5,12-14', 2:'6-8'}我尝试从Counter库中使用collection.Counter函数，但它只输出元素的计数。
如果没有任何内置的函数，请指导我如何在我自己的函数中实现这一点，而不是整个代码，只是一个指南。
回答 2
Stack Overflow用户
回答已采纳
发布于 2018-04-19 08:23:13
您可以使用itertools.groupby和collections.defaultdict创建自定义函数，以获得列表形式的数字范围，如下所示：
from itertools import groupby
from collections import defaultdict
def index_range(my_list):
my_dict = defaultdict(list)
for i, j in groupby(enumerate(my_list), key=lambda x: x[1]):
index_range, numlist = list(zip(*j))
my_dict[numlist[0]].append((index_range[0], index_range[-1]))
return my_dict样本运行：
>>> index_range([-1,-1,-1,1,1,1,2,2,2,-1,-1,-1,1,1,1])
{1: [(3, 5), (12, 14)], 2: [(6, 8)], -1: [(0, 2), (9, 11)]}为了使在dict中以字符串的形式获得值，您可以修改上面的函数，或者在字典理解中使用函数的返回值如下：
>>> result_dict = index_range([-1,-1,-1,1,1,1,2,2,2,-1,-1,-1,1,1,1])
>>> {k: ','.join('{}:{}'.format(*i) for i in v)for k, v in result_dict.items()}
{1: '3:5,12:14', 2: '6:8', -1: '0:2,9:11'}Stack Overflow用户
发布于 2018-04-19 08:11:54
您可以使用一个dict，它将列表项用作键，将其索引用作值：
>>> lst = [-1,-1,-1,1,1,1,2,2,2,-1,-1,-1,1,1,1]
>>> indexes = {}
>>> for index, item in enumerate(lst):
... indexes.setdefault(value, []).append(index)
>>> indexes
{1: [3, 4, 5, 12, 13, 14], 2: [6, 7, 8], -1: [0, 1, 2, 9, 10, 11]}如果需要的话，可以将索引列表合并到范围中。必要的话我也可以帮你。
页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接：
https://stackoverflow.com/questions/49915793
复制相关文章
相似问题
腾讯云开发者
