python - numpy 在数组中的重复值中选择固定数量的值

Question

从一个具有重复值的简单数组开始:

a = np.array([2,3,2,2,3,3,2,1])

我正在尝试从中选择最多 2 个唯一值。生成的数组将显示为:

b = np.array([2,3,2,3,1])

无论项目的顺序如何。到目前为止，我试图通过以下方式找到独特的值(value):

In [20]: c = np.unique(a,return_counts=True)

In [21]: c
Out[21]: (array([1, 2, 3]), array([1, 4, 3]))

这很有用，因为它也返回值的频率，但我坚持按频率过滤。

Answer 1

你可以使用 np.repeat从 uniques 和 count 的数组中生成所需的数组:

import numpy as np

a = np.array([2,3,2,2,3,3,2,1])
uniques, count = np.unique(a,return_counts=True)
np.repeat(uniques, np.clip(count, 0, 2))

产量

array([1, 2, 2, 3, 3])

np.clip 用于强制 count 中的所有值都在 0 和 2 之间。因此，对于每个唯一值，您最多可以获得两个值。