第一次海报在这里。如果我有一个包含具有不同信息的子列表的列表,并且我想将具有相同键值的所有子列表分组到一个新的子列表中(例如,根据键值将列表唯一化),那么最有效的方法是什么?在
示例:
A = [[1,2,3], [1,3,7], [2,1,3], [8,9,6], [3,7,9], [2,3,8], [1,2,4]]
我想将“0”索引中具有相同项的所有子列表分组到一个新的子列表中,最好保持原来的顺序:
^{pr2}$这个问题我已经解决过很多次了,但是我想看看有没有更有效的方法来解决这个问题。是的,出于这个原因,我可以使用numpy包,但是我对list解决方案感兴趣。在
通常,我会首先唯一化每个子列表中第一个索引中的项(如果不担心效率或顺序,则使用set方法),然后循环整个列表,将子列表组合成一个新的子列表:
def getUniqueList(list):
seen = []
for e in list:
if e not in seen:
seen.append(e)
return seen
def uniquify(list):
# Get all items from the first column
new_list = [item[0] for item in list]
return getUniqueList(new_list)
def rearrangeList(A, A_0):
B = []
for i in range(len(A_0)):
B.append([])
for i in range(len(A)):
index = A_0.index(A[i][0])
B[index].append(A[i])
return B
A = [[1,2,3], [1,3,7], [2,1,3], [8,9,6], [3,7,9], [2,3,8], [1,2,4]]
A_0 = uniquify(A) # Contains all unique keys in index 0 of list A
B = rearrangeList(A, A_0)
print(B)
这个应该(我没有测试这个,但是我将如何编码它)结果是:
B = [ [[1,2,3], [1,3,7], [1,2,4]], [[2,1,3],[2,3,8]], [[8,9,6]], [[3,7,9]] ]
我想知道,1)在效率方面,是否有更好的方法来实现这一点,2)是否有一种方法可以以更少的步骤完成这项工作(不一定高效,只是好奇)。希望这个问题提出得很好,但如果需要编辑请告诉我!在
保持原始列表顺序的要求很有趣。在
假设您的列表是
A
,并且您希望结果是B
,那么下面的代码在使用输入时构造列表,因此保留列表的原始顺序:或者以下更简单的代码在defaultdict中创建列表,然后复制,但原始列表的顺序将丢失:
^{pr2}$请注意,子列表将按照它们最初出现的顺序排列。在
实际上,您需要一个分组操作。使用字典按键分组,使用OrderedDict保持顺序:
老派(但又好又简单)的方法:
相关问题 更多 >
编程相关推荐