对于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
有更好的方法吗?
答案 0 :(得分:2)
一种方法是在你正在做的时候计算homestead halt && homestead up --provision
,然后为每个元素计算一组距离或偏移量,并确保偏移量不能为0:
l1
然后,您可以将这些偏移添加到ofs = l2 = np.random.choice(sample[:-1], size=n)+1
并将结果包装回范围:
l1