在Python中使用matplotlib转换要打印的数组列表

2024-04-23 23:06:40 发布

您现在位置:Python中文网/ 问答频道 /正文

我尝试使用sklearn包生成滚动回归,然后使用matplotlib绘制它们。你知道吗

我已经能够生成滚动回归系数,但是附加输出会产生一个我很难绘制的三维列表。你知道吗

下面的代码出现以下错误:

ValueError: x and y can be no greater than 2-D, but have shapes (130,) and (130, 1, 5)

rCoeff = []
lm = sk_l.LinearRegression()
for iS in range(1, len(y)-(rollingN-1)):

    iE = iS+(rollingN-1)
    subX = X[iS:iE]
    suby = y[iS:iE]

    lm.fit(subX,suby)
    rCoeff.append(lm.coef_)

x = df_cpf.loc[36:166,'Date']
plt.plot_date(x,rCoeff)

有没有一种方法可以把3D列表压缩成二维,或者其他一些方法可以让它绘图?你知道吗


Tags: and方法代码列表matplotlibis绘制sklearn
2条回答

结果是使用extend()方法而不是append()方法完成了这个任务。你知道吗

rCoeff = []
lm = sk_l.LinearRegression()
for iS in range(1, len(y)-(rollingN-1)):

    iE = iS+(rollingN-1)
    subX = X[iS:iE]
    suby = y[iS:iE]

    lm.fit(subX,suby)
    rCoeff.extend(lm.coef_)

x = df_cpf.loc[36:166,'Date']
plt.plot_date(x,rCoeff)

可以通过执行以下操作重塑阵列:

X_reshaped = X.reshape(130, 5)

相关问题 更多 >