Фильтрация элементов списка на пандах

Question

У меня есть pandas dataframe foo, который выглядит следующим образом:

   a         b
0  1    [1, 2]
1  2    [1, 2]
2  3  [10, 11]

, и я пытаюсь извлечь все строки, содержащие элемент [1, 2] в столбце b.Есть ли хороший способ сделать это?

Answer 1

Стандартный способ сделать это с логическим индексированием :

mask = df['b'].apply(lambda x: x == [1, 2])
df[mask]

возвращает

   a       b
0  1  [1, 2]
1  2  [1, 2]

Если вы новичок в pandas, урок для новых пользователей - хорошее место для начала, в котором будут рассмотрены следующие вопросы.