import matplotlib.pyplot as plt
import numpy as np

# Match order (Latest to Earliest)
matches = ["Final/\nBronze","SF","QF","R16","R32","GS3","GS2","GS1"]

data = {
    "Spain":{
        "xGF":[2.29,1.63,1.96,1.69,2.32,0.86,2.30,2.10],
        "xGA":[0.22,0.31,0.34,0.63,0.32,0.20,0.14,0.20],
        "Opp":["ARG","FRA","BEL","POR","AUT","URU","KSA","CPV"]
    },

    "Argentina":{
        "xGF":[0.22,1.59,1.94,2.84,2.26,2.13,2.36,1.26],
        "xGA":[2.29,0.53,0.47,0.89,0.47,0.76,0.53,0.32],
        "Opp":["ESP","ENG","SUI","EGY","CPV","JOR","AUT","ALG"]
    },

    "England":{
        "xGF":[2.88,0.53,1.04,1.61,2.16,1.56,1.36,3.20],
        "xGA":[2.88,1.59,0.68,1.88,0.77,0.66,0.17,0.70],
        "Opp":["FRA","ARG","NOR","MEX","COD","PAN","GHA","CRO"]
    },

    "France":{
        "xGF":[2.88,0.31,3.69,1.45,3.24,1.50,2.67,1.79],
        "xGA":[2.88,1.63,0.14,0.13,0.70,1.70,0.63,0.53],
        "Opp":["ENG","ESP","MAR","PAR","SWE","NOR","IRQ","SEN"]
    }
}

fig, axes = plt.subplots(2,2,figsize=(15,10),sharex=True,sharey=True)
axes = axes.flatten()

for ax,(team,values) in zip(axes,data.items()):

    diff=np.array(values["xGF"])-np.array(values["xGA"])
    colors=["forestgreen" if d>=0 else "crimson" for d in diff]

    bars=ax.bar(range(len(matches)),
                diff,
                color=colors,
                edgecolor="black",
                linewidth=0.8)

    ax.axhline(0,color="black",linewidth=1.2)

    # Labels
    for i,(bar,d,opp) in enumerate(zip(bars,diff,values["Opp"])):

        y = d+0.08 if d>=0 else d-0.08
        va = "bottom" if d>=0 else "top"

        ax.text(i,
                y,
                f"{d:+.2f}",
                ha="center",
                va=va,
                fontsize=8,
                fontweight="bold")

        # Opponent label
        ax.text(i,
                0.05,
                opp,
                ha="center",
                va="bottom",
                fontsize=8,
                rotation=90,
                fontweight="bold")

    ax.set_title(team,
                 fontsize=15,
                 fontweight="bold")

    ax.set_xticks(range(len(matches)))
    ax.set_xticklabels(matches)

    ax.grid(axis="y",
            linestyle="--",
            alpha=0.35)

    ax.set_ylim(-2.5,3.5)

fig.suptitle(
    "2026 FIFA World Cup Semi-finalists\nMatch-by-Match xG Difference (xGF − xGA)",
    fontsize=19,
    fontweight="bold"
)

fig.supxlabel("Tournament Stage",fontsize=13,fontweight="bold")
fig.supylabel("xG Difference",fontsize=13,fontweight="bold")

plt.tight_layout(rect=[0,0,1,0.95])
plt.show()