Использование матриц в питоне

Question

Как я могу взять выходные данные (x_matrix и y_matrix) из get_x_and_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)