diff options
Diffstat (limited to 'src/describe.py')
| -rw-r--r-- | src/describe.py | 20 |
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()) |
