Серия фильтров pandas со списками строк в качестве значений

Итак, я пытаюсь сделать простой фильтр, который будет принимать фрейм данных и отфильтровывать все строки, которые не имеют целевого жанра. С кодом будет проще объяснить:

    import pandas as pd

test = [{
        "genre":["RPG","Shooter"]},
        {"genre":["RPG"]},
        {"genre":["Shooter"]}]
        
data =pd.DataFrame(test)

fil = data.genre.isin(['RPG'])

Я хочу, чтобы фильтр возвращал фрейм данных со следующими элементами:

[{"genre":["RPG"]},
{"genre":["RPG", "Shooter"]}]

Это ошибка, которую я получаю, когда пытаюсь использовать свой код:

SystemError: <built-in method view of numpy.ndarray object at 0x00000180D1DF2760> returned a result with an error set

@Vaishali Верно, что вопросы дублируются, но я считаю, что ответы здесь лучше, чем те, что приведены в другом.

— 11.12.2020 22:35

python pandas

11.12.2020 22:17

Почему в Python есть оператор "pass"?

Оператор pass в Python - это простая концепция, которую могут быстро освоить даже новички без опыта программирования.

Некоторые методы, о которых вы не знали, что они существуют в Python

Python - самый известный и самый простой в изучении язык в наши дни. Имея широкий спектр применения в области машинного обучения, Data Science,...

Основы Python Часть I

Вы когда-нибудь задумывались, почему в программах на Python вы видите приведенный ниже код?

LeetCode - 1579. Удаление максимального числа ребер для сохранения полной проходимости графа

Алиса и Боб имеют неориентированный граф из n узлов и трех типов ребер:

Оптимизация кода с помощью тернарного оператора Python

И последнее, что мы хотели бы показать вам, прежде чем двигаться дальше, это

Советы по эффективной веб-разработке с помощью Python

Как веб-разработчик, Python может стать мощным инструментом для создания эффективных и масштабируемых веб-приложений.

Перейти к ответу Данный вопрос помечен как решенный

Ответы 2

Ответ принят как подходящий

Проблема в том, что элементами жанра являются списки, поэтому isin не работает. Использовать:

mask = data['genre'].apply(frozenset(['RPG']).issubset)
print(data[mask])

Выход

            genre
0  [RPG, Shooter]
1           [RPG]

Выражение:

frozenset(['RPG']).issubset

Проверяет, что в каждой строке содержится какой-либо список из документации:

Проверьте, находится ли каждый элемент в наборе в другом.

Таким образом, вы также можете легко проверить несколько значений, например:

mask = data['genre'].apply(frozenset(['RPG', "Shooter"]).issubset)
print(data[mask])

Выход

            genre
0  [RPG, Shooter]

11.12.2020 22:19

Вы хотите:

data[data.genre.apply(lambda x: 'RPG' in x)]

Или:

data[data.genre.explode().eq('RPG').any(level=0)]

Выход:

            genre
0  [RPG, Shooter]
1           [RPG]

11.12.2020 22:19

Другие вопросы по теме

Как определить зоны в таблице с помощью панд?

Необходимо подсчитывать повторяющиеся последовательные значения в фрейме данных python в группе

Невозможно удалить пустые строки в фрейме данных

Используйте Pandas для объединения двух DataFrames

Группировать по любому общему значению в нескольких столбцах

Разделить элементы списка на подсписки на основе символа в элементе списка

Совокупная сумма в Python без текущей строки

Заполнить столбец в фрейме данных panda результатом groupby

Скопировать вывод из команд описания и информации в разные кадры данных python

Как искать в DataFrame определенную строку с помощью подстановочного знака

Серия фильтров pandas со списками строк в качестве значений

Ответы 2

Другие вопросы по теме

Похожие вопросы