Smartphone sales

Comparing smartphone sales units using a stacked area chart.

Importing libraries and packages

# Mathematical operations and data manipulation
import pandas as pd

# Plotting
import matplotlib.pyplot as plt

# Warnings
import warnings

warnings.filterwarnings("ignore")

%matplotlib inline

Set paths

# Path to datasets directory
data_path = "./datasets"
# Path to assets directory (for saving results to)
assets_path = "./assets"

Loading dataset

dataset = pd.read_csv(f"{data_path}/smartphone_sales.csv")

Exploring dataset

# Shape of the dataset
print("Shape of the dataset: ", dataset.shape)
# Head
dataset

Shape of the dataset:  (8, 7)

	Unnamed: 0	Quarter	Apple	Samsung	Huawei	Xiaomi	OPPO
0	0	3Q16	43001	71734	32490	14926	24591
1	1	4Q16	77039	76783	40804	15751	26705
2	2	1Q17	51993	78776	34181	12707	30922
3	3	2Q17	44315	82855	35964	21179	26093
4	4	3Q17	45442	85605	36502	26853	29449
5	5	4Q17	73175	74027	43887	28188	25660
6	6	1Q18	54059	78565	40426	28498	28173
7	7	2Q18	44715	72336	49847	32826	28511

Cleaning dataset

dataset.drop(dataset.columns[[0]], axis=1, inplace=True)
dataset.head()

	Quarter	Apple	Samsung	Huawei	Xiaomi	OPPO
0	3Q16	43001	71734	32490	14926	24591
1	4Q16	77039	76783	40804	15751	26705
2	1Q17	51993	78776	34181	12707	30922
3	2Q17	44315	82855	35964	21179	26093
4	3Q17	45442	85605	36502	26853	29449

Visualisation

# Create figure
plt.figure(figsize=(10, 6), dpi=300)
# Create stacked area chart
labels = dataset.columns[2:]
plt.stackplot(
    "Quarter",
    "Apple",
    "Samsung",
    "Huawei",
    "Xiaomi",
    "OPPO",
    data=dataset,
    labels=labels,
)
# Add legend
plt.legend()
# Add labels and title
plt.xlabel("Quarters")
plt.ylabel("Sales units in thousands")
plt.title("Smartphone sales units")
# Show plot
plt.show()