Numpy:生成2个n个随机数的序列,没有重叠

时间:2016-03-26 00:22:56

标签: python numpy random random-sample

对于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

有更好的方法吗?

1 个答案:

答案 0 :(得分:2)

一种方法是在你正在做的时候计算homestead halt && homestead up --provision,然后为每个元素计算一组距离或偏移量,并确保偏移量不能为0:

l1

然后,您可以将这些偏移添加到ofs = l2 = np.random.choice(sample[:-1], size=n)+1 并将结果包装回范围:

l1