Handling Missing Data - Example - Part 2

Let’s start with LoanDurationDays. Since we are filling missing values with the number of days between LoanStartDate and LoanEndDate, it’s important that we first ensure that the dates are in correct format.

This code snippet cleans and standardizes date data in a DataFrame, specifically in the 'LoanStartDate' column. It defines a function, parse_dates, to convert dates into a consistent format, handling two common formats and marking unparseable dates as missing (pd.NaT). This function is applied to the 'LoanStartDate' column of loan_data_cleaned.

# Fix issues with start date and end date
loan_data_cleaned = loan_data_cleaned.copy()
from dateutil.parser import parse

def parse_dates(date):
    for fmt in ("%d/%m/%y", "%Y-%m-%d"):  # Add more formats here if needed
        try:
            return pd.to_datetime(date, format=fmt)
        except ValueError:
            continue
    return pd.NaT  # Return NaT for unparseable dates

# Apply the custom parse function to the 'LoanStartDate' column
loan_data_cleaned['LoanStartDate'] = loan_data_cleaned['LoanStartDate'].apply(parse_dates)

# Check for any NaT values indicating invalid or unparseable dates
invalid_dates = loan_data_cleaned[loan_data_cleaned['LoanStartDate'].isna()]
print("Invalid or unparseable dates:\n", invalid_dates[['CustomerName', 'LoanStartDate']])

We will apply the same date transformation to LoanEndDate also.

# Apply the custom parse function to the 'LoanStartDate' column
loan_data_cleaned['LoanEndDate'] = loan_data_cleaned['LoanEndDate'].apply(parse_dates)
loan_data_cleaned.head()

At this stage all the dates have been formatted to a common format, and invalid dates have been marked is NoT. We should decide how to handle these NaT values before calculating the LoanDurationDays. Typically, if you do not have a reliable way to determine what the correct start date should be for the entries with NaT, you might consider:

  • Removing those rows, especially if they are a small fraction of the data.
  • Imputing them, if you have a business rule or logic for imputation.

In our case we will remove the rows containing NaT values.

# Remove rows where 'LoanStartDate' is NaT
loan_data_cleaned = loan_data_cleaned.dropna(subset=['LoanStartDate'])

# Check the result
loan_data_cleaned['LoanStartDate'].isna().sum(), loan_data_cleaned.shape
loan_data_cleaned.head()

Once you've handled the NaT values, here's how you could fill in the missing LoanDurationDays with the number of days between the start and end date:

# Assuming 'LoanStartDate' and 'LoanEndDate' have been converted to datetime and
# any NoT or incorrect dates have been handled appropriately.

# Calculate the duration only for the rows where 'LoanDurationDays' is missing
loan_data_cleaned['LoanDurationDays'] = loan_data_cleaned.apply(
    lambda row: (row['LoanEndDate'] - row['LoanStartDate']).days
    if pd.isnull(row['LoanDurationDays']) else row['LoanDurationDays'],
    axis=1
)

Make sure both LoanStartDate and LoanEndDate are in the datetime format before performing this operation. If they are not, convert them using pd.to_datetime(). The lambda function in the apply method calculates the duration only for the rows with missing LoanDurationDays. If LoanDurationDays is not missing, the original value is retained.

Related Downloads

Finance Train Premium
Accelerate your finance career with cutting-edge data skills.
Join Finance Train Premium for unlimited access to a growing library of ebooks, projects and code examples covering financial modeling, data analysis, data science, machine learning, algorithmic trading strategies, and more applied to real-world finance scenarios.
I WANT TO JOIN
JOIN 30,000 DATA PROFESSIONALS

Free Guides - Getting Started with R and Python

Enter your name and email address below and we will email you the guides for R programming and Python.

Saylient AI Logo

Accelerate your finance career with cutting-edge data skills.

Join Finance Train Premium for unlimited access to a growing library of ebooks, projects and code examples covering financial modeling, data analysis, data science, machine learning, algorithmic trading strategies, and more applied to real-world finance scenarios.