我如何检查另一个较大的阵列中存在哪一行？

Question

我如何检查另一个较大的阵列中存在哪一行？

给定以下设置：

final_batch = np.emtpy((batch_size,2))
batch_size = 4
a = np.array(range(10))
b = np.array(range(10,20))
edges = np.array([[0,11],[0,12],[1,11],[1,12],[0,17]])


c1 = np.random.choice(a,batch).reshape(-1,1)
c2 = np.random.choice(b,batch).reshape(-1,1)
samples = np.append(c1,c2,axis=1)

现在可以在样本和边缘中存在副词，我想继续制作np.random.choice，如果它们还不存在，则仅将它们添加到final_batch中。这样做的简单方法就是将它们1 x 1 x 1 x循环中，

while len(final_batch)<batch_size+1:
    c1 = np.random.choice(a,1).reshape(-1,1)
    c2 = np.random.choice(b,1).reshape(-1,1)
    if not np.isin(c1,c2).any():
        final_batch = np.append(final_batch,np.append(c1,c2,axis=1),axis=0)    

final_batch = final_batch[1:]

但所有a，b和edges都可以很大批处理大小将是10k，但是由于要立即采样许多元素的速度更快，因此我想看看是否有更快的方法。请

while len(final_batch)<batch_size+1:
     c1 = np.random.choice(a,batch).reshape(-1,1)
     c2 = np.random.choice(b,batch).reshape(-1,1)
     samples = np.append(c1,c2,axis=1)
     full_batch.append(samples NOT IN edges)
     

注意，C1和C2是互斥的，所以我觉得我应该能够以某种方式使用它。

Answer 1

如果我理解您的问题，您正在寻找类似的

samples = np.empty((10, 2), dtype=int)
samples[:,0] = np.random.choice(a, 10)
samples[:,1] = np.random.choice(b, 10)
new_indices = (samples != edges[:,None]).any(axis=2).all(axis=0)
new_samples = samples[new_indices]

意思我生成10个新样本，那么我会看看它们是否匹配边缘。这并不是最佳的操作号码，因为即使我找到了匹配项，我仍会继续检查平等，但这是用numpy矢量化的，numpy通常比尽快停止更快。