Numpy：生成 2 个无重叠的 n 个随机数序列

Question

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

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

我想确保 l1[i] != l2[i] 对所有 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)

Numpy：生成 2 个无重叠的 n 个随机数序列

Numpy: Generate 2 sequences of n random numbers with no overlaps

python

random

numpy

random-sample