在python中使用矩阵

Question

如何从get_x_and_y_matrix获取输出（x_matrix和y_matrix），并将这些矩阵用作ols_coefficients函数的输入？每次我都会收到一条错误信息

def get_x_and_y_matrix(data_set):
    ones = np.ones(shape=(len(data_set), 1))
    x = data_set[:, 0:2]
    y_matrix = data_set[:, 2]
    x_matrix = np.concatenate((ones, x), axis=1)
    return x_matrix, y_matrix

def ols_coefficients(x, y):
    x, y = get_x_and_y_matrix(data_set=generate_data(n_obs))
    intercept = np.ones(shape=(len(x), 1))
    new_x = np.concatenate((intercept, x), axis=1)
    beta_hat = np.dot(np.linalg.inv(np.dot(intercept, new_x)), np.dot(intercept, y))
    return beta_hat

Answer 1

您可以将来自get_x_and_y_matrix函数的返回保存在变量x和y中，然后将它们作为参数传递给ols_coefficients

def get_x_and_y_matrix(data_set):
    # rest of the function
    return x_matrix, y_matrix

def ols_coefficients(x, y):
    # rest of the function
    return beta_hat

x, y = get_x_and_y_matrix(data_set)
beta = ols_coefficients(x, y)