我有一份我想重新订购的列表清单:
qvalues = [[0.1, 0.3, 0.6],[0.7, 0.1, 0.2],[0.3, 0.4, 0.3],[0.1, 0.3, 0.6],[0.1, 0.3, 0.6],[0.1, 0.3, 0.6]]
如果我有一个包含我想要的订单的列表,我知道如何重新排序此列表(例如here)。这个棘手的部分是获得这个订单。
我拥有的是:
locations = [(['Loc1','Loc1'], 3), (['Loc2'], 1), (['Loc3', 'Loc3', 'Loc3'], 2)]
这是一个元组列表,其中每个元组的第一个元素是一个带有位置名称的列表,为该位置中的每个个体重复,第二个元素是这些个体在{{1}上的顺序。 } list(qvalues
为qvalues[0]
,'Loc2'
为qvalues[1:4]
,'Loc3'
为qvalues[4:6]
。
我想要的是将'Loc1'
中的列表顺序更改为它们在qvalues
中显示的顺序:首先locations
,然后'Loc1'
,最后{{} 1}}。
这只是一个小例子,我的真实数据集有数百个人和17个地点。
提前感谢您提供的任何帮助。
答案 0 :(得分:1)
您需要建立一个偏移和长度列表,而不是locations
列表中提供的长度和位置。然后,您将能够根据您链接的答案重新排序:
qvalues = [[0.1, 0.3, 0.6],[0.7, 0.1, 0.2],[0.3, 0.4, 0.3],[0.1, 0.3, 0.6],[0.1, 0.3, 0.6],[0.1, 0.3, 0.6]]
locations = [(['Loc1','Loc1'], 3), (['Loc2'], 1), (['Loc3', 'Loc3', 'Loc3'], 2)]
locations_dict = {pos:(index,len(loc)) for index,(loc,pos) in enumerate(locations)}
# if python2: locations_dict = dict([(pos,(index,len(loc))) for index,(loc,pos) in enumerate(locations)])
offsets = [None]*len(locations)
def compute_offset(pos):
# compute new offset from offset and length of previous position. End of recursion at position 1: we’re at the beginning of the list
offset = sum(compute_offset(pos-1)) if pos > 1 else 0
# get index at where to store current offset + length of current location
index, length = locations_dict[pos]
offsets[index] = (offset, length)
return offsets[index]
compute_offset(len(locations))
qvalues = [qvalues[offset:offset+length] for offset,length in offsets]
您最终将qvalues
作为列表列表而不是“简单”列表列表。如果你想将它展平以保持初始布局,请使用此列表理解:
qvalues = [value for offset,length in offsets for value in qvalues[offset:offset+length]]
使用第一个版本输出
[[[0.1, 0.3, 0.6], [0.1, 0.3, 0.6]], [[0.1, 0.3, 0.6]], [[0.7, 0.1, 0.2], [0.3, 0.4, 0.3], [0.1, 0.3, 0.6]]]
以第二个版本输出
[[0.1, 0.3, 0.6], [0.1, 0.3, 0.6], [0.1, 0.3, 0.6], [0.7, 0.1, 0.2], [0.3, 0.4, 0.3], [0.1, 0.3, 0.6]]