代码之家  ›  专栏  ›  技术社区  ›  Chris

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

  •  1
  • Chris  · 技术社区  · 6 年前

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

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

    和(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列表压缩成二维,或者其他一些方法可以让它绘图?

    2 回复  |  直到 6 年前
        1
  •  0
  •   Lafa    6 年前

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

    X_reshaped = X.reshape(130, 5)
    
        2
  •  0
  •   Chris    6 年前

    结果是用 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)