nut-butter-sales-example.py

import numpy as np
import pandas as pd

np.random.seed(0)

sales_amounts = np.random.randint(20, size=(5, 3))

weekly_sales = pd.DataFrame(sales_amounts,
                            index=["Mon", "Tues", "Wed", "Thurs", "Fri"],
                            columns=["Almond butter", "Peanut butter", "Cashew butter"])

prices = np.array([10, 8, 12])

print(f"Weekly sales shape: {weekly_sales.shape}")

butter_prices = pd.DataFrame(prices.reshape(1, 3),
                             index=["Price"],
                             columns=["Almond butter", "Peanut butter", "Cashew butter"])

print(f"Butter prices shape: {butter_prices.shape}")
print(f"Weekly sales shape (transposed): {weekly_sales.T.shape}")

daily_sales = butter_prices.dot(weekly_sales.T)

print(f"Daily sales shape: {daily_sales.shape}")

weekly_sales["Total"] = daily_sales.T

weekly_sales