Python 随机更改numpy数组值

我有一个数组数据

import numpy as np

data = np.array([[1,0,4,1,1,2,0],
                 [1,0,0,4,0,1,0],
                 [4,0,4,3,2,1,0],
                 [4,0,1,1,2,1,1]])

print(data)

如果一些非零元素（1,2,3,4）的计数超过5，我想随机抽取并保留5个位置，并将所有其他位置替换为0

uniques, counts = np.unique(data, return_counts=True)
for unique, count in zip(uniques, counts):
    print (unique, count)

    if count > 5:       
       
       ids = np.random.choice(range(count), 5, replace=False)

---

我该怎么做？

当计数超过5时，只保留5个位置或唯一非零元素的所有位置？如果计数超过5，则只保留5个位置。这行的替代方法是什么，

data[tuple（I[id]表示np中的I.where（data==unique））]=0

？我无法理解该行可能这行更容易理解，

x，y=np.where（data==unique）

对于ID中的I:data[x[I]，y[I]=0

uniques, counts = np.unique(data, return_counts=True)
for unique, count in zip(uniques, counts):
    print (unique, count)
    if unique != 0 and count > 5:
       ids = np.random.choice(count, count-5, replace=False)
       data[tuple(i[ids] for i in np.where(data == unique))] = 0