我有嵌套列表a = [(2,0),(3,0),(4,2),(10,3),(11,5)]
我要做的是在n位置添加内部元组(0,n),其中{}是{}中缺失元素的位置。每个内部列表中的第二个元素应该以1为增量递增,如果存在间隙,则应该在该间隙处插入(0,n)。在
所以列表a的预期结果是:
^{pr2}$
也就是说,由于a中的第一个和第二个元素是(3,0)和{},所以在它们之间插入了soa(0,1)。在
我的解决方案是可行的,但我想知道是否有一种更像Python的方法来解决这个问题?我一直在查找Python的itertools库,但找不到一个简明的解决方案。在
到目前为止,我的代码是:l1 = [n[1] for n in a]
l2 = range(max(l1)+1)
l3 = [n for n in l2 if not in l1]
zeros = [0]*len(l3)
inserts = zip(zeros,l3)
a_full = a + inserts
a_out = sorted(a_full, key = itemgetter(1))
有人能提出更好的解决办法吗??在
编辑:
一般来说,可能有许多元素具有相同的第二个内部元素(例如(2,0)和{}出现在a中)。然而,我可以将这些组合起来,而不失概括性。在
嵌套列表a可以表示为a_sum = [(5,0),(4,2),(10,3),(11,5)]
通过使用代码a_group = [sum([x for x, y in group]) for key, group in groupby(a, key=itemgetter(1))]
a_sum = zip(output,list(set(l1)))
编辑二:
a的长度总是600,但根据研究的进展,这可能会增加到10**3阶。在