如何随机选择一些pandas数据帧行？

我有一个包含列数的pandas数据帧df.对于许多行,金额为零.我想随机删除金额为零的50％的行,保留金额非零的所有行.我怎样才能做到这一点？

解决方法:

大熊猫

使用查询示例

df.drop(df.query('amount == 0').sample(frac=.5).index)

考虑数据帧df

df = pd.DataFrame(dict(amount=[0, 1] * 10))

df.drop(df.query('amount == 0').sample(frac=.5).index)

numpy的

iszero = df.amount.values == 0
count_zeros = iszero.sum()
idx = np.arange(iszero.shape[0])
keep_these = np.random.choice(idx[iszero], int(iszero.sum() * .5), replace=False)

df.iloc[np.sort(np.concatenate([idx[~iszero], keep_these]))]

    amount
1        1
2        0
3        1
5        1
6        0
7        1
8        0
9        1
10       0
11       1
12       0
13       1
15       1
17       1
19       1

时间测试

根据@tomcy的评论,您可以使用参数inplace = True从df中删除行而无需重新分配df

df.drop(df.query('amount == 0').sample(frac=.5).index, inplace=True)
df

    amount
1        1
2        0
3        1
5        1
6        0
7        1
8        0
9        1
10       0
11       1
12       0
13       1
15       1
17       1
19       1