Comment tracer un nuage de points à partir de deux colonnes d'une dataframe avec pandas en python ?

Exemples de comment tracer un nuage de points à partir de deux colonnes d'une dataframe avec pandas en python:

Tracer un nuage un nuage de points (exemple 1)

Créons une simple dataframe avec pandas

import pandas as pd
import matplotlib.pyplot as plt

data = {'c':['a','b','c','d','e','f','g','h','i','f'], 
        'x':[0,1,2,3,4,5,6,7,8,9],
        'y':[0,0,0,0,0,0,0,0,0,0]}

data['y'] = [i* 2.0 + 1.0 for i in  data['x'] ]

df = pd.DataFrame(data)

print(df)

donne

   c  x     y
0  a  0   1.0
1  b  1   3.0
2  c  2   5.0
3  d  3   7.0
4  e  4   9.0
5  f  5  11.0
6  g  6  13.0
7  h  7  15.0
8  i  8  17.0
9  f  9  19.0

pour tracer un nuage de points à partir des colonnes x et y, une solution est d''utiliser pandas.DataFrame.plot:

df.plot(x='x', y='y', style='o')

plt.savefig("pandas_scatter_plot_01.png", bbox_inches='tight', dpi=100)

donne

Tracer un nuage un nuage de points (exemple 2)

Un autre exemple en améliorant un peu l'apparence de la figure:

ax = df.plot(x='x', y='y', style='o', legend=False)

ax.set_xlabel("x label")
ax.set_ylabel("y label")

ax.set_title("Create a scatter plot with pandas")

ax.set_xlim(0,10)
ax.set_ylim(0,20)

ax.grid()

plt.savefig("pandas_scatter_plot_02.png", bbox_inches='tight', dpi=100)

donne

Références

• How to plot two columns of a pandas data frame using points?
• Add x and y labels to a pandas plot
• Pandas plot() without a legend