exercise_1.py

import pandas


def remove_bad_cols(dataframe):
    return df.drop(
        labels=df.columns.difference(["passengerid", "price_paid"]), axis=1
    )


def add_inflation_to_price(dataframe):
    df["price_paid_inflation"] = df.price_paid * 1.3
    return df


df = pandas.DataFrame(
    {
        "col1": [1, 2],
        "col2": [3, 4],
        "passengerid": [2, 3],
        "price_paid": [4, 5],
    }
)