def remove(data, columns):
    """ Simply remove columns containing missing values.

        :param data: AutoData data
        :param columns: Column(s) to impute
        :return: Imputed data
        :rtype: autopandas.AutoData
    """
    data = data.dropna(axis=0, subset=columns)
    return data
