python - 根据条件从数据框中随机删除行

Question

给定一个在特定列中包含数值的数据框，我想随机删除特定列中的值位于特定范围内的特定百分比的行。

例如给定以下数据框:

df = pd.DataFrame({'col1': [1,2,3,4,5,6,7,8,9,10]})
df
   col1
0     1
1     2
2     3
3     4
4     5
5     6
6     7
7     8
8     9
9    10

应随机删除 col1 低于 6 的行的 2/5。

最简洁的方法是什么？

Answer 1

使用sample + drop

df.drop(df.query('col1 < 6').sample(frac=.4).index)

   col1
1     2
3     4
4     5
5     6
6     7
7     8
8     9
9    10

对于一个范围

df.drop(df.query('2 < col1 < 8').sample(frac=.4).index)

   col1
0     1
1     2
3     4
4     5
5     6
7     8
8     9
9    10