Solution Review: Group By Aggregations

import pandas as pd
# Loading dataset
def read_csv():
    # Define the column names as a list
    names = ["mpg", "cylinders", "displacement", "horsepower", 
             "weight", "acceleration", "model_year", "origin", "car_name"]
    # Read in the CSV file using regex for whitespace separation
    df = pd.read_csv("auto-mpg.data", header=None, names=names, sep=r"\s+")
    return df
# Describing data
def group_aggregation(df, group_var, agg_var):
    # Grouping the data and taking mean
    grouped_df = df.groupby([group_var])[agg_var].mean()
    return grouped_df
# Calling the function
print(group_aggregation(read_csv(), "cylinders", "mpg"))