为python中的列表生成新订单

时间:2015-09-29 10:17:24

标签: python list sorting tuples

我有一份我想重新订购的列表清单:

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(qvaluesqvalues[0]'Loc2'qvalues[1:4]'Loc3'qvalues[4:6]

我想要的是将'Loc1'中的列表顺序更改为它们在qvalues中显示的顺序:首先locations,然后'Loc1',最后{{} 1}}。

这只是一个小例子,我的真实数据集有数百个人和17个地点。

提前感谢您提供的任何帮助。

1 个答案:

答案 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]]