Création d'ellipses de confiance dans un sccatterplot à l'aide de matplotlib

Comment créer de la Confiance des Ellipses dans un sccatterplot à l'aide de matplotlib?

Le code suivant fonctionne jusqu'à ce que la création d'un nuage de points. Alors, quiconque est familier avec la mise en
La confiance des Ellipses sur le nuage de points?

import numpy as np
import matplotlib.pyplot as plt
x = [5,7,11,15,16,17,18]
y = [8, 5, 8, 9, 17, 18, 25]

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

Qui suit est la référence pour la Confiance Ellipses de SAS.

http://support.sas.com/documentation/cdl/en/grstatproc/62603/HTML/default/viewer.htm#a003160800.htm

Le code de la sas est comme ceci:

proc sgscatter data=sashelp.iris(where=(species="Versicolor"));
  title "Versicolor Length and Width";
  compare y=(sepalwidth petalwidth)
          x=(sepallength petallength)
          / reg ellipse=(type=mean) spacing=4;
run;

source d'informationauteur 2964502

  1. 14

    Le code suivant dessine un, deux, et trois écart-type de la taille des ellipses:

    x = [5,7,11,15,16,17,18]
y = [8, 5, 8, 9, 17, 18, 25]
cov = np.cov(x, y)
lambda_, v = np.linalg.eig(cov)
lambda_ = np.sqrt(lambda_)
from matplotlib.patches import Ellipse
import matplotlib.pyplot as plt
ax = plt.subplot(111, aspect='equal')
for j in xrange(1, 4):
    ell = Ellipse(xy=(np.mean(x), np.mean(y)),
                  width=lambda_[0]*j*2, height=lambda_[1]*j*2,
                  angle=np.rad2deg(np.arccos(v[0, 0])))
    ell.set_facecolor('none')
    ax.add_artist(ell)
plt.scatter(x, y)
plt.show()

    Création d'ellipses de confiance dans un sccatterplot à l'aide de matplotlib

  2. 12

    Après avoir accepté de répondre à un aller, j'ai trouvé qu'il ne choisit pas le quadrant correctement lors du calcul de thêta, car elle repose sur np.arccos:

    Création d'ellipses de confiance dans un sccatterplot à l'aide de matplotlib

    De prendre un coup d'oeil à la 'double' et Joe kingston portant solution sur githubj'ai arrosé son code:

    import numpy as np
import matplotlib.pyplot as plt
from matplotlib.patches import Ellipse

def eigsorted(cov):
    vals, vecs = np.linalg.eigh(cov)
    order = vals.argsort()[::-1]
    return vals[order], vecs[:,order]

x = [5,7,11,15,16,17,18]
y = [25, 18, 17, 9, 8, 5, 8]

nstd = 2
ax = plt.subplot(111)

cov = np.cov(x, y)
vals, vecs = eigsorted(cov)
theta = np.degrees(np.arctan2(*vecs[:,0][::-1]))
w, h = 2 * nstd * np.sqrt(vals)
ell = Ellipse(xy=(np.mean(x), np.mean(y)),
              width=w, height=h,
              angle=theta, color='black')
ell.set_facecolor('none')
ax.add_artist(ell)
plt.scatter(x, y)
plt.show()

    Création d'ellipses de confiance dans un sccatterplot à l'aide de matplotlib