выбрать только указанное количество строк от первой до последней или случайным образом для заданного значения столбца

Question

я пытаюсь выбрать случайные строки для определенного значения столбца

df:

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

Я хочу извлечь только две строки случайным образом, где df.B == 'x'

вывод:

user    A
1   x
3   x

код:

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

Answer 1

Если хотите случайные 2 строки, используйте DataFrame.sample после фильтрации boolean indexing:

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

Если хотите первые 2 строки используйте DataFrame.head:

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