Package: pandas-dq
pandas_dqpandas-dqis the ultimate data quality toolkit for pandas dataframes.Table of ContentsWhat is pandas_dqWhat are its main componentsHow to use pandas_dqHow to install pandas_dqUsageAPIMaintainersContributingLicenseIntroductionpandas_dqis a new python library for data quality analysis and improvement. It is fast, efficient and scalable.pandas-dqis:A smart and simple way to clean and improve your pandas dataframes.A powerful way to boost your data analysis with high-quality pandas dataframes.A powerful and flexible library for data quality management in pandas.Data quality made easy with pandas and scikit-learn transformersThe newpandas_dqlibrary in Python is a great addition to thepandasecosystem. It provides a set of tools for data quality assessment, which can be used to identify and address potential problems with data sets. This can help to improve the quality of data analysis and ensure that results are reliable.Thepandas_dqlibrary is still under development, but it already includes a number of useful features. These include:Data profiling: pandas_dq displays a report either in-line or in HTML to give you a quick overview of your data, including its features, feature types, their null and unique value percentages, their maximum and minimum values.Train Test comparison: pandas_dq displays a comparison report either in-line or in HTML to give you a quick comparison of your train and test dataasets, including their distributional differences (using the KS Test), and comparing their null and unique value percentages.Data cleaning: pandas_dq allows you to quickly identify and remove data quality issues and inconsistencies in your data set.Data imputation: pandas_dq allows you to fill missing values with your own choice of values for each feature in your data. For example, you can have one default foragefeature and another forincomefeature.Data transformation: pandas_dq allows you to transform skewed features into a more normal-like distribution.Thepandas_dqlibrary is a valuable tool for anyone who works with data. It can help you to improve the quality of your data analysis and ensure that your results are reliable.Here are some of the benefits of using the pandas_dq library:It can help you to identify and address potential problems with data sets before modeling.It can fix data quality issues and improve the quality of your data.It is easy to use and can be integrated with otherscikit-learnpipelines.Alert!: If you are usingpandas version 2.0("the new pandas"), beware that weird errors are popping up in all kinds of libraries that use pandas underneath. Ourpandas_dqlibrary is no exception. So if you plan to usepandas_dqwithpandas version 2.0, beware that you may see weird errors and we can't and won't fix them!Componentspandas_dqhas the following main modules:dq_report: The data quality report displays a data quality report either inline or in HTML after it analyzes your dataset for various issues, such as missing values, outliers, duplicates, correlations, etc. It also checks the relationship between the features and the target variable (if provided) to detect data leakage.dc_report: The data comparison report displays a comparison report between train and test datasets either inline or in HTML after it analyzes both datasets for various issues, such as missing values, unique values, min and max, etc. It also checks provides a Statistical Test (KS test) to compare the distribitional differences of numeric features to detect data drift. You can exclude target column(s) from comparison between train and test.Fix_DQ: This class is a scikit-learn compatible transformer that can detect and fix data quality issues in one line of code. It can remove ID columns, zero-variance columns, rare categories, infinite values, mixed data types, outliers, high cardinality features, highly correlated features, duplicate rows and columns, skewed distributions and imbalanced classes.DataSchemaChecker: This class can check your dataset data types against a specific schema and report any mismatches or errors.pandas_dqis designed to provide you the cleanest features with the fewest steps.Usespandas_dqhas multiple important modules:dq_report,Fix_DQand nowDataSchemaChecker.1.  dq_report function`dq_report` displays a data quality report (inline or HTML) after it analyzes your dataset looking for these issues:It detects ID columnsIt detects zero-variance columnsIt identifies rare categories (less than 5% of categories in a column)It finds infinite values in a columnIt detects mixed data types (i.e. a column that has more than a single data type)It detects outliers (i.e. a float column that is beyond the Inter Quartile Range)It detects high cardinality features (i.e. a feature that has more than 100 categories)It detects highly correlated features (i.e. two features that have an absolute correlation higher than 0.8)It detects duplicate rows (i.e. the same row occurs more than once in the dataset)It detects duplicate columns (i.e. the same column occurs twice or more in the dataset)It detects skewed distributions (i.e. a feature that has a skew more than 1.0)It detects imbalanced classes (i.e. target variable has one class more than other in a significant way)It detects feature leakage (i.e. a feature that is highly correlated to target with correlation > 0.8)Notice that for large datasets, this report generation may take time, hence we read a 100K sample from your CSV file. If you want us to read the whole data, then send it in as a dataframe.2.  dc_report functiondc_reportis a data comparison tool that accepts two pandas dataframes as input and returns a report highlighting any differences between them. For example:The function uses our function `dqr = dq_report(df)` to generate a data quality report for each dataframe and compares the results using the column names from the report.It also computes the Kolmogorov-Smirnov test statistic to measure the distribution difference for numeric columns with low cardinality.It also compares the Missing Values% and Unique Values% between the two dataframes and adds a comment in the "Distribution Difference" column if the two percentages are different.You can exclude target column(s) from comparison between train and test.- Notice that for large datasets, this report generation may take time. So make sure you take a sample of your train and test data before calling this report!3.  Fix_DQ class: a scikit_learn transformer which can detect data quality issues and fix them all in one line of code`Fix_DQ` is a great way to clean an entire train data set and apply the same steps in an MLOps pipeline to a test dataset.  `Fix_DQ` can be used to detect most issues in your data (similar to dq_report but without the `target` related issues) in one step. Then it fixes those issues it finds during the `fit` method by the `transform` method. This transformer can then be saved (or "pickled") for applying the same steps on test data either at the same time or later.Fix_DQ will perform following data quality cleaning steps:It removes ID columns from further processingIt removes zero-variance columns from further processingIt identifies rare categories and groups them into a single category called "Rare"It finds infinite values and replaces them with an upper bound based on Inter Quartile RangeIt detects mixed data types and drops those mixed-type columns from further processingIt detects outliers and suggests to remove them or use robust statistics.It detects high cardinality features but leaves them as it is.It detects highly correlated features and drops one of them (whichever comes first in the column sequence)It detects duplicate rows and drops one of them or keeps only one copy of duplicate rowsIt detects duplicate columns and drops one of them or keeps only one copyIt detects skewed distributions and applies log or box-cox transformations on themIt detects imbalanced classes and leaves them as it isIt detects feature leakage and drops one of those features if they are highly correlated to targetHow can we use Fix_DQ in GridSearchCV to find the best model pipeline?This is another way to find the best data cleaning steps for your train data and then use the cleaned data in hyper parameter tuning using GridSearchCV or RandomizedSearchCV along with a LightGBM or an XGBoost or a scikit-learn model.4.  DataSchemaChecker class: a scikit_learn transformer that can check whether a pandas dataframe conforms to a given schema and coerces the data to conform to it.The DataSchemaChecker class has two methods: fit and transform. You need to initialize the class with a schema that you want to compare your data's dtypes against. A schema is a dictionary that maps column names to data types.The fit method takes a dataframe as an argument and checks if it matches the schema. The fit method first checks if the number of columns in the dataframe and the schema are equal. If not, it creates an exception. Finally, the fit method displays a table of exceptions it found in your data against the given schema.The transform method takes a dataframe as an argument and based on the given schema and the exceptions, converts all the exception data columns to the given schema. If it is not able to transform the column, it skips the column and displays out an error message.InstallPrerequsites:pandas_dq is built using pandas, numpy and scikit-learn - that's all.It should run on almost all Python3 Anaconda installations without additional installs. You won't have to import any special libraries.The best method to install pandas_dq is to use pip:pip install pandas_dqTo install from source:cd <pandas_dq_Destination>
git clone git@github.com:AutoViML/pandas_dq.gitor download and unziphttps://github.com/AutoViML/pandas_dq/archive/master.zipconda create -n <your_env_name> python=3.7 anaconda
conda activate <your_env_name> # ON WINDOWS: `source activate <your_env_name>`
cd pandas_dq
pip install -r requirements.txtUsageTo get a quick profile of your data, can simply call dq_reportfrom pandas_dq import dq_report
dqr = dq_report(data, target=target, html=False, csv_engine="pandas", verbose=1)It displays a data quality report like this inline or in HTML format (and it saves the HTML to your machine):To get a quick comparison of two sets of data frames,  simply call dc_reportdc_reportis data comparison tool that accepts two pandas dataframes as input and returns a report highlighting any differences between them. It can also provide a report in HTML format as below.from pandas_dq import dc_report
dc_report = dc_report(train, test, exclude=[], html=True, verbose=1)To fix your data quality issues, useFix_DQas a scikit-learn compatible transformerfrom pandas_dq import Fix_DQ

# Create an instance of the fix_data_quality transformer with default parameters
fdq = Fix_DQ()

# Fit the transformer on X_train and transform it
X_train_transformed = fdq.fit_transform(X_train)

# Transform X_test using the fitted transformer
X_test_transformed = fdq.transform(X_test)To validate that your data conforms to a given schema, use DataSchemaChecker:Once you define the schema as below, you can use it as follows:schema = {'name': 'string',
        'age': 'float32',
        'gender': 'object',
        'income': 'float64',
        'date': 'date',
        'target': 'integer'}from pandas_dq import DataSchemaChecker

ds = DataSchemaChecker(schema=schema)
ds.fit_transform(X_train)
df.transform(X_test)APIpandas_dq has a very simple API with one major goal: find data quality issues in your data and fix them.Argumentsdq_reporthas the following arguments:Caution:For very large data sets, we randomly sample 100K rows from your CSV file to speed up reporting. If you want a larger sample, simply read in your file offline into a pandas dataframe and send it in as input, and we will load it as it is. This is one way to go around our speed limitations:Inputs:data: You can provide any kind of file format (string) or even a pandas DataFrame (df). It reads parquet, csv, feather, arrow, all kinds of file formats straight from disk. You just have to tell it the path to the file and the name of the file.target: default:None. Otherwise, it should be a string name representing the name of a column in df. You can leave it asNoneif you don't want any target related issues.html: default isFalse. If you want to display your report in HTML in a browser, set it toTrue. Otherwise, it defaults to inline in a notebook or prints on the terminal. It also saves the HTML file in your working directory in your machine.csv_engine: default ispandas. If you want to load your CSV file using any other backend engine such asarroworparquetplease specify it here. This option only impacts CSV files.verbose: This has 2 possible states:0summary report. displays only the summary level data quality issues in the dataset. Great for managers.1detailed report. displays all the gory details behind each DQ issue in your dataset and what to do about them. Great for engineers.Outputs:dataframe: If verbose=1, it returns a dataframe with detailed data quality issues with your data. If verbose=0, it returns with a dataframe containing only the highlights of the data quality issues.dc_reportreturns a dataframe highlighting differences between two dataframes, typically train and test. It has the following inputs and outputs:Inputs:train: a dataframetest: a dataframeexclude: an empty list or a list of columns that you want to exclude from comparison in both dataframeshtml: return a HTML file containing the differences between the two dataframesverbose: 0 will return just the highlights of differences. 1 will return a detailed description of differences between the two dataframes.Outputs:dataframe: If verbose=1, it returns a dataframe with the following column names: Column Name, Data Type Train, Data Type Test, Missing Values% Train, Missing Values% Test, Unique Values% Train, Unique Values% Test, Minimum Value Train, Minimum Value Test, Maximum Value Train, Maximum Value Test, DQ Issue Train, DQ Issue Test, Distribution Difference. If verbose=0, it will return only the following columns: Column Name, DQ Issue Train, DQ Issue Test, Distribution Difference.Fix_DQis a scikit-learn transformer. It finds and fixes data quality issues in your dataCaution:X_train and X_test in Fix_DQ must be pandas Dataframes or pandas Series. I have not tested it on numpy arrays. You can try your luck.Inputs:X_train: a pandas dataframeX_test: a pandas dataframequantile: float (0.75): Define a threshold for IQR for outlier detection. Could be any float between 0 and 1. If quantile is set toNone, then no outlier detection will take place.cat_fill_value: string ("missing") or a dictionary: Define a fill value for missing categories in your object or categorical variables. This is a global default for your entire dataset. You can also give a dictionary where you specify different fill values for different columns.num_fill_value: integer (99) or float value (999.0) or a dictionary: Define a fill value for missing numbers in your integer or float variables.  This is a global default for your entire dataset. You can also give a dictionary where you specify different fill values for different columns.rare_threshold: float (0.05):  Define a threshold for rare categories. If a certain category in a column is less than say 5% (0.05) of samples, then it will be considered rare. All rare categories in that column will be merged under a new category named "Rare".correlation_threshold: float (0.8): Define a correlation limit. Anything above this limit, if two variables are correlated, one of them will be dropped. The program will tell you which variable is being dropped. You can switch the sequence of variables in your dataset if you want the one or other dropped.DataSchemaCheckeris a scikit-learn transformer. It checks you data against a given schemaInputs:schema: dictionary. A schema (dict) is a dictionary that maps column names to data types. This schema will determine the data types of whatever dataframe you want to comply with.DataSchemaChecker has two methods:fitmethod: Checks if the given dataframe matches the schema and displays a table of errors if any.transformmethod: Transforms the given dataframe's dtypes to the given schema and displays errors if any.Maintainers@AutoViMLContributingSeethe contributing file!PRs accepted.LicenseApache License 2.0 © 2020 Ram SeshadriNote of GratitudeThis libray would not have been possible without the help of ChatGPT and Bard. This library is dedicated to the thousands of people who worked to create LLM's.DISCLAIMERThis project is not an official Google project. It is not supported by Google and Google specifically disclaims all warranties as to its quality, merchantability, or fitness for a particular purpose.