python - 我如何通过特定的垃圾箱过滤数据框的行

Question

假设我有一个数据框:

col1  col2
15    2
17    4
14    5
20    10
21    11
19    11
18    12
20    6
22    7
21    8
22    8
25    6
21    7

首先，我对 col2 使用 pandas.value_counts() ，并使用三个容器 (0,5] (5,10] (10,15 ].

然后，我想过滤 col2 值属于频率最高的前两个 bin 的行，在本例中为 bin (5,10] 和(10,15]。

Answer 1

用途:

s = df.col2.value_counts(bins=[0,5,10,15]).nlargest(2)
df[pd.cut(df.col2, s.index).notnull()]

输出:

    col1  col2
3     20    10
4     21    11
5     19    11
6     18    12
7     20     6
8     22     7
9     21     8
10    22     8
11    25     6
12    21     7