python - 为什么导入的函数不能正常工作？

Question

我使用 numpy.hstack() 方法编写了一个应该水平堆叠矩阵和向量的函数。但是当我将它导入另一个 python 文件时，它并没有做它应该做的事情。这是函数及其使用位置：

我尝试过直接编写没有该功能的代码并且它有效。当我尝试将其作为函数调用时，就会出现问题。

def augmented_matrix(A, b):
    A = np.hstack((A, b))
    return A

A = np.array([[2., 2., 3., 4., 6.],
              [3., 0., 3., 1., 11.],
              [2., 4., 0., 2., 5.],
              [5., 2., 12., 0., 6.],
              [6., 4., 1., 5., 8.]])

b = np.array([[1.], [2.], [4.], [6.], [8.]])

augmented_matrix(A, b)

它仍然给出了结果：

[[ 2.  2.  3.  4.  6.]
[ 3.  0.  3.  1. 11.]
[ 2.  4.  0.  2.  5.]
[ 5.  2. 12.  0.  6.]
[ 6.  4.  1.  5.  8.]]

我期望 5x6 矩阵：

[[ 2.  2.  3.  4.  6.  1.]
 [ 3.  0.  3.  1. 11.  2.]
 [ 2.  4.  0.  2.  5.  4.]
 [ 5.  2. 12.  0.  6.  6.]
 [ 6.  4.  1.  5.  8.  8.]]