Pandas выбор кадра данных на основе содержимого объекта списка в кадре данных

Question

У меня есть pandas фрейм данных, где один столбец содержит объекты списка, что дает мне dtype: object. Объекты списка имеют разную длину.

        features           other_features
0        ["foo", "bar"]                2
1        ["foo", "dog"]                1
2        ["dog"]                       4

Я хочу выбрать строки в моем фрейме данных, где список содержит элементы, которые находятся в другом списке external_list = ["dog", "cat"].

В этом случае я хотел бы, чтобы результат был строки, которые содержат собаку или кошку вместе с другими coulmns, которые были бы:

        features           other_features
1        ["foo", "dog"]                1
2        ["dog"]                       4

Я пробовал isin, но из того, что я понимаю, требует, чтобы тип столбца элемента, чтобы смотреть, чтобы не быть список из более чем одного объекта.

filter = df["features"].isin(["dog", "cat"])

Это приводит к тому, что каждый элемент имеет значение False, чего не должно быть, поскольку некоторые должны иметь значение True, и он не возвращает остальные столбцы .

Как я могу решить эту проблему?

Answer 1

Использовать map при сравнении списка, преобразованного в наборы, проверенные isdisjoint, ~ для обратной маски:

#if string repr of list
#import ast
#df['features'] = df['features'].apply(ast.literal_eval)

external_list = ["dog", "cat"]
df = df[~df.features.map(set(external_list).isdisjoint)]
print (df)
     features  other_features
1  [foo, dog]               1
2       [dog]               4

Answer 2

Давайте сделаем «взрыв» на оси = 0, тогда мы можем сделать isin

m=pd.DataFrame(df.features.tolist(),index=df.index).isin(external_list).any(1)
df=df[~m]