Как сделать агрегат в стиле R в Python Pandas?

Question

Мне нужно сделать агрегат (по крайней мере, то, что вы бы назвали в R) для набора данных mtcars, который я загрузил в python. Конечная цель - получить среднее число миль на галлон для каждого значения цил в наборе данных (есть три значения цил, 4,6,8). Вот код R для того, что я хочу сделать

mean_each_gear <- агрегат (mtcars $ mpg ~ mtcars $ cyl, FUN = mean) </p>

Выход: цил миль на галлон 1 4 26,66364 2 6 19,74286 3 8 15.10000

Самое близкое, с чем я пришел в Пандах, это

mtcars.agg ([ 'значит'])

Я не уверен, как бы я это сделал в Пандах. Любая помощь будет оценена!

Answer 1

Вы хотите, чтобы панды groupby ()!

import pandas as pd

my_dataframe = pd.read_csv('my_input_data.csv') //insert your data here
pd.groupby(['col1'])['col2'].mean()

где 'col1' - это столбец, по которому вы хотите сгруппировать, а 'col2' - это столбец, среднее значение которого вы хотите получить. Также смотрите здесь:

https://pandas.pydata.org/pandas-docs/stable/generated/pandas.DataFrame.groupby.html