La régression linéaire avec matplotlib / numpy
Je suis en train de générer une régression linéaire sur un nuage de points, j'ai généré, mais mes données sont sous forme de liste, et tous les exemples que je peux trouver de l'aide polyfit
besoin de l'aide des arange
. arange
n'accepte pas les listes si. J'ai cherché haut et bas sur la façon de convertir une liste à un tableau et rien ne semble évident. Ai-je raté quelque chose?
Suite, comment puis-je utiliser ma liste d'entiers comme intrants pour la polyfit
?
ici est la polyfit exemple je suis:
from pylab import *
x = arange(data)
y = arange(data)
m,b = polyfit(x, y, 1)
plot(x, y, 'yo', x, m*x+b, '--k')
show()
- Essayez
regplot
deseaborn
: stackoverflow.com/a/42263217/911945
Vous devez vous connecter pour publier un commentaire.
arange
génère listes (bien, tableaux numpy); type dehelp(np.arange)
pour plus de détails. Vous n'avez pas besoin d'appeler les listes existantes.Je dois ajouter que j'ai tendance à utiliser
poly1d
ici plutôt que d'écrire "m*x+b" et l'ordre supérieur équivalents, donc ma version de votre code devrait ressembler à quelque chose comme ceci:Ce code:
donne une liste avec les éléments suivants:
Source
Utiliser cette ..
Une autre rapide et sale réponse est que vous pouvez simplement convertir votre liste à un tableau à l'aide de: