Получите среднее общее количество, сгруппированное по нескольким столбцам в Pandas

У меня есть фрейм данных Pandas, который выглядит так:

тип расположение проходить зачислен студент НАС да средняя школа студент Калифорния да средняя школа учитель НАС да колледж учитель НАС нет колледж студент НАС нет средняя школа студент Калифорния да колледж студент Калифорния нет колледж

Я хочу получить процент сдачи и сгруппировать его по типу, местоположению и зачислению, чтобы он выглядел примерно так:

тип расположение зачислен pass_rate студент НАС средняя школа .5 студент Калифорния средняя школа 1.0 студент Калифорния колледж .5 учитель НАС колледж 1.0

Чтобы создать вышеуказанный фрейм данных:

import pandas as pd

list_of_dict = [
    {"type": "student", "location": "US", "pass": "yes", "enrolled": "highschool"},
    {"type": "student", "location": "CA", "pass": "yes", "enrolled": "highschool"},
    {"type": "teacher", "location": "US", "pass": "yes", "enrolled": "college"},
    {"type": "teacher", "location": "US", "pass": "no", "enrolled": "college"},
    {"type": "student", "location": "US", "pass": "no", "enrolled": "highschool"},
    {"type": "student", "location": "CA", "pass": "yes", "enrolled": "college"},
    {"type": "student", "location": "CA", "pass": "no", "enrolled": "college"},
]
df = pd.DataFrame(list_of_dict)

Я знаю, что мне нужно получить .count() при группировке по «типу», «местоположению» и «зачислению».

Я пробовал:

df = df.groupby(["type", "location", "enrolled"]).count().mean()

Но это просто дает мне целое число.

Почему в Python есть оператор "pass"?
Почему в Python есть оператор "pass"?
Оператор pass в Python - это простая концепция, которую могут быстро освоить даже новички без опыта программирования.
Некоторые методы, о которых вы не знали, что они существуют в Python
Некоторые методы, о которых вы не знали, что они существуют в Python
Python - самый известный и самый простой в изучении язык в наши дни. Имея широкий спектр применения в области машинного обучения, Data Science,...
Основы Python Часть I
Основы Python Часть I
Вы когда-нибудь задумывались, почему в программах на Python вы видите приведенный ниже код?
LeetCode - 1579. Удаление максимального числа ребер для сохранения полной проходимости графа
LeetCode - 1579. Удаление максимального числа ребер для сохранения полной проходимости графа
Алиса и Боб имеют неориентированный граф из n узлов и трех типов ребер:
Оптимизация кода с помощью тернарного оператора Python
Оптимизация кода с помощью тернарного оператора Python
И последнее, что мы хотели бы показать вам, прежде чем двигаться дальше, это
Советы по эффективной веб-разработке с помощью Python
Советы по эффективной веб-разработке с помощью Python
Как веб-разработчик, Python может стать мощным инструментом для создания эффективных и масштабируемых веб-приложений.
0
0
57
2
Перейти к ответу Данный вопрос помечен как решенный

Ответы 2

Вы можете заменить 'yes' и 'no' на 1 и 0 соответственно в столбце 'pass', затем groupby в других столбцах и получить mean.

df["pass"] = df["pass"].map({"yes": 1, "no": 0})

df = (
    df.groupby(["type", "location", "enrolled"])["pass"]
    .mean()
    .reset_index(name = "pass_rate")
)
      type location    enrolled  pass_rate
0  student       CA     college        0.5
1  student       CA  highschool        1.0
2  student       US  highschool        0.5
3  teacher       US     college        0.5

Плюс 1 от меня... отличное решение с использованием среднего значения.

Scott Boston 09.07.2024 03:50

Вы также можете выполнить замену на месте df.replace({'pass': {'yes':1, 'no':0}}).groupby.....

Anant Gupta 09.07.2024 09:21
Ответ принят как подходящий

Код

используйте crosstab с normalize=0

out = (pd.crosstab([df['type'], df['location'], df['enrolled']], 
                    df['pass'], normalize=0)['yes']
         .reset_index(name='pass_rate')
)

вне:

      type location    enrolled  pass_rate
0  student       CA     college        0.5
1  student       CA  highschool        1.0
2  student       US  highschool        0.5
3  teacher       US     college        0.5

если вы хотите использовать groupby, используйте следующий код:

out = (
    df['pass'].eq('yes')
    .groupby([df['type'], df['location'], df['enrolled']])
    .mean()
    .reset_index(name='pass_rate')
)

тот же результат

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