Массивы с несовместимыми номерами - PullRequest
Новая
       25

Массивы с несовместимыми номерами

0 голосов
/ 05 июня 2019

Это всего лишь пример кода из учебника, который я копировал, и у меня постоянно появляется эта ошибка.Может кто-нибудь дать мне четкое и подробное решение. 

import matplotlib.pyplot as plt
import numpy as np



from sklearn import datasets,linear_model


house_price=[245, 312, 279, 308, 199, 405, 324, 319, 255]
size= [1400, 1600, 1700, 1875, 1100, 1550, 2350, 2450, 1425, 1700]

size2 = np.array(size).reshape((-1,1))
print(size2)



regr = linear_model.LinearRegression()
regr.fit(size2, house_price)
print("coefficient: \n" ,regr.coef_)
print("intercept: \n", regr.intercept_)


def graph(formula, x_range):
    x = np.array(x_range)
    y = eval(formula)
    plt.plot(x,y)

1 Ответ

1 голос
/ 05 июня 2019

Количество цен на жилье не соответствует количеству размеров (элементов в массиве).Они должны соответствовать.В противном случае добавьте np.nan (или вмененное значение, например, медиану), где значение отсутствует.Например: 

house_price=[245, 312, 279, 308, 199, 405, 324, 319, 255, np.nan]
size= [1400, 1600, 1700, 1875, 1100, 1550, 2350, 2450, 1425, 1700]
...