python - 您如何从满足某些条件的数组中随机排除/删除元素？

Question

例如，如果我有一个 1x100 数组，它包含 90 个 0，我想以某种方式排除随机选择的 80 个 0。我已经为这个问题苦苦挣扎了一段时间，不幸的是我取得的进展很小。

Answer 1

因为你有一个 numpy 标签:

import numpy as np

def solution(arr, value, size):
    return np.delete(arr, np.random.choice(np.flatnonzero(arr==value), size, False))

arr = np.array([0]*90 + [1]*10)
np.random.shuffle(arr)

print(solution(arr, 0, 80)) # [1 0 1 0 0 1 0 0 1 0 1 1 1 0 0 0 0 1 1 1]
print(solution(arr, 0, 90)) # [1 1 1 1 1 1 1 1 1 1]
print(solution(arr, 0, 100)) # Raises error (expected behavior)