python - 从 numpy 数组中选择 'some' 个随机点

Question

我有两个相关的 numpy 数组，X 和 y。我需要从 X 中选择 n 随机行并将其存储在一个数组中，相应的 y 值和附加到它的索引随机选择的点。

我有另一个数组index，它存储了一个我不想采样的索引列表。

我该怎么做？

示例数据:

index = [2,3]
X = np.array([[0.3,0.7],[0.5,0.5] ,[0.2,0.8], [0.1,0.9]])
y = np.array([[0], [1], [0], [1]])

如果这些 X 是随机选择的(其中 n=2):

randomylSelected = np.array([[0.3,0.7],[0.5,0.5]])

期望的输出是:

index = [0,1,2,3]
randomlySelectedY = [0,1]

我该怎么做？

Answer 1

我会管理一个 bool 值数组，我不断使用它来切片索引数组并从结果中随机选择。

n = X.shape[0]
sampled = np.empty(n, dtype=np.bool)
sampled.fill(False)
rng = np.arange(n)

k = 2

while not sampled.all():
    sample = np.random.choice(rng[~sampled], size=k, replace=False)
    print(X[sample])
    print()
    print(y[sample])
    print()
    sampled[sample] = True

[[ 0.2  0.8]
 [ 0.5  0.5]]

[[0]
 [1]]

[[ 0.3  0.7]
 [ 0.1  0.9]]

[[0]
 [1]]