Удаление строки при дублировании значения столбца, если другой столбец соответствует критерию
У меня есть DataFrame, в котором я хочу удалить строку (или строки), где несколько последовательных строк имеют одинаковое значение для столбца на основе значения другого столбца. В этом случае я хочу сохранить строку с самым высоким значением в B, если оно высокое, или с самым низким, если оно низкое. По сути, я пытаюсь, чтобы максимумы следовали за минимумами, а минимумы - максимумами.
df = pd.DataFrame({'A': ['low', 'high', 'high', 'low', 'low','low'],
'B': [10, 70, 90, 40, 50,60]})
Выход:
A B
0 low 10
1 high 70
2 high 90
3 low 40
4 low 50
5 low 60
Желанный:
A B
0 low 10
2 high 90
3 low 40
Пытаюсь понять, как реализовать логику, и наткнулся на кирпичную стену.
Почему в Python есть оператор "pass"?
Оператор pass в Python - это простая концепция, которую могут быстро освоить даже новички без опыта программирования.
Некоторые методы, о которых вы не знали, что они существуют в Python
Python - самый известный и самый простой в изучении язык в наши дни. Имея широкий спектр применения в области машинного обучения, Data Science,...
Основы Python Часть I
Вы когда-нибудь задумывались, почему в программах на Python вы видите приведенный ниже код?
LeetCode - 1579. Удаление максимального числа ребер для сохранения полной проходимости графа
Алиса и Боб имеют неориентированный граф из n узлов и трех типов ребер:
Оптимизация кода с помощью тернарного оператора Python
И последнее, что мы хотели бы показать вам, прежде чем двигаться дальше, это
Советы по эффективной веб-разработке с помощью Python
Как веб-разработчик, Python может стать мощным инструментом для создания эффективных и масштабируемых веб-приложений.
0
0
55
2
Перейти к ответу
Данный вопрос помечен как решенный
Ответы 2
Вот быстрый и грязный способ использования groupby.apply:
out = (df.groupby(['A', df['A'].ne(df['A'].shift()).cumsum()])
.apply(lambda x: x.max() if x['A'].iat[0]=='high' else x.min())
.droplevel(0).sort_index().reset_index(drop=True))
Другой способ — сначала найти groupby + max; затем mask «низкие» значения и замените их значениями groupby + min:
g = df.groupby(['A', df['A'].ne(df['A'].shift()).cumsum()], sort=False)['B']
out = g.max().mask(lambda x: x.index.get_level_values(0)=='low', g.min()).droplevel(1).reset_index()
Выход:
A B
0 low 10
1 high 90
2 low 40
Ответ принят как подходящий
Вот еще один способ:
d = {'low':-1}
(df.assign(B = df['B'].mul(df['A'].map(d),fill_value=1))
.groupby(['A',pd.Series(pd.factorize(df['A'])[0]).diff().ne(0).cumsum()]).max()
.abs()
.sort_index(level=1)
.droplevel(1)
.reset_index())
Выход:
A B
0 low 10.0
1 high 90.0
2 low 40.0
Другие вопросы по теме
Создание функции для объединения строк на основе len(массива)
Python 3 зацикливается с помощью if else elif
Кнопки, нажимаемые веб-драйвером selenium, не работают/не найдены
Json нормализует массив объектов
Генерация кадра данных из группового преобразования
Когда генерируется ошибка переполнения Python?
Самый простой способ создать подкласс Set таким образом, чтобы изменить методы Set?
Хранить только определенные слова в строке python
Удалите строки с одинаковым значением в столбцах Dataframe
Выход из цикла через метод класса
Похожие вопросы
Как динамически установить порядок параметров сортировки с нулевым последним в API django?
Проверьте, сколько раз число появляется внутри другого без строк или списков (Python)
Как я могу выбрать переменную внутри тега скрипта в качестве значения JSON?
Python — самый быстрый способ удалить конечные нули из битового представления числа
Переименуйте дубликаты в списке словарей, добавив прогрессивные числа в конце
Включение изображений в библиотеку с установкой pip
Matplotlib продолжает писать одну и ту же цифру. Мне нужны результаты регрессии в виде отдельных файлов png. Как мне изменить код?
Создание функции для объединения строк на основе len(массива)
Получите количество идентификаторов, которые имеют одинаковую комбинацию различных значений в столбце «местоположения»
Python 3 зацикливается с помощью if else elif