#!/usr/bin/env python3

import sys
import pandas
import matplotlib.pyplot as plt
import seaborn as sns
import plotly.express as px


def main():
    if (len(sys.argv) == 1):
        print("scatter_plot.py: missing dataset")
    else:
        dataset = open(str(sys.argv[1]), 'r')
        df = pandas.read_csv(dataset)
        df = df.drop(columns=['Index', 'First Name', 'Last Name', 'Birthday', 'Best Hand'])
        sns.pairplot(df, hue='Hogwarts House', markers=".")
        plt.show()
        #fig = px.scatter_matrix(df, color="Hogwarts House")
        #fig.show()
        dataset.close()

if __name__ == "__main__":
    # execute only if run as a script
    main()
