Annotating scatterplot points with DF column text Matplotlib

Question

I'm fairly new to Python and I'm struggling annotating plots at the minute. I've come from R so I'm used to the ease of being able to annotate scatterplot points with minimum code.

Code:

import pandas as pd 
import numpy as np
import matplotlib.pyplot as plt 
import matplotlib as mpl

url = ('https://fbref.com/en/share/nXtrf')
df = pd.read_html(url)[0]
df = df[['Unnamed: 1_level_0', 'Unnamed: 2_level_0', 'Play', 'Perf']].copy()
df.columns = df.columns.droplevel()
df = df[['Player','Squad','Min','SoTA','Saves']]
df = df.drop([25])
df['Min'] = pd.to_numeric(df['Min'])
df['SoTA'] = pd.to_numeric(df['SoTA'])
df['Saves'] = pd.to_numeric(df['Saves'])
df['Min'] = df[df['Min'] > 1600]['Min']
df = df.dropna()

df.plot(x = 'Saves', y = 'SoTA', kind = "scatter")

I've tried numerous ways to annotate this plot. I'd like the points to be annotated with corresponding data from 'Player' column.

I've tried using a label_point function that I've found while trying to find a work around buy I keep getting Key Error 0 on most ways I try.

Any assistance would be great. Thanks.

JohanC · Accepted Answer · 2021-01-31 17:51:35Z

1

You could loop through both columns and add a text for each entry. Note that you need to save the ax returned by df.plot(...).

ax = df.plot(x='Saves', y='SoTA', kind="scatter")

for x, y, player in zip(df['Saves'], df['SoTA'], df['Player']):
    ax.text(x, y, f'{player}', ha='left', va='bottom')
xmin, xmax = ax.get_xlim()
ax.set_xlim(xmin, xmax + 0.15 * (xmax - xmin))  # some more margin to fit the texts

An alternative is to use the mplcursors library to show an annotation while hovering (or after a click):

import mplcursors

mplcursors.cursor(hover=True)

edited Jan 31, 2021 at 17:51

answered Jan 31, 2021 at 17:16

JohanC

81.4k8 gold badges54 silver badges90 bronze badges

Sign up to request clarification or add additional context in comments.

2 Comments

ScoutingForJay Over a year ago

thanks for that, in the first example you gave me. That returns the x,y coordinates and pastes them as text. How would I replace the xy coordinates with text from the 'Players' column in the df?

JohanC Over a year ago

To avoid overlapping texts, you could try Matplotlib overlapping annotations / text. That library can be installed as pip install adjustText. Note that the mplcursor library also facilitates customizations such as showing a third column.

Collectives™ on Stack Overflow

Annotating scatterplot points with DF column text Matplotlib

1 Answer 1

2 Comments

Your Answer

Linked

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

2 Comments

Your Answer

Sign up or log in

Post as a guest

Linked

Related