python - 随机选择两个列表的某些索引创建两个新列表

Question

假设我有两个列表:

A = ['cat', 'dog', 'cow', 'pig', 'monkey']
B = ['Felix', 'Fido', 'Moo', 'Trump', 'King Kong']

现在我想通过选择 3 个随机索引(无替换)来创建 lists A 和 list B，同时保持 A< 的值之间的关系 和 B。

例如，

随机选择的索引:4, 0, 3

因此，

A = ['monkey', 'cat', 'pig']
B = ['King Kong', 'Felix', 'Trump']

有没有一种方法可以在不编写迭代 3 次以选择随机索引的 for 循环的情况下执行此操作？

Answer 1

您可以 zip列表，然后选择 3 个随机对 random.sample ，最后再次将这些对拆分为单独的列表:

import random

pairs = list(zip(A, B))  # make pairs out of the two lists
pairs = random.sample(pairs, 3)  # pick 3 random pairs
A1, B1 = zip(*pairs)  # separate the pairs

这是逐步发生的事情:

>>> list(zip(A, B))
[('cat', 'Felix'), ('dog', 'Fido'), ('cow', 'Moo'), ('pig', 'Trump'), ('monkey', 'King Kong')]
>>> random.sample(_, 3)
[('monkey', 'King Kong'), ('pig', 'Trump'), ('dog', 'Fido')]
>>> list(zip(*_))
[('monkey', 'pig', 'dog'), ('King Kong', 'Trump', 'Fido')]