Как изменить порядок оси X линейного графика Matplotlib - PullRequest
Новая
       4

Как изменить порядок оси X линейного графика Matplotlib

0 голосов
/ 12 апреля 2020

Я пытаюсь визуализировать результаты регрессионного анализа склеарна, но не могу отсортировать ось х по индексу (месяцу). Есть идеи как это сделать? Вот код и результат: 

data = pd.DataFrame({
    'Month':['2014.01','2014.02','2014.03','2014.04','2014.05','2014.06','2014.07','2014.08','2014.09','2014.10','2014.11','2014.12'], 
    'A': ['101','102','103','104','105','106','107','108','109','110','111','112'],
    'B': ['11','12','13','14','15','16','17','18','19','20','21','22']})    

    y = data['B']
    x = data[['A']]
    x_train, x_test, y_train, y_test = train_test_split(x, y, test_size=0.2, random_state=1)
    linear = LinearRegression().fit(x_train,y_train)
    predicted_price = linear.predict(x_test)
    predicted_price = pd.DataFrame(predicted_price, index = y_test.index, columns = ['Forecast'])
    predicted_price.plot(figsize = (20,5))
    y_test.plot(figsize = (20,5))
    plt.show()

Ось X не отсортирована по месяцам

Спасибо!

1 Ответ

0 голосов
/ 12 апреля 2020

В matplotlib вы можете указать x, y, чтобы вы могли сделать 

plt.plot(y_test.index,y_test)
plt.show()
...