Dataset parent of Analysis, scatter plot and pair_plot dirty scripts

1 files changed, 3 insertions, 12 deletions

diff --git a/src/describe.py b/src/describe.py
index 7a968f1..4a3c5bc 100644
--- a/src/describe.py
+++ b/src/describe.py
@@ -1,20 +1,11 @@
 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)
+        raise "Usage: {} dataset_path".format(sys.argv[0])
+    a = Analysis(sys.argv[1])
     a.describe()
-    print(df.describe())
+    print(a.df_scores.describe())