Custom statistics modulde, describe program

1 files changed, 20 insertions, 0 deletions

diff --git a/src/describe.py b/src/describe.py
index e69de29..7a968f1 100644
--- a/src/describe.py
+++ b/src/describe.py
@@ -0,0 +1,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())