src/scatter_plot.py


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17

#!/bin/python3

import sys

import matplotlib.pyplot as plt

from dataset import Dataset


if __name__ == '__main__':
    if len(sys.argv) != 2:
        raise "Usage: {} dataset_path".format(sys.argv[0])
    d = Dataset(sys.argv[1])
    plt.scatter(d.df['astronomy'], d.df['defense_dark_arts'], s=5)
    plt.xlabel('astronomy')
    plt.ylabel('defense_dark_arts')
    plt.show()