def column(matrix, i):
    """
    Returning all the values in a specific columns
        Parameters:
            X: the input matrix
            i: the column
        Return value: an array with desired column
    """
    return [row[i] for row in matrix]