为什么导入的功能不能发挥作用？

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.]]