python - 从两列随机生成唯一组合

Question

假设我在一个大文件中有两列

pro1 lig1
pro2 lig2
pro3 lig3
pro4 lig1
.....

其次是列冗余。我想要新的双倍大小的随机组合，它不应该匹配给定的组合，例如

pro1 lig2
pro1 lig4
pro2 lig1
pro2 lig3
pro3 lig4
pro3 lig2
pro4 lig2
pro4 lig3
.....

谢谢。

Answer 1

如果您希望第一列中的每个值恰好有两个结果，我会强制使用不匹配的部分，像这样:

import random

def gen_random_data(inputfile):
    with open(inputfile, "r") as f:
        column_a, column_b = zip(*(line.strip().split() for line in f))

    for a, b in zip(column_a, column_b):
        r = random.sample(column_b, 2)
        while b in r: # resample if we hit a duplicate of the original pair
            r = random.sample(column_b, 2)

        yield a, r[0]
        yield a, r[1]