Python如何恢复列表重排的模式

Question

因此，我正在根据索引模式重新排列列表，并希望找到一种方法来计算模式，以便将列表恢复到原始顺序。

对于我的例子，我使用了一个由5个项目组成的列表，因为我可以计算出将列表恢复到其原始状态所需的模式。

然而，当处理100个列表项时，这就不那么容易了。

def rearrange(pattern: list, L: list):
new_list = []
for i in pattern:
new_list.append(L[i-1])
return new_list
print(rearrange([2,5,1,3,4], ['q','t','g','x','r']))
#['t', 'r', 'q', 'g', 'x']

为了让它回到原来的模式我会使用

print(rearrange([3,1,4,5,2],['t', 'r', 'q', 'g', 'x']))
#['q', 't', 'g', 'x', 'r']

我正在寻找的是一种计算模式"；[3，1，4，5，2]"；关于上述示例。whist运行脚本，这样我就可以将列表设置回原来的顺序。

使用一个更大的例子：

print(rearrange([18,20,10,11,13,1,9,12,16,6,15,5,3,7,17,2,19,8,14,4],['e','p','b','i','s','r','q','h','m','f','c','g','d','k','l','t','a','n','j','o']))
#['n', 'o', 'f', 'c', 'd', 'e', 'm', 'g', 't', 'r', 'l', 's', 'b', 'q', 'a', 'p', 'j', 'h', 'k', 'i']

但我需要知道这个新列表要使用的模式，以便将其返回到原始状态。

print(rearrange([???],['n', 'o', 'f', 'c', 'd', 'e', 'm', 'g', 't', 'r', 'l', 's', 'b', 'q', 'a', 'p', 'j', 'h', 'k', 'i']))
#['e','p','b','i','s','r','q','h','m','f','c','g','d','k','l','t','a','n','j','o']

Answer 1

这通常被称为"argsort"；。但是，由于您使用的是基于1的索引，因此您偏离了1。你可以用numpy:获得它

>>> pattern
[2, 5, 1, 3, 4]
>>> import numpy as np
>>> np.argsort(pattern) + 1
array([3, 1, 4, 5, 2])

无numpy：

>>> [1 + i for i in sorted(range(len(pattern)), key=pattern.__getitem__)]
[3, 1, 4, 5, 2]

Answer 2

下面这样的东西怎么样：

def revert_pattern(pattern):
pattern_i = [0]*len(pattern)
for k in range(len(pattern)):
pattern_i[pattern[k]-1] = k+1 
return pattern_i
print(revert_pattern([2, 5, 1, 3, 4]))
# [3, 1, 4, 5, 2]

注意：我遵循了你的逻辑，但我建议你使用0作为最小索引，而不是1，因为它需要一些额外的+1/-1，可以避免

Answer 3

def rearrange(p, l):
arr = [l[i - 1] for i in p]
d = {v : i + 1 for i, v in enumerate(arr)}
order = [d[k] for k in l]
return arr, order
a = [2, 5, 1, 3, 4]
b = ['q', 't', 'g', 'x', 'r']
rearrange(a, b)
# (['t', 'r', 'q', 'g', 'x'], [3, 1, 4, 5, 2])

或者

def revert(p):
z = zip(p, list(range(len(p))))
return [x + 1 for _, x in sorted(z)]
a = [2, 5, 1, 3, 4]   
revert(a)
# [3, 1, 4, 5, 2]