python - 为 armodel 创建矩阵，但 np.arange 返回 none 类型，即使矩阵正确。如何转换为数组？以下是我到目前为止的代码

Question

我有一个函数可以为 ar_model 生成矩阵 x 和 y，以使用最小二乘法计算系数。但是，当我正确打印时，np.arange 会按原样打印 x 矩阵，但是当我将 np.arange 值转换为数组时，它是不正确的。请帮助我如何正确生成矩阵的数组版本。谢谢！

#example lists
x = [1,2,3,4,5]
y = [2,4,6,8,10,12,14]

def matrix(x, na): #na is order or armodel
    X = np.array(x)
    N = len(X)
    p = na
    for n in range(p, N):
       u = X[np.arange((n-1),(n-p-1),-1)]
       matrix = print(u)
       array = np.array(matrix) #not correct
       #need to get the negative versions of u
       #but u isnt an array so I wasn't able multiply by -1
       
       #matrix_y
       y = X[na:]
       
    return matrix, array

标签： pythonpandasnumpytime-series