python - 反转 numpy 数组中键的随机选择

Question

我有一个名为 arr 的巨大 np.array，它有 N 个值，并通过以下方式随机选择这些值的 10%:

choice=random.sample(range(N), int(N*percent))  # percent has values 0-1
newarr=arr[choice]

N 可能超过 200 万个值。

实际上我还需要一个包含其他 90% 值的数组。所以目前我使用的是非常慢的以下内容:

def buildRevChoice(choice, nevents):
        revChoice=[]
        for i in range(N):
            if not i in choice:
                revChoice.append(i)
        return revChoice

你能想出一种方法来固定它吗？

Answer 1

你可以 random.shuffle列表，然后根据需要拆分它。

def choice(N, percent):
    tmp = range(N)
    random.shuffle(tmp)
    cut = int(N * percent)
    return tmp[:cut], tmp[cut:]

您将获得两个列表，第一个包含所选列表，第二个包含其余列表。