Comment créer un nuage de points à partir d’un DataFrame Pandas

Il existe deux façons de créer un nuage de points à l’aide des données d’un DataFrame pandas :

1. Utilisez pandas.DataFrame.plot.scatter

Une façon de créer un nuage de points consiste à utiliser la fonction pandas plot.scatter() intégrée :

import pandas as pd

df.plot.scatter(x = 'x_column_name', y = 'y_columnn_name')

2. Utilisez matplotlib.pyplot.scatter

Une autre façon de créer un nuage de points consiste à utiliser la fonction Matplotlib pyplot.scatter() :

import matplotlib.pyplot as plt

plt.scatter(df.x, df.y)

Ce didacticiel fournit un exemple d’utilisation de chacune de ces méthodes.

Exemple 1 : utiliser des pandas

Le code suivant montre comment utiliser la fonction plot.scatter() pour créer un nuage de points simple :

import pandas as pd

#create DataFrame
df = pd.DataFrame({'x': [1, 3, 3, 4, 5, 7, 9, 12, 13, 15],
                   'y': [5, 7, 9, 7, 6, 12, 14, 18, 15, 22]})

#create scatterplot
df.plot.scatter(x='x', y='y')

Notez que vous pouvez utiliser les arguments s et c pour modifier respectivement la taille et la couleur des points :

df.plot.scatter(x='x', y='y', s=60, c='green')

Exemple 2 : utiliser Matplotlib

Le code suivant montre comment utiliser la fonction pyplot.scatter() pour créer un nuage de points :

import pandas as pd
import matplotlib.pyplot as plt

#create DataFrame
df = pd.DataFrame({'x': [1, 3, 3, 4, 5, 7, 9, 12, 13, 15],
                   'y': [5, 7, 9, 7, 6, 12, 14, 18, 15, 22]})

#create scatterplot
plt.scatter(df.x, df.y)

Notez que vous pouvez utiliser les arguments s et c pour modifier respectivement la taille et la couleur des points :

plt.scatter(df.x, df.y, s=60, c='purple')

Vous pouvez trouver plus de didacticiels Python ici .