pair_plot.py

import numpy as np
import pandas as pd
import seaborn as sns
import matplotlib.pyplot as plt


classes = ['Hogwarts House', 'Arithmancy', 'Astronomy', 'Herbology',
           'Defense Against the Dark Arts', 'Divination', 'Muggle Studies',
           'Ancient Runes', 'History of Magic', 'Transfiguration', 'Potions',
           'Care of Magical Creatures', 'Charms', 'Flying']


# Pair plot
def pair_plot (filename : str):
    # Read data
    try:
        df = pd.read_csv(filename)
    except:
        print("error when trying to read dataset", file=sys.stderr)
        sys.exit(1)
    
    # Select features
    df = df[classes]

    # Draw pairplot
    sns.pairplot(df, hue="Hogwarts House", diag_kind="hist")

    # Show plot
    plt.show()


if __name__ == "__main__":
    pair_plot("data/dataset_train.csv")