Линейная регрессия с matplotlib / numpy

Я пытаюсь создать линейную регрессию на графике рассеяния, который я сгенерировал, однако мои данные представлены в формате списка, и все примеры, которые я могу найти, используют polyfit требует использования arange . arange не принимает списки. Я много раз искал, как преобразовать список в массив, и ничего не кажется ясным. Я что-то упускаю?

Далее, как лучше всего использовать мой список целых чисел в качестве входных данных для polyfit ?

вот пример полифита, которому я следую:

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