Notes

import pandas as pd
import matplotlib.pyplot as plt

# Sample data for demonstration (replace this with your actual data)
# Let's assume you have sales data for several time periods.
time_periods = ['Period 1', 'Period 2', 'Period 3', 'Period 4', 'Period 5']
test_sales = [150, 180, 160, 170, 155] # Average sales for test stores over time
control_sales = [200, 210, 205, 220, 215] # Average sales for control stores over the same periods

# Create a DataFrame for better handling
data = pd.DataFrame({
'Time Period': time_periods,
'Test Sales': test_sales,
'Control Sales': control_sales
})

# Set the Time Period as the index
data.set_index('Time Period', inplace=True)

# Create the line plot
plt.figure(figsize=(12, 6))
plt.plot(data.index, data['Test Sales'], marker='o', label='Test Stores (Powerbase)', color='orange')
plt.plot(data.index, data['Control Sales'], marker='o', label='Control Stores', color='blue')

# Adding labels and title
plt.title('Sales Comparison: Test vs Control Stores Over Time')
plt.xlabel('Time Period')
plt.ylabel('Average Sales')
plt.xticks(rotation=45)
plt.legend()
plt.grid()

# Show the plot
plt.tight_layout()
plt.show()