# Mini Python Practice Lab
# Module 0 - Mini Python practice lab
# Dataset: /data/wage_sample.csv

# Mini Python Practice Lab
#
# Run the basic workflow: load the data, inspect it, summarize it, graph it, and write a cautious interpretation.

# %% Cell 2
import pandas as pd
import matplotlib.pyplot as plt

df = pd.read_csv("wage_sample.csv")
print(df.head())

# %% Cell 3
summary = df[["wage", "education", "experience"]].describe()
print(summary)

# %% Cell 4
print("Correlation between wage and education:")
print(round(df["wage"].corr(df["education"]), 3))

# %% Cell 5
plt.scatter(df["education"], df["wage"])
plt.xlabel("Education")
plt.ylabel("Wage")
plt.title("Practice lab: wage and education")
plt.show()

# Interpretation prompt
#
# Describe the pattern in the sample. Then add one caution about why this is not automatically causal.