TransWikia.com

Linha de tendencia em scatterplot no matplotlib em Python

Stack Overflow em Português Asked by vinifrm on November 27, 2021

Supondo que eu tenha duas listas simples em Python e crie um gráfico de dispersão:

x = [548, 677, 987, 2, 29, 1114, 521, 999] 
y = [96, 775, 258, 369, 410, 99, 5, 1117]
import matplotlib.pyplot as plt
plt.scatter(x, y)

Como posso criar uma linha de tendência dentro desse mesmo gráfico?

One Answer

Você pode fazer isso rapidamente através da função Polyfit do numpy.

import matplotlib.pyplot as plt
import numpy as np

x = [548, 677, 987, 2, 29, 1114, 521, 999] 
y = [96, 775, 258, 369, 410, 99, 5, 1117]

plt.scatter(x, y)

z = np.polyfit(x, y, 1)
p = np.poly1d(z)
plt.plot(x,p(x),"r--")

Saída:

inserir a descrição da imagem aqui

Essa que eu fiz é o basicão, mas dá pra fazer algumas coisas bem legais, como você pode ver aqui.

Answered by Evilmaax on November 27, 2021

Add your own answers!

Ask a Question

Get help from others!

© 2024 TransWikia.com. All rights reserved. Sites we Love: PCI Database, UKBizDB, Menu Kuliner, Sharing RPP