对于列的给定值，从第一行到最后一行或随机选择指定的行数

Question

am尝试为特定列值选择随机行

df:

user    A
1   x
2   y
3   x
4   x
5   x
6   y
7   y
9   y
8   y

我只想随机提取两行，其中df.B == 'x'

ouptut:

user    A
1   x
3   x

代码：

df.loc[1:2,[df['B'] == 'x]]

Answer 1

如果想要随机2行，请在boolean indexing:筛选后使用DataFrame.sample

df[df['B'] == 'x'].sample(2)

如果需要前2行，请使用DataFrame.head:

df[df['B'] == 'x'].head(2)