dptools 0.2.1

Data Preprocessing Tools

dptools: data preprocessing functions for Python

---

---

Overview

The dptools python package provides helper functions to simplify common data preprocessing tasks, including feature engineering, working with missing values, aggregating data and more.

The package currently features the following functions:

• add_date_features(): adds date-based features
• add_text_features(): adds text-based features
• aggregate_data(): adds aggregation-based features
• fill_missings(): replaces missings with specific values
• print_missings(): counts missing values and prints the results
• label_encoding(): performs label encoding on partitioned data
• print_factor_levels(): prints levels of categorical features

Installation

The latest stable release can be installed from PyPI:

pip install dptools

You may also install the development version from Github:

pip install git+https://github.com/kozodoi/dptools.git

After the installation, you can import the included functions:

from dptools import *

Examples

This section contains a few examples of using functions from dptools for different data preprocessing tasks.

Creating a toy data set

First, let us create a toy data frame to demonstarte the package functionality.

# import dependecies
import pandas as pd
import numpy as np

# create data frame
data = {'age': [27, np.nan, 30, 25, np.nan], 
        'height': [170, 168, 173, 177, 165], 
        'gender': ['female', 'male', np.nan, 'male', 'female'],
        'income': ['high', 'medium', 'low', 'low', 'no income']}
df = pd.DataFrame(data)

age	height	gender	income
27.0	170	female	high
NaN	168	male	medium
30.0	173	NaN	low
25.0	177	male	low
NaN	165	female	no income

Working with missings

# print statistics on missing values
from dptools import print_missings
print_missings(df)

	Total	Percent
age	2	0.4
gender	1	0.2

Aggregating features

# aggregating the data
from dptools import aggregate_data
df_new = aggregate_data(df, group_var = 'gender', num_stats = ['mean', 'max'])

gender	age_mean	age_max	height_mean	height_max	income_count	income_mode
female	27.0	27.0	167.5	170	2	0
male	25.0	25.0	172.5	177	2	1

Feature engineering

# creating text-based features
from dptools import add_text_features
df_new = add_text_features(df, string_vars = ['income'])

age	height	gender	income_word_count	income_char_count	income_tfidf_0	...	income_tfidf_3
27.0	170	female	1	4	1.0	...	0.0
NaN	168	male	1	6	0.0	...	1.0
30.0	173	NaN	1	3	0.0	...	0.0
25.0	177	male	1	3	0.0	...	0.0
NaN	165	female	2	9	0.0	...	0.0

Dependencies

Installation requires Python 3.6+ and the following packages:

• numpy
• pandas
• sklearn
• scipy

Feedback

In case you need help on the included data preprocseeing functions or you want to report an issue, please do so at the corresponding GitHub page.