data_crosses = data_crosses[data_crosses['is_cross']].reset_index(drop=True)
data_crosses = pd.merge(data_crosses, team_map[['team_id', 'fotmob_id']], how='left', on='team_id')
data_crosses

fig = plt.figure(figsize=(4,4), dpi=300)
ax = plt.subplot(111)
pitch = VerticalPitch(
    pitch_type='opta',
    goal_type='box',
    linewidth=.85,
    line_color='black',
    half=True
)
pitch.draw(ax = ax)

# -- Split successful corners from unsuccessful attempts.
assists = test_df[test_df['is_assist'] == True]
others = test_df[test_df['is_assist'] != True]
other_s = test_df[test_df['outcome_type'] == 'Successful']
other_u = test_df[test_df['outcome_type'] != 'Successful']

lines(other_s.y, other_s.x, other_s.end_y, other_s.end_x, 
    ax=ax, comet=True, alpha_start=0.01, alpha_end=0.05, lw=.3, transparent=True, color='#307351')

lines(other_u.y, other_u.x, other_u.end_y, other_u.end_x, 
    ax=ax, comet=True, alpha_start=0.01, alpha_end=0.05, lw=.3, transparent=True, color='#70163C')

for index, a in assists.iterrows():
    # ax.plot([a['y'], a['end_y']], [a['x'], a['end_x']], lw=1, color='#307351')
    pass_arrow = mpatches.FancyArrowPatch((a.y, a.x), (a.end_y, a.end_x), ec='#307351',
                                        fc='#efe9e6',arrowstyle='-|>,head_length=5.6,head_width=2.2', zorder=4)
    ax.add_patch(pass_arrow)

ax.scatter(assists['y'], assists['x'], marker='o', fc=ax.get_facecolor(), ec='#307351', zorder=4)
ax.scatter(other_s['y'], other_s['x'], marker='o', fc=ax.get_facecolor(), ec='#307351', zorder=3, alpha=.5, s=5, lw=0.2)
ax.scatter(other_u['y'], other_u['x'], marker='o', fc=ax.get_facecolor(), ec='#70163C', zorder=3, alpha=.5, s=5, lw=0.2)

# Separate left from right hand side
# -- Count which side had most crosses:
left_side = test_df[test_df['y'] > 50].shape[0]
right_side = test_df[test_df['y'] < 50].shape[0]
left_success = other_s[other_s['y'] > 50].shape[0]
right_success = other_s[other_s['y'] < 50].shape[0]

if right_side > left_side:
    color_left = '#e7deda'
    color_right = '#d7c8c1'
else:
    color_right = '#e7deda'
    color_left = '#d7c8c1'

ax.fill_between(
    x=[0.35,50], y1=ax.get_ylim()[0], y2=99.65,
    color=color_right, zorder=-1
)
ax.fill_between(
    x=[99.35,50], y1=ax.get_ylim()[0], y2=99.65,
    color=color_left, zorder=-1
)

# -- Annotations
ax_text(
    x=99, y=52,
    s=f'Crosses: {left_side:.0f} | <Success rate: {left_success/left_side:.1%}>',
    ha='left', va='center',
    weight='bold', textalign='top',
    highlight_textprops=[{'color':'#307351'}],
    size=4,
    ax=ax
)

ax_text(
    x=34, y=52,
    s=f'Crosses: {right_side:.0f} | <Success rate: {right_success/right_side:.1%}>',
    ha='left', va='center',
    weight='bold', textalign='top',
    highlight_textprops=[{'color':'#307351'}],
    size=4,
    ax=ax
)

ax.annotate(
    xy=(50,105),
    text=f"{test_df['team_name'].iloc[0].upper()}",
    size=9.5,
    color='black',
    ha='center',
    va='center',
    weight='bold',
    annotation_clip=False
)

# -- Add Logo
DC_to_FC = ax.transData.transform
FC_to_NFC = fig.transFigure.inverted().transform
# -- Take data coordinates and transform them to normalized figure coordinates
DC_to_NFC = lambda x: FC_to_NFC(DC_to_FC(x))
ax_coords = DC_to_NFC((11,102))
ax_size = 0.05
image_ax = fig.add_axes(
    [ax_coords[0], ax_coords[1], ax_size, ax_size],
    fc='None'
)
fotmob_url = 'https://images.fotmob.com/image_resources/logo/teamlogo/'
club_icon = Image.open(urllib.request.urlopen(f'{fotmob_url}{test_df["fotmob_id"].iloc[0]:.0f}.png')).convert('LA')
image_ax.imshow(club_icon)
image_ax.axis('off')

order_teams = data_crosses.groupby(['team_name', 'fotmob_id'])['is_cross'].sum().reset_index().sort_values(by='is_cross', ascending=False).reset_index(drop=True)
order_teams

fig, axs = plt.subplots(nrows=4, ncols=5, figsize=(18,18), dpi=300)
axs = np.array(axs)
for index, ax in enumerate(axs.reshape(-1)):
    plot_crosses_by_team_name(ax, order_teams['team_name'].iloc[index])

plt.subplots_adjust(wspace=0, hspace=-.4)
# We add the logos once we've reshaped the figure
for index, ax in enumerate(axs.reshape(-1)):
    # -- Transformation functions
    DC_to_FC = ax.transData.transform
    FC_to_NFC = fig.transFigure.inverted().transform
    # -- Take data coordinates and transform them to normalized figure coordinates
    DC_to_NFC = lambda x: FC_to_NFC(DC_to_FC(x))
    ax_coords = DC_to_NFC((14,87))
    ax_size = 0.017
    image_ax = fig.add_axes(
        [ax_coords[0], ax_coords[1], ax_size, ax_size],
        fc='None'
    )
    fotmob_url = 'https://images.fotmob.com/image_resources/logo/teamlogo/'
    club_icon = Image.open(urllib.request.urlopen(f'{fotmob_url}{order_teams["fotmob_id"].iloc[index]:.0f}.png')).convert('LA')
    image_ax.imshow(club_icon)
    image_ax.axis('off')


fig_text(
    x = 0.5, y = .87, 
    s = "WHERE DO CROSSES COME FROM IN THE PREMIER LEAGUE?",
    va = "bottom", ha = "center",
    fontsize = 30, color = "black", font = "DM Sans", weight = "bold"
)
fig_text(
	x = 0.5, y = .85, 
    s = "Non-corner crosses for EPL teams, <arrows denote assists> | Season 2022/2023 | viz by @sonofacorner, data from Opta as of 2nd of January 2023",
    highlight_textprops=[{"weight": "bold", "color": "#307351"}],
	va = "bottom", ha = "center",
	fontsize = 14, color = "#4E616C", font = "DM Sans"
)

plt.savefig(
	"figures/01022023_epl_crosses.png",
	dpi = 600,
	facecolor = "#EFE9E6",
	bbox_inches="tight",
    edgecolor="none",
	transparent = False
)

plt.savefig(
	"figures/01022023_epl_crosses_tr.png",
	dpi = 600,
	facecolor = "none",
	bbox_inches="tight",
    edgecolor="none",
	transparent = True
)