src/describe.py


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

import sys

import pandas as pd
from analysis import Analysis


if __name__ == "__main__":
    if len(sys.argv) != 2:
        print("Usage: {} dataset_path".format(sys.argv[0]))
        sys.exit(1)
    try:
        df = pd.read_csv(sys.argv[1])
    except FileNotFoundError:
        print("Could not find dataset at: {}".format(sys.argv[1]))
        sys.exit(1)
    df = df.loc[:, 'Arithmancy':'Flying']
    df.dropna(inplace=True)
    a = Analysis(df)
    a.describe()
    print(df.describe())