1. Learn
    2. /
  2. Courses
    3. /
  3. Exploratory Data Analysis in Python

Exercise

Extracting features for correlation

In this exercise, you'll work with a version of the salaries dataset containing a new column called "date_of_response".

The dataset has been read in as a pandas DataFrame, with "date_of_response" as a datetime data type.

Your task is to extract datetime attributes from this column and then create a heat map to visualize the correlation coefficients between variables.

Seaborn has been imported for you as sns, pandas as pd, and matplotlib.pyplot as plt.

Instructions

100 XP
  • Extract the month from "date_of_response", storing it as a column called "month".
  • Create the "weekday" column, containing the weekday that the participants completed the survey.
  • Plot a heat map, including the Pearson correlation coefficient scores.