打印X_train形状.你看到了什么?我敢打赌X_train是2d(矩阵有一列),而y_train1d(向量).反过来你会得到不同的尺寸.
X_train
y_train
我认为X_train[:,0]用于绘图(来自错误的起源)应该可以解决问题
X_train[:,0]