python画二次函数_用matplotlib绘制二次函数/模型

我在玩python，想用matplotlib绘制二次线性回归。问题是，我的绘图结果是很多连接的线/点，而不是一个函数：

通常我认为这是行向量与列向量的问题。但当我换位的时候似乎什么都没变。在

我的代码是：from sklearn import datasets, linear_model

import pandas as pd

import numpy as np

import matplotlib.pyplot as plt

# Load diabetes dataset JSON

dsDiabetes = datasets.load_diabetes()

# Create feature- and outcome vectors

lin_train = np.array(dsDiabetes.data[:, np.newaxis, 2])

quad_train = np.concatenate((lin_train, lin_train**2), axis=1)

outcome = np.array(dsDiabetes.target)

# Create regression objects

lin_model = linear_model.LinearRegression()

quad_model = linear_model.LinearRegression()

# lin_train model

lin_model.fit(lin_train, outcome)

quad_model.fit(quad_train, outcome)

# Plot

plt.style.use('fivethirtyeight')

plt.scatter(lin_train, outcome, color='black')

# plt.plot(lin_train, quad_model.predict(quad_train))

# plt.plot(lin_train, lin_model.predict(lin_train), color='blue', linewidth=1)

plt.plot(lin_train, quad_model.predict(quad_train), color='red', linewidth=1)

plt.show()

我错过了什么？在