python - Numpy:生成 2 个无重叠的 n 个随机数序列

Question

对于 n = 5，[0, 2] 中的两个随机整数序列可能如下所示:

 l1 = [2, 0, 2, 2, 0]
 l2 = [0, 1, 2, 2, 2]

我想确保所有 i 的 l1[i] != l2[i]。我当前的方法(遍历所有值并替换重叠部分)看起来很笨拙:

n = 5
sample = np.arange(3)
l1 = np.random.choice(sample, size=n)
l2 = np.random.choice(sample, size=n)
mask = np.ones(sample.size, dtype=bool)
for i in xrange(n):
    if l1[i] == l2[i]:
        mask[l1[i]] = False
        l2[i] = np.random.choice(sample[mask])
        mask[l1[i]] = True

有更好的方法吗？

Answer 1

一种方法是计算 l1，然后计算每个元素的一组距离或偏移量，并确保偏移量不能为 0:

ofs = l2 = np.random.choice(sample[:-1], size=n)+1

然后您可以将这些偏移量添加到 l1 并将结果换回到范围内:

l2 = (l1 + ofs) % (sample[-1] + 1)