Estou tentando gerar uma regressão linear em um gráfico de dispersão que gere, no entanto, meus dados estão no formato de lista e todos os exemplos que posso encontrar do uso depolyfit requer o uso dearange. arange não aceita listas. Eu pesquisei alto e baixo sobre como converter uma lista em uma matriz e nada parece claro. Estou esquecendo de algo

A seguir, qual a melhor forma de usar minha lista de números inteiros como entradas para opolyfit?

aqui é o exemplo de polyfit que estou seguindo:

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()