Как отфильтровать строки в кадре данных Pandas на основе столбца со списком в качестве значений?

Question

У меня есть фрейм данных Pandas, где в одном из столбцов хранятся списки строковых литералов.Например, мои данные выглядят следующим образом:

field1, field2
x1, ['tag1', 'tag2']
x2, ['tag1', 'tag3']

Я хочу получить только записи в этом кадре данных, чей список field2 содержит tag1 или tag2.Итак, в приведенном выше примере должна быть возвращена только запись, в которой field1=x1.

Как я могу выполнить эту фильтрацию на фрейме данных Pandas?

Answer 1

Вам необходимо проверить на tag2 в списке field2.Используя .apply () , вы можете добиться этого.

df[df.apply(lambda x: 'tag2' in x['field2'],axis=1)]

Выход:

    field1  field2
0   x1      [tag1, tag2]