python：关于其他属性的洗牌列表

Question

我有两个列表，我想根据另一个列表中的属性对其中的值进行洗牌。例如：

list1 = np.array([1,1,1, 2,2,2, 3,3,3])  # spaces for better understanding
list2 = np.array([1,2,3, 4,5,6, 7,8,9])
result = [4,5,6, 1,2,3, 7,8,9]

我解决了这个问题

y = split(list2, len(np.unique(list1)))
np.random.shuffle(y)
result = np.array(y).flatten()

我希望它也适用于 list1 中的属性不在一起的情况。示例：

list1 = np.array([1,2,3,1,2,3,1,2,3])
list2 = np.array([1,2,3,4,5,6,7,8,9])
result = [2,1,3,5,4,6,8,7,9]

Answer 1

解决了：

uniques = np.unique(list1)
shuffled = uniques.copy()
np.random.shuffle(shuffled)

result = list2.copy()
for orig, new in zip(uniques, shuffled):
    result[np.where(list1==orig)] = list2[np.where(list1==new)]