Python - Split DataFrame to make training set

Question

I have a DataFrame that looks something like this called "sales_csv":

    id     TV  radio  newspaper  sales  invested_amount  successful_campaign
0