从有限制的列表中使用随机选择时解决 ValueError

Question

给定一个随机列表和一个限制列表：

>>> import numpy as np
>>> x = np.random.randint(0, 50, 10)
>>> x
array([27, 14, 42,  1,  9, 43, 16, 39, 27,  3])
>>> y = [1,2,5, 19, 27]
>>> n = 5

我想从 X 中采样（不放回）N 而没有 Y 中的值，我可以这样做：

>>> np.random.choice(list(set(x).difference(y)), n, replace=False)
array([39,  9, 43, 14, 16])

N 是一个肯定小于 len(x) 的用户输入，但考虑到我不知道 N 是否大于 X-Y 的子集，我可能会遇到抛出 ValueError:

>>> np.random.choice(list(set(x).difference(y)), 8, replace=False)
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
  File "mtrand.pyx", line 1150, in mtrand.RandomState.choice (numpy/random/mtrand/mtrand.c:18113)
ValueError: Cannot take a larger sample than population when 'replace=False'

鉴于我必须为 N 预设一个最大值，例如：

>>> n = min(n, len(list(set(x).difference(y))) )
>>> n
7

但在那种情况下，N 不再是 8，用户输入：

>>> np.random.choice(list(set(x).difference(y)), n, replace=False)
array([14, 39, 43,  3, 42,  9, 16])

所以我必须post-添加输出：

>>> list(np.random.choice(list(set(x).difference(y)), _n, replace=False)) + [-1]*(n-_n)
[43, 42, 9, 16, 3, 39, 14, -1]

总而言之，我要抽样N个。没有从不在 Y 中的 X 值的子集中替换的元素，如果子集的长度小于 N，我需要用 -1 填充 "gaps"。

我可以用上面的代码来完成，但是有没有更简洁（希望也更有效）的方法来实现相同的输出？

Answer 1

我大概会用np.in1d取差，然后np.append进行post处理：

x = np.random.randint(0, 50, 10)
y = [1, 2, 5, 19, 27]
n = 12    

x_y = x[~np.in1d(x,y)]                                                    
arr = np.append(np.random.choice(x_y, len(x_y), replace=False), [-1]*(n-len(x_y)))
print arr
# array([35, 46, 39, 21,  9, 37, 17, 23,  8, -1, -1, -1])

如果 n 小于差异的长度，则不会追加任何内容。

从有限制的列表中使用随机选择时解决 ValueError

Resolve ValueError when using random choice from a list with restrictions

python

random

numpy

valueerror