def column(mat, row_i):
    """
    Grab a column from a vector as a numpy column vector.
    :param row_i: row index
    :return: the column vector (shape (n, 1))
    """
    return mat[:, row_i : row_i + 1]